|
|
@ -21,6 +21,8 @@ import (
|
|
|
|
"strconv"
|
|
|
|
"strconv"
|
|
|
|
"time"
|
|
|
|
"time"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
"github.com/OpenIMSDK/tools/log"
|
|
|
|
|
|
|
|
|
|
|
|
"github.com/OpenIMSDK/protocol/constant"
|
|
|
|
"github.com/OpenIMSDK/protocol/constant"
|
|
|
|
|
|
|
|
|
|
|
|
"github.com/OpenIMSDK/protocol/user"
|
|
|
|
"github.com/OpenIMSDK/protocol/user"
|
|
|
@ -51,7 +53,7 @@ type UserCache interface {
|
|
|
|
GetUserGlobalRecvMsgOpt(ctx context.Context, userID string) (opt int, err error)
|
|
|
|
GetUserGlobalRecvMsgOpt(ctx context.Context, userID string) (opt int, err error)
|
|
|
|
DelUsersGlobalRecvMsgOpt(userIDs ...string) UserCache
|
|
|
|
DelUsersGlobalRecvMsgOpt(userIDs ...string) UserCache
|
|
|
|
GetUserStatus(ctx context.Context, userIDs []string) ([]*user.OnlineStatus, error)
|
|
|
|
GetUserStatus(ctx context.Context, userIDs []string) ([]*user.OnlineStatus, error)
|
|
|
|
SetUserStatus(ctx context.Context, list []*user.OnlineStatus) error
|
|
|
|
SetUserStatus(ctx context.Context, userID string, status, platformID int32) error
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
type UserCacheRedis struct {
|
|
|
|
type UserCacheRedis struct {
|
|
|
@ -198,58 +200,72 @@ func (u *UserCacheRedis) GetUserStatus(ctx context.Context, userIDs []string) ([
|
|
|
|
return nil, errs.Wrap(err)
|
|
|
|
return nil, errs.Wrap(err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
onlineStatus.UserID = userID
|
|
|
|
onlineStatus.UserID = userID
|
|
|
|
|
|
|
|
onlineStatus.Status = constant.Online
|
|
|
|
res = append(res, &onlineStatus)
|
|
|
|
res = append(res, &onlineStatus)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return res, nil
|
|
|
|
return res, nil
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
// 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, userID string, status, platformID int32) error {
|
|
|
|
for _, status := range list {
|
|
|
|
UserIDNum := crc32.ChecksumIEEE([]byte(userID))
|
|
|
|
var isNewKey int64
|
|
|
|
|
|
|
|
UserIDNum := crc32.ChecksumIEEE([]byte(status.UserID))
|
|
|
|
|
|
|
|
modKey := strconv.Itoa(int(UserIDNum % statusMod))
|
|
|
|
modKey := strconv.Itoa(int(UserIDNum % statusMod))
|
|
|
|
key := olineStatusKey + modKey
|
|
|
|
key := olineStatusKey + modKey
|
|
|
|
jsonData, err := json.Marshal(status)
|
|
|
|
log.ZDebug(ctx, "SetUserStatus args", "userID", userID, "status", status,
|
|
|
|
|
|
|
|
"platformID", platformID, "modKey", modKey, "key", key)
|
|
|
|
|
|
|
|
isNewKey, err := u.rdb.Exists(ctx, key).Result()
|
|
|
|
if err != nil {
|
|
|
|
if err != nil {
|
|
|
|
return errs.Wrap(err)
|
|
|
|
return errs.Wrap(err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
isNewKey, err = u.rdb.Exists(ctx, key).Result()
|
|
|
|
if isNewKey == 0 {
|
|
|
|
|
|
|
|
if status == constant.Online {
|
|
|
|
|
|
|
|
onlineStatus := user.OnlineStatus{
|
|
|
|
|
|
|
|
UserID: userID,
|
|
|
|
|
|
|
|
Status: constant.Online,
|
|
|
|
|
|
|
|
PlatformIDs: []int32{platformID},
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
jsonData, err := json.Marshal(onlineStatus)
|
|
|
|
if err != nil {
|
|
|
|
if err != nil {
|
|
|
|
return errs.Wrap(err)
|
|
|
|
return errs.Wrap(err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
if isNewKey == 0 {
|
|
|
|
_, err = u.rdb.HSet(ctx, key, userID, string(jsonData)).Result()
|
|
|
|
_, err = u.rdb.HSet(ctx, key, status.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 {
|
|
|
|
return nil
|
|
|
|
result, err := u.rdb.HGet(ctx, key, status.UserID).Result()
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
isNil := false
|
|
|
|
|
|
|
|
result, err := u.rdb.HGet(ctx, key, userID).Result()
|
|
|
|
if err != nil {
|
|
|
|
if err != nil {
|
|
|
|
|
|
|
|
if err == redis.Nil {
|
|
|
|
|
|
|
|
isNil = true
|
|
|
|
|
|
|
|
} else {
|
|
|
|
return errs.Wrap(err)
|
|
|
|
return errs.Wrap(err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if status == constant.Offline {
|
|
|
|
|
|
|
|
if isNil {
|
|
|
|
|
|
|
|
log.ZWarn(ctx, "this user not online,maybe trigger order not right",
|
|
|
|
|
|
|
|
err, "userStatus", status)
|
|
|
|
|
|
|
|
return nil
|
|
|
|
|
|
|
|
}
|
|
|
|
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 {
|
|
|
|
|
|
|
|
var newPlatformIDs []int32
|
|
|
|
var newPlatformIDs []int32
|
|
|
|
for _, val := range onlineStatus.PlatformIDs {
|
|
|
|
for _, val := range onlineStatus.PlatformIDs {
|
|
|
|
if val != status.PlatformIDs[0] {
|
|
|
|
if val != platformID {
|
|
|
|
newPlatformIDs = append(newPlatformIDs, val)
|
|
|
|
newPlatformIDs = append(newPlatformIDs, val)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
if newPlatformIDs == nil {
|
|
|
|
if newPlatformIDs == nil {
|
|
|
|
onlineStatus.Status = constant.Offline
|
|
|
|
_, err = u.rdb.HDel(ctx, key, 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 +275,32 @@ 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, userID, string(newjsonData)).Result()
|
|
|
|
if err != nil {
|
|
|
|
if err != nil {
|
|
|
|
return errs.Wrap(err)
|
|
|
|
return errs.Wrap(err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
} else {
|
|
|
|
onlineStatus.Status = constant.Online
|
|
|
|
var onlineStatus user.OnlineStatus
|
|
|
|
// Judging whether to be kicked out.
|
|
|
|
if !isNil {
|
|
|
|
flag := false
|
|
|
|
err = json.Unmarshal([]byte(result), &onlineStatus)
|
|
|
|
for _, val := range onlineStatus.PlatformIDs {
|
|
|
|
if err != nil {
|
|
|
|
if val == status.PlatformIDs[0] {
|
|
|
|
return errs.Wrap(err)
|
|
|
|
flag = true
|
|
|
|
|
|
|
|
break
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
}
|
|
|
|
if !flag {
|
|
|
|
|
|
|
|
onlineStatus.PlatformIDs = append(onlineStatus.PlatformIDs, status.PlatformIDs[0])
|
|
|
|
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
onlineStatus.Status = constant.Online
|
|
|
|
|
|
|
|
onlineStatus.UserID = userID
|
|
|
|
|
|
|
|
onlineStatus.PlatformIDs = append(onlineStatus.PlatformIDs, platformID)
|
|
|
|
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, userID, string(newjsonData)).Result()
|
|
|
|
if err != nil {
|
|
|
|
if err != nil {
|
|
|
|
return errs.Wrap(err)
|
|
|
|
return errs.Wrap(err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
return nil
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
}
|
|
|
|