withchao
1 year ago
parent
006766cd14
commit
fcda73f4bc
@ -0,0 +1,148 @@
|
|||||||
|
package msggateway
|
||||||
|
|
||||||
|
//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)
|
||||||
|
// }
|
||||||
|
// }
|
||||||
|
//}
|
||||||
Loading…
Reference in new issue