batch to mongo

This commit is contained in:
skiffer-git
2022-05-20 11:00:11 +08:00
parent a2888b009d
commit d4f81f086c
3 changed files with 139 additions and 9 deletions
+15 -8
View File
@@ -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)
}