|
|
|
|
@ -2,174 +2,180 @@ package msggateway
|
|
|
|
|
|
|
|
|
|
import (
|
|
|
|
|
"context"
|
|
|
|
|
"github.com/openimsdk/open-im-server/v3/pkg/common/storage/cache/cachekey"
|
|
|
|
|
"github.com/openimsdk/open-im-server/v3/pkg/util/useronline"
|
|
|
|
|
"encoding/json"
|
|
|
|
|
"github.com/openimsdk/protocol/constant"
|
|
|
|
|
"github.com/openimsdk/protocol/sdkws"
|
|
|
|
|
"github.com/openimsdk/tools/log"
|
|
|
|
|
"github.com/openimsdk/tools/mcontext"
|
|
|
|
|
"github.com/redis/go-redis/v9"
|
|
|
|
|
"math/rand"
|
|
|
|
|
"strconv"
|
|
|
|
|
"github.com/openimsdk/tools/utils/datautil"
|
|
|
|
|
"github.com/openimsdk/tools/utils/idutil"
|
|
|
|
|
"google.golang.org/protobuf/proto"
|
|
|
|
|
"sync"
|
|
|
|
|
"time"
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
func (ws *WsServer) SubscriberUserOnlineStatusChanges(rdb redis.UniversalClient) {
|
|
|
|
|
ctx := mcontext.SetOperationID(context.Background(), cachekey.OnlineChannel+strconv.FormatUint(rand.Uint64(), 10))
|
|
|
|
|
for message := range rdb.Subscribe(ctx, cachekey.OnlineChannel).Channel() {
|
|
|
|
|
userID, platformIDs, err := useronline.ParseUserOnlineStatus(message.Payload)
|
|
|
|
|
if err != nil {
|
|
|
|
|
log.ZError(ctx, "OnlineCache redis subscribe parseUserOnlineStatus", err, "payload", message.Payload, "channel", message.Channel)
|
|
|
|
|
func (ws *WsServer) subscriberUserOnlineStatusChanges(ctx context.Context, userID string, platformIDs []int32) {
|
|
|
|
|
if ws.clients.RecvSubChange(userID, platformIDs) {
|
|
|
|
|
log.ZDebug(ctx, "gateway receive subscription message and go back online", "userID", userID, "platformIDs", platformIDs)
|
|
|
|
|
} else {
|
|
|
|
|
log.ZDebug(ctx, "gateway ignore user online status changes", "userID", userID, "platformIDs", platformIDs)
|
|
|
|
|
}
|
|
|
|
|
ws.pushUserIDOnlineStatus(ctx, userID, platformIDs)
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func (ws *WsServer) SubUserOnlineStatus(ctx context.Context, client *Client, data *Req) ([]byte, error) {
|
|
|
|
|
var sub sdkws.SubUserOnlineStatus
|
|
|
|
|
if err := proto.Unmarshal(data.Data, &sub); err != nil {
|
|
|
|
|
return nil, err
|
|
|
|
|
}
|
|
|
|
|
ws.subscription.Sub(client, sub.SubscribeUserID, sub.UnsubscribeUserID)
|
|
|
|
|
var resp sdkws.SubUserOnlineStatusTips
|
|
|
|
|
if len(sub.SubscribeUserID) > 0 {
|
|
|
|
|
resp.Subscribers = make([]*sdkws.SubUserOnlineStatusElem, 0, len(sub.SubscribeUserID))
|
|
|
|
|
for _, userID := range sub.SubscribeUserID {
|
|
|
|
|
platformIDs, err := ws.online.GetUserOnlinePlatform(ctx, userID)
|
|
|
|
|
if err != nil {
|
|
|
|
|
return nil, err
|
|
|
|
|
}
|
|
|
|
|
resp.Subscribers = append(resp.Subscribers, &sdkws.SubUserOnlineStatusElem{
|
|
|
|
|
UserID: userID,
|
|
|
|
|
OnlinePlatformIDs: platformIDs,
|
|
|
|
|
})
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
return proto.Marshal(&resp)
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
type subClient struct {
|
|
|
|
|
clients map[string]*Client
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func newSubscription() *Subscription {
|
|
|
|
|
return &Subscription{
|
|
|
|
|
userIDs: make(map[string]*subClient),
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
type Subscription struct {
|
|
|
|
|
lock sync.RWMutex
|
|
|
|
|
userIDs map[string]*subClient
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func (s *Subscription) GetClient(userID string) []*Client {
|
|
|
|
|
s.lock.RLock()
|
|
|
|
|
defer s.lock.RUnlock()
|
|
|
|
|
cs, ok := s.userIDs[userID]
|
|
|
|
|
if !ok {
|
|
|
|
|
return nil
|
|
|
|
|
}
|
|
|
|
|
clients := make([]*Client, 0, len(cs.clients))
|
|
|
|
|
for _, client := range cs.clients {
|
|
|
|
|
clients = append(clients, client)
|
|
|
|
|
}
|
|
|
|
|
return clients
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func (s *Subscription) DelClient(client *Client) {
|
|
|
|
|
client.subLock.Lock()
|
|
|
|
|
userIDs := datautil.Keys(client.subUserIDs)
|
|
|
|
|
for _, userID := range userIDs {
|
|
|
|
|
delete(client.subUserIDs, userID)
|
|
|
|
|
}
|
|
|
|
|
client.subLock.Unlock()
|
|
|
|
|
if len(userIDs) == 0 {
|
|
|
|
|
return
|
|
|
|
|
}
|
|
|
|
|
addr := client.ctx.GetRemoteAddr()
|
|
|
|
|
s.lock.Lock()
|
|
|
|
|
defer s.lock.Unlock()
|
|
|
|
|
for _, userID := range userIDs {
|
|
|
|
|
sub, ok := s.userIDs[userID]
|
|
|
|
|
if !ok {
|
|
|
|
|
continue
|
|
|
|
|
}
|
|
|
|
|
if ws.clients.RecvSubChange(userID, platformIDs) {
|
|
|
|
|
log.ZDebug(ctx, "gateway receive subscription message and go back online", "userID", userID, "platformIDs", platformIDs)
|
|
|
|
|
} else {
|
|
|
|
|
log.ZDebug(ctx, "gateway ignore user online status changes", "userID", userID, "platformIDs", platformIDs)
|
|
|
|
|
delete(sub.clients, addr)
|
|
|
|
|
if len(sub.clients) == 0 {
|
|
|
|
|
delete(s.userIDs, userID)
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
//import (
|
|
|
|
|
// "context"
|
|
|
|
|
// "encoding/json"
|
|
|
|
|
// "github.com/openimsdk/protocol/constant"
|
|
|
|
|
// "github.com/openimsdk/protocol/sdkws"
|
|
|
|
|
// "github.com/openimsdk/tools/log"
|
|
|
|
|
// "github.com/openimsdk/tools/utils/datautil"
|
|
|
|
|
// "github.com/openimsdk/tools/utils/idutil"
|
|
|
|
|
// "sync"
|
|
|
|
|
// "time"
|
|
|
|
|
//)
|
|
|
|
|
//
|
|
|
|
|
//type subClient struct {
|
|
|
|
|
// clients map[string]*Client
|
|
|
|
|
//}
|
|
|
|
|
//
|
|
|
|
|
//func newSubscription() *Subscription {
|
|
|
|
|
// return &Subscription{
|
|
|
|
|
// userIDs: make(map[string]*subClient),
|
|
|
|
|
// }
|
|
|
|
|
//}
|
|
|
|
|
//
|
|
|
|
|
//type Subscription struct {
|
|
|
|
|
// lock sync.RWMutex
|
|
|
|
|
// userIDs map[string]*subClient
|
|
|
|
|
//}
|
|
|
|
|
//
|
|
|
|
|
//func (s *Subscription) GetClient(userID string) []*Client {
|
|
|
|
|
// s.lock.RLock()
|
|
|
|
|
// defer s.lock.RUnlock()
|
|
|
|
|
// cs, ok := s.userIDs[userID]
|
|
|
|
|
// if !ok {
|
|
|
|
|
// return nil
|
|
|
|
|
// }
|
|
|
|
|
// clients := make([]*Client, 0, len(cs.clients))
|
|
|
|
|
// for _, client := range cs.clients {
|
|
|
|
|
// clients = append(clients, client)
|
|
|
|
|
// }
|
|
|
|
|
// return clients
|
|
|
|
|
//}
|
|
|
|
|
//
|
|
|
|
|
//func (s *Subscription) DelClient(client *Client) {
|
|
|
|
|
// client.subLock.Lock()
|
|
|
|
|
// userIDs := datautil.Keys(client.subUserIDs)
|
|
|
|
|
// for _, userID := range userIDs {
|
|
|
|
|
// delete(client.subUserIDs, userID)
|
|
|
|
|
// }
|
|
|
|
|
// client.subLock.Unlock()
|
|
|
|
|
// if len(userIDs) == 0 {
|
|
|
|
|
// return
|
|
|
|
|
// }
|
|
|
|
|
// addr := client.ctx.GetRemoteAddr()
|
|
|
|
|
// s.lock.Lock()
|
|
|
|
|
// defer s.lock.Unlock()
|
|
|
|
|
// for _, userID := range userIDs {
|
|
|
|
|
// sub, ok := s.userIDs[userID]
|
|
|
|
|
// if !ok {
|
|
|
|
|
// continue
|
|
|
|
|
// }
|
|
|
|
|
// delete(sub.clients, addr)
|
|
|
|
|
// if len(sub.clients) == 0 {
|
|
|
|
|
// delete(s.userIDs, userID)
|
|
|
|
|
// }
|
|
|
|
|
// }
|
|
|
|
|
//}
|
|
|
|
|
//
|
|
|
|
|
//func (s *Subscription) Sub(client *Client, addUserIDs, delUserIDs []string) {
|
|
|
|
|
// if len(addUserIDs)+len(delUserIDs) == 0 {
|
|
|
|
|
// return
|
|
|
|
|
// }
|
|
|
|
|
// var (
|
|
|
|
|
// del = make(map[string]struct{})
|
|
|
|
|
// add = make(map[string]struct{})
|
|
|
|
|
// )
|
|
|
|
|
// client.subLock.Lock()
|
|
|
|
|
// for _, userID := range delUserIDs {
|
|
|
|
|
// if _, ok := client.subUserIDs[userID]; !ok {
|
|
|
|
|
// continue
|
|
|
|
|
// }
|
|
|
|
|
// del[userID] = struct{}{}
|
|
|
|
|
// delete(client.subUserIDs, userID)
|
|
|
|
|
// }
|
|
|
|
|
// for _, userID := range addUserIDs {
|
|
|
|
|
// delete(del, userID)
|
|
|
|
|
// if _, ok := client.subUserIDs[userID]; ok {
|
|
|
|
|
// continue
|
|
|
|
|
// }
|
|
|
|
|
// client.subUserIDs[userID] = struct{}{}
|
|
|
|
|
// }
|
|
|
|
|
// client.subLock.Unlock()
|
|
|
|
|
// if len(del)+len(add) == 0 {
|
|
|
|
|
// return
|
|
|
|
|
// }
|
|
|
|
|
// addr := client.ctx.GetRemoteAddr()
|
|
|
|
|
// s.lock.Lock()
|
|
|
|
|
// defer s.lock.Unlock()
|
|
|
|
|
// for userID := range del {
|
|
|
|
|
// sub, ok := s.userIDs[userID]
|
|
|
|
|
// if !ok {
|
|
|
|
|
// continue
|
|
|
|
|
// }
|
|
|
|
|
// delete(sub.clients, addr)
|
|
|
|
|
// if len(sub.clients) == 0 {
|
|
|
|
|
// delete(s.userIDs, userID)
|
|
|
|
|
// }
|
|
|
|
|
// }
|
|
|
|
|
// for userID := range add {
|
|
|
|
|
// sub, ok := s.userIDs[userID]
|
|
|
|
|
// if !ok {
|
|
|
|
|
// sub = &subClient{clients: make(map[string]*Client)}
|
|
|
|
|
// s.userIDs[userID] = sub
|
|
|
|
|
// }
|
|
|
|
|
// sub.clients[addr] = client
|
|
|
|
|
// }
|
|
|
|
|
//}
|
|
|
|
|
//
|
|
|
|
|
//func (ws *WsServer) pushUserIDOnlineStatus(ctx context.Context, userID string, platformIDs []int32) {
|
|
|
|
|
// clients := ws.subscription.GetClient(userID)
|
|
|
|
|
// if len(clients) == 0 {
|
|
|
|
|
// return
|
|
|
|
|
// }
|
|
|
|
|
// msgContent, err := json.Marshal(platformIDs)
|
|
|
|
|
// if err != nil {
|
|
|
|
|
// log.ZError(ctx, "pushUserIDOnlineStatus json.Marshal", err)
|
|
|
|
|
// return
|
|
|
|
|
// }
|
|
|
|
|
// now := time.Now().UnixMilli()
|
|
|
|
|
// msgID := idutil.GetMsgIDByMD5(userID)
|
|
|
|
|
// msg := &sdkws.MsgData{
|
|
|
|
|
// SendID: userID,
|
|
|
|
|
// ClientMsgID: msgID,
|
|
|
|
|
// ServerMsgID: msgID,
|
|
|
|
|
// SenderPlatformID: constant.AdminPlatformID,
|
|
|
|
|
// SessionType: constant.NotificationChatType,
|
|
|
|
|
// ContentType: constant.UserSubscribeOnlineStatusNotification,
|
|
|
|
|
// Content: msgContent,
|
|
|
|
|
// SendTime: now,
|
|
|
|
|
// CreateTime: now,
|
|
|
|
|
// }
|
|
|
|
|
// for _, client := range clients {
|
|
|
|
|
// msg.RecvID = client.UserID
|
|
|
|
|
// if err := client.PushMessage(ctx, msg); err != nil {
|
|
|
|
|
// log.ZError(ctx, "UserSubscribeOnlineStatusNotification push failed", err, "userID", client.UserID, "platformID", client.PlatformID, "changeUserID", userID, "content", msgContent)
|
|
|
|
|
// }
|
|
|
|
|
// }
|
|
|
|
|
//}
|
|
|
|
|
func (s *Subscription) Sub(client *Client, addUserIDs, delUserIDs []string) {
|
|
|
|
|
if len(addUserIDs)+len(delUserIDs) == 0 {
|
|
|
|
|
return
|
|
|
|
|
}
|
|
|
|
|
var (
|
|
|
|
|
del = make(map[string]struct{})
|
|
|
|
|
add = make(map[string]struct{})
|
|
|
|
|
)
|
|
|
|
|
client.subLock.Lock()
|
|
|
|
|
for _, userID := range delUserIDs {
|
|
|
|
|
if _, ok := client.subUserIDs[userID]; !ok {
|
|
|
|
|
continue
|
|
|
|
|
}
|
|
|
|
|
del[userID] = struct{}{}
|
|
|
|
|
delete(client.subUserIDs, userID)
|
|
|
|
|
}
|
|
|
|
|
for _, userID := range addUserIDs {
|
|
|
|
|
delete(del, userID)
|
|
|
|
|
if _, ok := client.subUserIDs[userID]; ok {
|
|
|
|
|
continue
|
|
|
|
|
}
|
|
|
|
|
client.subUserIDs[userID] = struct{}{}
|
|
|
|
|
}
|
|
|
|
|
client.subLock.Unlock()
|
|
|
|
|
if len(del)+len(add) == 0 {
|
|
|
|
|
return
|
|
|
|
|
}
|
|
|
|
|
addr := client.ctx.GetRemoteAddr()
|
|
|
|
|
s.lock.Lock()
|
|
|
|
|
defer s.lock.Unlock()
|
|
|
|
|
for userID := range del {
|
|
|
|
|
sub, ok := s.userIDs[userID]
|
|
|
|
|
if !ok {
|
|
|
|
|
continue
|
|
|
|
|
}
|
|
|
|
|
delete(sub.clients, addr)
|
|
|
|
|
if len(sub.clients) == 0 {
|
|
|
|
|
delete(s.userIDs, userID)
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
for userID := range add {
|
|
|
|
|
sub, ok := s.userIDs[userID]
|
|
|
|
|
if !ok {
|
|
|
|
|
sub = &subClient{clients: make(map[string]*Client)}
|
|
|
|
|
s.userIDs[userID] = sub
|
|
|
|
|
}
|
|
|
|
|
sub.clients[addr] = client
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func (ws *WsServer) pushUserIDOnlineStatus(ctx context.Context, userID string, platformIDs []int32) {
|
|
|
|
|
clients := ws.subscription.GetClient(userID)
|
|
|
|
|
if len(clients) == 0 {
|
|
|
|
|
return
|
|
|
|
|
}
|
|
|
|
|
msgContent, err := json.Marshal(platformIDs)
|
|
|
|
|
if err != nil {
|
|
|
|
|
log.ZError(ctx, "pushUserIDOnlineStatus json.Marshal", err)
|
|
|
|
|
return
|
|
|
|
|
}
|
|
|
|
|
now := time.Now().UnixMilli()
|
|
|
|
|
msgID := idutil.GetMsgIDByMD5(userID)
|
|
|
|
|
msg := &sdkws.MsgData{
|
|
|
|
|
SendID: userID,
|
|
|
|
|
ClientMsgID: msgID,
|
|
|
|
|
ServerMsgID: msgID,
|
|
|
|
|
SenderPlatformID: constant.AdminPlatformID,
|
|
|
|
|
SessionType: constant.NotificationChatType,
|
|
|
|
|
ContentType: constant.UserSubscribeOnlineStatusNotification,
|
|
|
|
|
Content: msgContent,
|
|
|
|
|
SendTime: now,
|
|
|
|
|
CreateTime: now,
|
|
|
|
|
}
|
|
|
|
|
for _, client := range clients {
|
|
|
|
|
msg.RecvID = client.UserID
|
|
|
|
|
if err := client.PushMessage(ctx, msg); err != nil {
|
|
|
|
|
log.ZError(ctx, "UserSubscribeOnlineStatusNotification push failed", err, "userID", client.UserID, "platformID", client.PlatformID, "changeUserID", userID, "content", msgContent)
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
withchao
1 year ago