mirror of
https://github.com/openimsdk/open-im-server.git
synced 2025-12-09 06:02:54 +08:00
fix: user status change.
This commit is contained in:
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
|
||||||
|
|||||||
89
pkg/common/db/cache/user.go
vendored
89
pkg/common/db/cache/user.go
vendored
@ -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 {
|
||||||
if err != nil {
|
_, err = u.rdb.HSet(ctx, key, userStatus.UserID, string(jsonData)).Result()
|
||||||
return errs.Wrap(err)
|
if err != nil {
|
||||||
|
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 {
|
||||||
return errs.Wrap(err)
|
//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)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
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,31 +267,36 @@ 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 {
|
||||||
onlineStatus.Status = constant.Online
|
//user all terminal offline directly set online
|
||||||
// Judging whether to be kicked out.
|
if onlineStatus.UserID == "" {
|
||||||
flag := false
|
onlineStatus.Status = constant.Online
|
||||||
for _, val := range onlineStatus.PlatformIDs {
|
onlineStatus.UserID = userStatus.UserID
|
||||||
if val == status.PlatformIDs[0] {
|
onlineStatus.PlatformIDs = userStatus.PlatformIDs
|
||||||
flag = true
|
newjsonData, err := json.Marshal(&onlineStatus)
|
||||||
break
|
if err != nil {
|
||||||
|
return errs.Wrap(err)
|
||||||
|
}
|
||||||
|
_, err = u.rdb.HSet(ctx, key, userStatus.UserID, string(newjsonData)).Result()
|
||||||
|
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)
|
||||||
|
if err != nil {
|
||||||
|
return errs.Wrap(err)
|
||||||
|
}
|
||||||
|
_, err = u.rdb.HSet(ctx, key, userStatus.UserID, string(newjsonData)).Result()
|
||||||
|
if err != nil {
|
||||||
|
return errs.Wrap(err)
|
||||||
}
|
}
|
||||||
}
|
|
||||||
if !flag {
|
|
||||||
onlineStatus.PlatformIDs = append(onlineStatus.PlatformIDs, status.PlatformIDs[0])
|
|
||||||
}
|
|
||||||
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 {
|
|
||||||
return errs.Wrap(err)
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@ -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 {
|
||||||
return nil, errs.Wrap(err)
|
if err == mongo.ErrNoDocuments {
|
||||||
|
return []string{}, nil
|
||||||
|
} else {
|
||||||
|
return nil, errs.Wrap(err)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
return user.UserIDList, nil
|
return user.UserIDList, nil
|
||||||
}
|
}
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user