fix: user status change.

pull/1171/head
Gordon 2 years ago
parent 8c787b0bc3
commit 5147467618

@ -289,7 +289,8 @@ func (s *userServer) SubscribeOrCancelUsersStatus(ctx context.Context, req *pbus
} }
// GetUserStatus Get the online status of the user. // GetUserStatus Get the online status of the user.
func (s *userServer) GetUserStatus(ctx context.Context, req *pbuser.GetUserStatusReq) (resp *pbuser.GetUserStatusResp, err error) { func (s *userServer) GetUserStatus(ctx context.Context, req *pbuser.GetUserStatusReq) (resp *pbuser.GetUserStatusResp,
err error) {
onlineStatusList, err := s.UserDatabase.GetUserStatus(ctx, req.UserIDs) onlineStatusList, err := s.UserDatabase.GetUserStatus(ctx, req.UserIDs)
if err != nil { if err != nil {
return nil, err return nil, err
@ -298,7 +299,8 @@ func (s *userServer) GetUserStatus(ctx context.Context, req *pbuser.GetUserStatu
} }
// SetUserStatus Synchronize user's online status. // SetUserStatus Synchronize user's online status.
func (s *userServer) SetUserStatus(ctx context.Context, req *pbuser.SetUserStatusReq) (resp *pbuser.SetUserStatusResp, err error) { func (s *userServer) SetUserStatus(ctx context.Context, req *pbuser.SetUserStatusReq) (resp *pbuser.SetUserStatusResp,
err error) {
err = s.UserDatabase.SetUserStatus(ctx, req.StatusList) err = s.UserDatabase.SetUserStatus(ctx, req.StatusList)
if err != nil { if err != nil {
return nil, err return nil, err

@ -17,6 +17,7 @@ package cache
import ( import (
"context" "context"
"encoding/json" "encoding/json"
"github.com/OpenIMSDK/tools/log"
"hash/crc32" "hash/crc32"
"strconv" "strconv"
"time" "time"
@ -205,12 +206,12 @@ func (u *UserCacheRedis) GetUserStatus(ctx context.Context, userIDs []string) ([
// SetUserStatus Set the user status and save it in redis. // SetUserStatus Set the user status and save it in redis.
func (u *UserCacheRedis) SetUserStatus(ctx context.Context, list []*user.OnlineStatus) error { func (u *UserCacheRedis) SetUserStatus(ctx context.Context, list []*user.OnlineStatus) error {
for _, status := range list { for _, userStatus := range list {
var isNewKey int64 var isNewKey int64
UserIDNum := crc32.ChecksumIEEE([]byte(status.UserID)) UserIDNum := crc32.ChecksumIEEE([]byte(userStatus.UserID))
modKey := strconv.Itoa(int(UserIDNum % statusMod)) modKey := strconv.Itoa(int(UserIDNum % statusMod))
key := olineStatusKey + modKey key := olineStatusKey + modKey
jsonData, err := json.Marshal(status) jsonData, err := json.Marshal(userStatus)
if err != nil { if err != nil {
return errs.Wrap(err) return errs.Wrap(err)
} }
@ -219,37 +220,44 @@ func (u *UserCacheRedis) SetUserStatus(ctx context.Context, list []*user.OnlineS
return errs.Wrap(err) return errs.Wrap(err)
} }
if isNewKey == 0 { if isNewKey == 0 {
_, err = u.rdb.HSet(ctx, key, status.UserID, string(jsonData)).Result() if userStatus.Status == constant.Online {
_, err = u.rdb.HSet(ctx, key, userStatus.UserID, string(jsonData)).Result()
if err != nil { if err != nil {
return errs.Wrap(err) return errs.Wrap(err)
} }
u.rdb.Expire(ctx, key, userOlineStatusExpireTime) u.rdb.Expire(ctx, key, userOlineStatusExpireTime)
}
} else { } else {
result, err := u.rdb.HGet(ctx, key, status.UserID).Result() result, err := u.rdb.HGet(ctx, key, userStatus.UserID).Result()
if err != nil { if err != nil {
//redis do not have this user key
if err == redis.Nil {
if userStatus.Status == constant.Offline {
log.ZWarn(ctx, "this user not online,maybe trigger order not right",
err, "userStatus", userStatus)
continue
}
} else {
return errs.Wrap(err) return errs.Wrap(err)
} }
}
var onlineStatus user.OnlineStatus var onlineStatus user.OnlineStatus
err = json.Unmarshal([]byte(result), &onlineStatus) err = json.Unmarshal([]byte(result), &onlineStatus)
if err != nil { if err != nil {
return errs.Wrap(err) return errs.Wrap(err)
} }
onlineStatus.UserID = status.UserID
if status.Status == constant.Offline { onlineStatus.UserID = userStatus.UserID
if userStatus.Status == constant.Offline {
var newPlatformIDs []int32 var newPlatformIDs []int32
for _, val := range onlineStatus.PlatformIDs { for _, val := range onlineStatus.PlatformIDs {
if val != status.PlatformIDs[0] { if val != userStatus.PlatformIDs[0] {
newPlatformIDs = append(newPlatformIDs, val) newPlatformIDs = append(newPlatformIDs, val)
} }
} }
if newPlatformIDs == nil { if newPlatformIDs == nil {
onlineStatus.Status = constant.Offline _, err = u.rdb.HDel(ctx, key, userStatus.UserID).Result()
onlineStatus.PlatformIDs = []int32{}
newjsonData, err := json.Marshal(&onlineStatus)
if err != nil {
return errs.Wrap(err)
}
_, err = u.rdb.HSet(ctx, key, status.UserID, string(newjsonData)).Result()
if err != nil { if err != nil {
return errs.Wrap(err) return errs.Wrap(err)
} }
@ -259,34 +267,39 @@ func (u *UserCacheRedis) SetUserStatus(ctx context.Context, list []*user.OnlineS
if err != nil { if err != nil {
return errs.Wrap(err) return errs.Wrap(err)
} }
_, err = u.rdb.HSet(ctx, key, status.UserID, string(newjsonData)).Result() _, err = u.rdb.HSet(ctx, key, userStatus.UserID, string(newjsonData)).Result()
if err != nil { if err != nil {
return errs.Wrap(err) return errs.Wrap(err)
} }
} }
} else { } else {
//user all terminal offline directly set online
if onlineStatus.UserID == "" {
onlineStatus.Status = constant.Online onlineStatus.Status = constant.Online
// Judging whether to be kicked out. onlineStatus.UserID = userStatus.UserID
flag := false onlineStatus.PlatformIDs = userStatus.PlatformIDs
for _, val := range onlineStatus.PlatformIDs { newjsonData, err := json.Marshal(&onlineStatus)
if val == status.PlatformIDs[0] { if err != nil {
flag = true return errs.Wrap(err)
break
}
} }
if !flag { _, err = u.rdb.HSet(ctx, key, userStatus.UserID, string(newjsonData)).Result()
onlineStatus.PlatformIDs = append(onlineStatus.PlatformIDs, status.PlatformIDs[0]) if err != nil {
return errs.Wrap(err)
} }
} else {
onlineStatus.Status = constant.Online
onlineStatus.PlatformIDs = append(onlineStatus.PlatformIDs, userStatus.PlatformIDs[0])
newjsonData, err := json.Marshal(&onlineStatus) newjsonData, err := json.Marshal(&onlineStatus)
if err != nil { if err != nil {
return errs.Wrap(err) return errs.Wrap(err)
} }
_, err = u.rdb.HSet(ctx, key, status.UserID, string(newjsonData)).Result() _, err = u.rdb.HSet(ctx, key, userStatus.UserID, string(newjsonData)).Result()
if err != nil { if err != nil {
return errs.Wrap(err) return errs.Wrap(err)
} }
} }
} }
} }
}
return nil return nil
} }

@ -176,7 +176,11 @@ func (u *UserMongoDriver) GetSubscribedList(ctx context.Context, userID string)
bson.M{"user_id": SubscribedPrefix + userID}) bson.M{"user_id": SubscribedPrefix + userID})
err = cursor.Decode(&user) err = cursor.Decode(&user)
if err != nil { if err != nil {
if err == mongo.ErrNoDocuments {
return []string{}, nil
} else {
return nil, errs.Wrap(err) return nil, errs.Wrap(err)
} }
}
return user.UserIDList, nil return user.UserIDList, nil
} }

Loading…
Cancel
Save