mirror of
https://github.com/openimsdk/open-im-server.git
synced 2026-05-04 00:55:59 +08:00
batch to mongo
This commit is contained in:
@@ -2,20 +2,27 @@ package logic
|
||||
|
||||
import (
|
||||
"Open_IM/pkg/common/db"
|
||||
"Open_IM/pkg/common/log"
|
||||
pbMsg "Open_IM/pkg/proto/chat"
|
||||
"Open_IM/pkg/utils"
|
||||
)
|
||||
|
||||
func saveUserChat(uid string, msg *pbMsg.MsgDataToMQ) error {
|
||||
//time := utils.GetCurrentTimestampByMill()
|
||||
//seq, err := db.DB.IncrUserSeq(uid)
|
||||
//if err != nil {
|
||||
// log.NewError(msg.OperationID, "data insert to redis err", err.Error(), msg.String())
|
||||
// return err
|
||||
//}
|
||||
//msg.MsgData.Seq = uint32(seq)
|
||||
time := utils.GetCurrentTimestampByMill()
|
||||
seq, err := db.DB.IncrUserSeq(uid)
|
||||
if err != nil {
|
||||
log.NewError(msg.OperationID, "data insert to redis err", err.Error(), msg.String())
|
||||
return err
|
||||
}
|
||||
msg.MsgData.Seq = uint32(seq)
|
||||
pbSaveData := pbMsg.MsgDataToDB{}
|
||||
pbSaveData.MsgData = msg.MsgData
|
||||
//log.NewInfo(msg.OperationID, "IncrUserSeq cost time", utils.GetCurrentTimestampByMill()-time)
|
||||
log.NewInfo(msg.OperationID, "IncrUserSeq cost time", utils.GetCurrentTimestampByMill()-time)
|
||||
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 {
|
||||
log.Info(operationID, utils.GetSelfFuncName(), "args ", uid, len(msgList))
|
||||
return db.DB.BatchInsertChat(uid, msgList, operationID)
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user