mirror of
https://github.com/openimsdk/open-im-server.git
synced 2026-05-05 17:45:59 +08:00
redis replace to go_redis
This commit is contained in:
Gordon
+141
-174
@@ -1,19 +1,7 @@
|
||||
package db
|
||||
|
||||
import (
|
||||
"Open_IM/pkg/common/config"
|
||||
"Open_IM/pkg/common/constant"
|
||||
log2 "Open_IM/pkg/common/log"
|
||||
pbChat "Open_IM/pkg/proto/chat"
|
||||
pbCommon "Open_IM/pkg/proto/sdk_ws"
|
||||
"Open_IM/pkg/utils"
|
||||
"errors"
|
||||
"fmt"
|
||||
|
||||
"github.com/garyburd/redigo/redis"
|
||||
"github.com/golang/protobuf/jsonpb"
|
||||
//osconfig "google.golang.org/genproto/googleapis/cloud/osconfig/v1alpha"
|
||||
"strconv"
|
||||
)
|
||||
|
||||
const (
|
||||
@@ -54,122 +42,101 @@ func (d *DataBases) Exec(cmd string, key interface{}, args ...interface{}) (inte
|
||||
|
||||
return con.Do(cmd, params...)
|
||||
}
|
||||
func (d *DataBases) JudgeAccountEXISTS(account string) (bool, error) {
|
||||
key := accountTempCode + account
|
||||
return redis.Bool(d.Exec("EXISTS", key))
|
||||
}
|
||||
func (d *DataBases) SetAccountCode(account string, code, ttl int) (err error) {
|
||||
key := accountTempCode + account
|
||||
_, err = d.Exec("SET", key, code, "ex", ttl)
|
||||
return err
|
||||
}
|
||||
func (d *DataBases) GetAccountCode(account string) (string, error) {
|
||||
key := accountTempCode + account
|
||||
return redis.String(d.Exec("GET", key))
|
||||
}
|
||||
|
||||
//Perform seq auto-increment operation of user messages
|
||||
func (d *DataBases) IncrUserSeq(uid string) (uint64, error) {
|
||||
key := userIncrSeq + uid
|
||||
return redis.Uint64(d.Exec("INCR", key))
|
||||
}
|
||||
|
||||
//Get the largest Seq
|
||||
func (d *DataBases) GetUserMaxSeq(uid string) (uint64, error) {
|
||||
key := userIncrSeq + uid
|
||||
return redis.Uint64(d.Exec("GET", key))
|
||||
}
|
||||
|
||||
//set the largest Seq
|
||||
func (d *DataBases) SetUserMaxSeq(uid string, maxSeq uint64) error {
|
||||
key := userIncrSeq + uid
|
||||
_, err := d.Exec("SET", key, maxSeq)
|
||||
return err
|
||||
}
|
||||
|
||||
//Set the user's minimum seq
|
||||
func (d *DataBases) SetUserMinSeq(uid string, minSeq uint32) (err error) {
|
||||
key := userMinSeq + uid
|
||||
_, err = d.Exec("SET", key, minSeq)
|
||||
return err
|
||||
}
|
||||
|
||||
//Get the smallest Seq
|
||||
func (d *DataBases) GetUserMinSeq(uid string) (uint64, error) {
|
||||
key := userMinSeq + uid
|
||||
return redis.Uint64(d.Exec("GET", key))
|
||||
}
|
||||
|
||||
//Store Apple's device token to redis
|
||||
func (d *DataBases) SetAppleDeviceToken(accountAddress, value string) (err error) {
|
||||
key := appleDeviceToken + accountAddress
|
||||
_, err = d.Exec("SET", key, value)
|
||||
return err
|
||||
}
|
||||
|
||||
//Delete Apple device token
|
||||
func (d *DataBases) DelAppleDeviceToken(accountAddress string) (err error) {
|
||||
key := appleDeviceToken + accountAddress
|
||||
_, err = d.Exec("DEL", key)
|
||||
return err
|
||||
}
|
||||
|
||||
//Store userid and platform class to redis
|
||||
func (d *DataBases) AddTokenFlag(userID string, platformID int, token string, flag int) error {
|
||||
key := uidPidToken + userID + ":" + constant.PlatformIDToName(platformID)
|
||||
log2.NewDebug("", "add token key is ", key)
|
||||
_, err1 := d.Exec("HSet", key, token, flag)
|
||||
return err1
|
||||
}
|
||||
|
||||
func (d *DataBases) GetTokenMapByUidPid(userID, platformID string) (map[string]int, error) {
|
||||
key := uidPidToken + userID + ":" + platformID
|
||||
log2.NewDebug("", "get token key is ", key)
|
||||
return redis.IntMap(d.Exec("HGETALL", key))
|
||||
}
|
||||
func (d *DataBases) SetTokenMapByUidPid(userID string, platformID int, m map[string]int) error {
|
||||
key := uidPidToken + userID + ":" + constant.PlatformIDToName(platformID)
|
||||
_, err := d.Exec("hmset", key, redis.Args{}.Add().AddFlat(m)...)
|
||||
return err
|
||||
}
|
||||
func (d *DataBases) DeleteTokenByUidPid(userID string, platformID int, fields []string) error {
|
||||
key := uidPidToken + userID + ":" + constant.PlatformIDToName(platformID)
|
||||
_, err := d.Exec("HDEL", key, redis.Args{}.Add().AddFlat(fields)...)
|
||||
return err
|
||||
}
|
||||
|
||||
func (d *DataBases) SetSingleConversationRecvMsgOpt(userID, conversationID string, opt int32) error {
|
||||
key := conversationReceiveMessageOpt + userID
|
||||
_, err := d.Exec("HSet", key, conversationID, opt)
|
||||
return err
|
||||
}
|
||||
|
||||
func (d *DataBases) GetSingleConversationRecvMsgOpt(userID, conversationID string) (int, error) {
|
||||
key := conversationReceiveMessageOpt + userID
|
||||
return redis.Int(d.Exec("HGet", key, conversationID))
|
||||
}
|
||||
func (d *DataBases) GetAllConversationMsgOpt(userID string) (map[string]int, error) {
|
||||
key := conversationReceiveMessageOpt + userID
|
||||
return redis.IntMap(d.Exec("HGETALL", key))
|
||||
}
|
||||
func (d *DataBases) SetMultiConversationMsgOpt(userID string, m map[string]int) error {
|
||||
key := conversationReceiveMessageOpt + userID
|
||||
_, err := d.Exec("hmset", key, redis.Args{}.Add().AddFlat(m)...)
|
||||
return err
|
||||
}
|
||||
func (d *DataBases) GetMultiConversationMsgOpt(userID string, conversationIDs []string) (m map[string]int, err error) {
|
||||
m = make(map[string]int)
|
||||
key := conversationReceiveMessageOpt + userID
|
||||
i, err := redis.Ints(d.Exec("hmget", key, redis.Args{}.Add().AddFlat(conversationIDs)...))
|
||||
if err != nil {
|
||||
return m, err
|
||||
}
|
||||
for k, v := range conversationIDs {
|
||||
m[v] = i[k]
|
||||
}
|
||||
return m, nil
|
||||
|
||||
}
|
||||
//func (d *DataBases) JudgeAccountEXISTS(account string) (bool, error) {
|
||||
// key := accountTempCode + account
|
||||
// return redis.Bool(d.Exec("EXISTS", key))
|
||||
//}
|
||||
//func (d *DataBases) SetAccountCode(account string, code, ttl int) (err error) {
|
||||
// key := accountTempCode + account
|
||||
// _, err = d.Exec("SET", key, code, "ex", ttl)
|
||||
// return err
|
||||
//}
|
||||
//func (d *DataBases) GetAccountCode(account string) (string, error) {
|
||||
// key := accountTempCode + account
|
||||
// return redis.String(d.Exec("GET", key))
|
||||
//}
|
||||
//
|
||||
////Perform seq auto-increment operation of user messages
|
||||
//func (d *DataBases) IncrUserSeq(uid string) (uint64, error) {
|
||||
// key := userIncrSeq + uid
|
||||
// return redis.Uint64(d.Exec("INCR", key))
|
||||
//}
|
||||
//
|
||||
////Get the largest Seq
|
||||
//func (d *DataBases) GetUserMaxSeq(uid string) (uint64, error) {
|
||||
// key := userIncrSeq + uid
|
||||
// return redis.Uint64(d.Exec("GET", key))
|
||||
//}
|
||||
//
|
||||
////set the largest Seq
|
||||
//func (d *DataBases) SetUserMaxSeq(uid string, maxSeq uint64) error {
|
||||
// key := userIncrSeq + uid
|
||||
// _, err := d.Exec("SET", key, maxSeq)
|
||||
// return err
|
||||
//}
|
||||
//
|
||||
////Set the user's minimum seq
|
||||
//func (d *DataBases) SetUserMinSeq(uid string, minSeq uint32) (err error) {
|
||||
// key := userMinSeq + uid
|
||||
// _, err = d.Exec("SET", key, minSeq)
|
||||
// return err
|
||||
//}
|
||||
//
|
||||
////Get the smallest Seq
|
||||
//func (d *DataBases) GetUserMinSeq(uid string) (uint64, error) {
|
||||
// key := userMinSeq + uid
|
||||
// return redis.Uint64(d.Exec("GET", key))
|
||||
//}
|
||||
//
|
||||
//
|
||||
////Store userid and platform class to redis
|
||||
//func (d *DataBases) AddTokenFlag(userID string, platformID int, token string, flag int) error {
|
||||
// key := uidPidToken + userID + ":" + constant.PlatformIDToName(platformID)
|
||||
// log2.NewDebug("", "add token key is ", key)
|
||||
// _, err1 := d.Exec("HSet", key, token, flag)
|
||||
// return err1
|
||||
//}
|
||||
//
|
||||
//func (d *DataBases) GetTokenMapByUidPid(userID, platformID string) (map[string]int, error) {
|
||||
// key := uidPidToken + userID + ":" + platformID
|
||||
// log2.NewDebug("", "get token key is ", key)
|
||||
// return redis.IntMap(d.Exec("HGETALL", key))
|
||||
//}
|
||||
//func (d *DataBases) SetTokenMapByUidPid(userID string, platformID int, m map[string]int) error {
|
||||
// key := uidPidToken + userID + ":" + constant.PlatformIDToName(platformID)
|
||||
// _, err := d.Exec("hmset", key, redis.Args{}.Add().AddFlat(m)...)
|
||||
// return err
|
||||
//}
|
||||
//func (d *DataBases) DeleteTokenByUidPid(userID string, platformID int, fields []string) error {
|
||||
// key := uidPidToken + userID + ":" + constant.PlatformIDToName(platformID)
|
||||
// _, err := d.Exec("HDEL", key, redis.Args{}.Add().AddFlat(fields)...)
|
||||
// return err
|
||||
//}
|
||||
//
|
||||
//func (d *DataBases) SetSingleConversationRecvMsgOpt(userID, conversationID string, opt int32) error {
|
||||
// key := conversationReceiveMessageOpt + userID
|
||||
// _, err := d.Exec("HSet", key, conversationID, opt)
|
||||
// return err
|
||||
//}
|
||||
//
|
||||
//func (d *DataBases) GetSingleConversationRecvMsgOpt(userID, conversationID string) (int, error) {
|
||||
// key := conversationReceiveMessageOpt + userID
|
||||
// return redis.Int(d.Exec("HGet", key, conversationID))
|
||||
//}
|
||||
//func (d *DataBases) GetMultiConversationMsgOpt(userID string, conversationIDs []string) (m map[string]int, err error) {
|
||||
// m = make(map[string]int)
|
||||
// key := conversationReceiveMessageOpt + userID
|
||||
// i, err := redis.Ints(d.Exec("hmget", key, redis.Args{}.Add().AddFlat(conversationIDs)...))
|
||||
// if err != nil {
|
||||
// return m, err
|
||||
// }
|
||||
// for k, v := range conversationIDs {
|
||||
// m[v] = i[k]
|
||||
// }
|
||||
// return m, nil
|
||||
//
|
||||
//}
|
||||
|
||||
//func (d *DataBases) SetGetuiToken(token string, expireTime int64) error {
|
||||
// _, err := d.Exec("SET", getuiToken, token, "ex", expireTime)
|
||||
@@ -270,54 +237,54 @@ func (d *DataBases) SearchContentType() {
|
||||
// return result, err
|
||||
//}
|
||||
|
||||
func (d *DataBases) GetMessageListBySeq(userID string, seqList []uint32, operationID string) (seqMsg []*pbCommon.MsgData, failedSeqList []uint32, errResult error) {
|
||||
for _, v := range seqList {
|
||||
//MESSAGE_CACHE:169.254.225.224_reliability1653387820_0_1
|
||||
key := messageCache + userID + "_" + strconv.Itoa(int(v))
|
||||
//func (d *DataBases) GetMessageListBySeq(userID string, seqList []uint32, operationID string) (seqMsg []*pbCommon.MsgData, failedSeqList []uint32, errResult error) {
|
||||
// for _, v := range seqList {
|
||||
// //MESSAGE_CACHE:169.254.225.224_reliability1653387820_0_1
|
||||
// key := messageCache + userID + "_" + strconv.Itoa(int(v))
|
||||
//
|
||||
// result, err := redis.String(d.Exec("GET", key))
|
||||
// if err != nil {
|
||||
// errResult = err
|
||||
// failedSeqList = append(failedSeqList, v)
|
||||
// log2.NewWarn(operationID, "redis get message error:", err.Error(), v)
|
||||
// } else {
|
||||
// msg := pbCommon.MsgData{}
|
||||
// err = jsonpb.UnmarshalString(result, &msg)
|
||||
// if err != nil {
|
||||
// errResult = err
|
||||
// failedSeqList = append(failedSeqList, v)
|
||||
// log2.NewWarn(operationID, "Unmarshal err", result, err.Error())
|
||||
// } else {
|
||||
// log2.NewDebug(operationID, "redis get msg is ", msg.String())
|
||||
// seqMsg = append(seqMsg, &msg)
|
||||
// }
|
||||
//
|
||||
// }
|
||||
// }
|
||||
// return seqMsg, failedSeqList, errResult
|
||||
//}
|
||||
|
||||
result, err := redis.String(d.Exec("GET", key))
|
||||
if err != nil {
|
||||
errResult = err
|
||||
failedSeqList = append(failedSeqList, v)
|
||||
log2.NewWarn(operationID, "redis get message error:", err.Error(), v)
|
||||
} else {
|
||||
msg := pbCommon.MsgData{}
|
||||
err = jsonpb.UnmarshalString(result, &msg)
|
||||
if err != nil {
|
||||
errResult = err
|
||||
failedSeqList = append(failedSeqList, v)
|
||||
log2.NewWarn(operationID, "Unmarshal err", result, err.Error())
|
||||
} else {
|
||||
log2.NewDebug(operationID, "redis get msg is ", msg.String())
|
||||
seqMsg = append(seqMsg, &msg)
|
||||
}
|
||||
|
||||
}
|
||||
}
|
||||
return seqMsg, failedSeqList, errResult
|
||||
}
|
||||
|
||||
func (d *DataBases) SetMessageToCache(msgList []*pbChat.MsgDataToMQ, uid string, operationID string) error {
|
||||
var failedList []pbChat.MsgDataToMQ
|
||||
for _, msg := range msgList {
|
||||
key := messageCache + uid + "_" + strconv.Itoa(int(msg.MsgData.Seq))
|
||||
s, err := utils.Pb2String(msg.MsgData)
|
||||
if err != nil {
|
||||
log2.NewWarn(operationID, utils.GetSelfFuncName(), "Pb2String failed", msg.MsgData.String(), uid, err.Error())
|
||||
continue
|
||||
}
|
||||
log2.NewDebug(operationID, "convert string is ", s)
|
||||
_, err = d.Exec("SET", key, s, "ex", config.Config.MsgCacheTimeout)
|
||||
if err != nil {
|
||||
log2.NewWarn(operationID, utils.GetSelfFuncName(), "redis failed", "args:", key, *msg, uid, s)
|
||||
failedList = append(failedList, *msg)
|
||||
}
|
||||
}
|
||||
if len(failedList) != 0 {
|
||||
return errors.New(fmt.Sprintf("set msg to cache failed, failed lists: %q", failedList))
|
||||
}
|
||||
return nil
|
||||
}
|
||||
//func (d *DataBases) SetMessageToCache(msgList []*pbChat.MsgDataToMQ, uid string, operationID string) error {
|
||||
// var failedList []pbChat.MsgDataToMQ
|
||||
// for _, msg := range msgList {
|
||||
// key := messageCache + uid + "_" + strconv.Itoa(int(msg.MsgData.Seq))
|
||||
// s, err := utils.Pb2String(msg.MsgData)
|
||||
// if err != nil {
|
||||
// log2.NewWarn(operationID, utils.GetSelfFuncName(), "Pb2String failed", msg.MsgData.String(), uid, err.Error())
|
||||
// continue
|
||||
// }
|
||||
// log2.NewDebug(operationID, "convert string is ", s)
|
||||
// _, err = d.Exec("SET", key, s, "ex", config.Config.MsgCacheTimeout)
|
||||
// if err != nil {
|
||||
// log2.NewWarn(operationID, utils.GetSelfFuncName(), "redis failed", "args:", key, *msg, uid, s)
|
||||
// failedList = append(failedList, *msg)
|
||||
// }
|
||||
// }
|
||||
// if len(failedList) != 0 {
|
||||
// return errors.New(fmt.Sprintf("set msg to cache failed, failed lists: %q", failedList))
|
||||
// }
|
||||
// return nil
|
||||
//}
|
||||
|
||||
//func (d *DataBases) DelMsgFromCache(uid string, seqList []uint32, operationID string) {
|
||||
// for _, seq := range seqList {
|
||||
|
||||
Reference in New Issue
Block a user