/
push.go
62 lines (55 loc) · 1.55 KB
/
push.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
package logic
import (
"context"
"github.com/Terry-Mao/goim/internal/logic/model"
log "github.com/golang/glog"
)
// PushKeys push a message by keys.
func (l *Logic) PushKeys(c context.Context, op int32, keys []string, msg []byte) (err error) {
servers, err := l.dao.ServersByKeys(c, keys)
if err != nil {
return
}
pushKeys := make(map[string][]string)
for i, key := range keys {
server := servers[i]
if server != "" && key != "" {
pushKeys[server] = append(pushKeys[server], key)
}
}
for server := range pushKeys {
if err = l.dao.PushMsg(c, op, server, pushKeys[server], msg); err != nil {
return
}
}
return
}
// PushMids push a message by mid.
func (l *Logic) PushMids(c context.Context, op int32, mids []int64, msg []byte) (err error) {
keyServers, _, err := l.dao.KeysByMids(c, mids)
if err != nil {
return
}
keys := make(map[string][]string)
for key, server := range keyServers {
if key == "" || server == "" {
log.Warningf("push key:%s server:%s is empty", key, server)
continue
}
keys[server] = append(keys[server], key)
}
for server, keys := range keys {
if err = l.dao.PushMsg(c, op, server, keys, msg); err != nil {
return
}
}
return
}
// PushRoom push a message by room.
func (l *Logic) PushRoom(c context.Context, op int32, typ, room string, msg []byte) (err error) {
return l.dao.BroadcastRoomMsg(c, op, model.EncodeRoomKey(typ, room), msg)
}
// PushAll push a message to all.
func (l *Logic) PushAll(c context.Context, op, speed int32, msg []byte) (err error) {
return l.dao.BroadcastMsg(c, op, speed, msg)
}