blackteay/open-im-server
1
0
Fork 0
mirror of https://github.com/openimsdk/open-im-server.git synced 2025-04-06 04:15:46 +08:00
Code Issues Packages Projects Releases Wiki Activity

test modify

Browse Source
This commit is contained in:
Gordon 2021-11-30 19:37:14 +08:00
parent f18f54afc6
commit b416cd380a
1 changed files with 6 additions and 26 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
pkg/common/db/redisModel.go
View File

@ -3,7 +3,6 @@ package db
import ( import (
"Open_IM/pkg/common/constant" "Open_IM/pkg/common/constant"
log2 "Open_IM/pkg/common/log" log2 "Open_IM/pkg/common/log"
"Open_IM/pkg/utils"
"github.com/garyburd/redigo/redis" "github.com/garyburd/redigo/redis"
) )
@ -77,38 +76,19 @@ func (d *DataBases) DelAppleDeviceToken(accountAddress string) (err error) {
//Store userid and platform class to redis //Store userid and platform class to redis
func (d *DataBases) AddTokenFlag(userID string, platformID int32, token string, flag int) error { func (d *DataBases) AddTokenFlag(userID string, platformID int32, token string, flag int) error {
key := uidPidToken + userID + ":" + constant.PlatformIDToName(platformID) key := uidPidToken + userID + ":" + constant.PlatformIDToName(platformID)
var m map[string]int log2.NewDebug("", "add token key is ", key)
m = make(map[string]int) _, err1 := d.Exec("HSet", key, token, flag)
ls, err := redis.String(d.Exec("GET", key))
if err != nil && err != redis.ErrNil {
return err
}
if err == redis.ErrNil {
} else {
_ = utils.JsonStringToStruct(ls, &m)
}
m[token] = flag
s := utils.StructToJsonString(m)
_, err1 := d.Exec("SET", key, s)
return err1 return err1
} }
func (d *DataBases) GetTokenMapByUidPid(userID, platformID string) (m map[string]int, e error) { func (d *DataBases) GetTokenMapByUidPid(userID, platformID string) (map[string]int, error) {
key := uidPidToken + userID + ":" + platformID key := uidPidToken + userID + ":" + platformID
log2.NewDebug("", "key is ", key) log2.NewDebug("", "get token key is ", key)
s, e := redis.String(d.Exec("GET", key)) return redis.IntMap(d.Exec("HGETALL", key))
if e != nil {
return nil, e
} else {
m = make(map[string]int)
_ = utils.JsonStringToStruct(s, &m)
return m, nil
}
} }
func (d *DataBases) SetTokenMapByUidPid(userID string, platformID int32, m map[string]int) error { func (d *DataBases) SetTokenMapByUidPid(userID string, platformID int32, m map[string]int) error {
key := uidPidToken + userID + ":" + constant.PlatformIDToName(platformID) key := uidPidToken + userID + ":" + constant.PlatformIDToName(platformID)
s := utils.StructToJsonString(m) _, err := d.Exec("SET", key, m)
_, err := d.Exec("SET", key, s)
return err return err
} }