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

batch to mongo

Browse Source
This commit is contained in:
skiffer-git 2022-05-20 11:20:21 +08:00 committed by Xinwei Xiong(cubxxw-openim)
parent ae7e681bca
commit 7b2de29e30
3 changed files with 11 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
internal/msg_transfer/logic/db.go
View File

@ -22,7 +22,7 @@ func saveUserChat(uid string, msg *pbMsg.MsgDataToMQ) error {
// return db.DB.SaveUserChatMongo2(uid, pbSaveData.MsgData.SendTime, &pbSaveData) // return db.DB.SaveUserChatMongo2(uid, pbSaveData.MsgData.SendTime, &pbSaveData)
} }
func saveUserChatList(uid string, msgList []*pbMsg.MsgDataToMQ, operationID string) error { func saveUserChatList(userID string, msgList []*pbMsg.MsgDataToMQ, operationID string) error {
log.Info(operationID, utils.GetSelfFuncName(), "args ", uid, len(msgList)) log.Info(operationID, utils.GetSelfFuncName(), "args ", userID, len(msgList))
return db.DB.BatchInsertChat(uid, msgList, operationID) return db.DB.BatchInsertChat(userID, msgList, operationID)
} }

2
pkg/common/db/batch_insert_chat.go
View File

@ -74,5 +74,5 @@ func (d *DataBases) BatchInsertChat(userID string, msgList []*pbMsg.MsgDataToMQ,
} }
} }
log.NewDebug(operationID, "find mgo uid cost time", getCurrentTimestampByMill()-newTime) log.NewDebug(operationID, "find mgo uid cost time", getCurrentTimestampByMill()-newTime)
return nil return utils.Wrap(d.SetUserMaxSeq(userID, uint32(currentMaxSeq)), "")
} }

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

@ -62,6 +62,13 @@ func (d *DataBases) GetUserMaxSeq(uid string) (uint64, error) {
return redis.Uint64(d.Exec("GET", key)) return redis.Uint64(d.Exec("GET", key))
} }
//set the largest Seq
func (d *DataBases) SetUserMaxSeq(uid string, maxSeq uint32) error {
key := userIncrSeq + uid
_, err := redis.Uint64(d.Exec("SET", key, maxSeq))
return err
}
//Set the user's minimum seq //Set the user's minimum seq
func (d *DataBases) SetUserMinSeq(uid string, minSeq uint32) (err error) { func (d *DataBases) SetUserMinSeq(uid string, minSeq uint32) (err error) {
key := userMinSeq + uid key := userMinSeq + uid