|
|
|
@ -48,12 +48,29 @@ func DeleteMongoMsgAndResetRedisSeq(operationID, userID string) error {
|
|
|
|
|
return err
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func delMongoMsgs(operationID string, delMsgIDList *[][2]interface{}) error {
|
|
|
|
|
if len(*delMsgIDList) > 0 {
|
|
|
|
|
var IDList []string
|
|
|
|
|
for _, v := range *delMsgIDList {
|
|
|
|
|
IDList = append(IDList, v[0].(string))
|
|
|
|
|
}
|
|
|
|
|
err := db.DB.DelMongoMsgs(IDList)
|
|
|
|
|
if err != nil {
|
|
|
|
|
return utils.Wrap(err, "DelMongoMsgs failed")
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
return nil
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// recursion
|
|
|
|
|
func deleteMongoMsg(operationID string, ID string, index int64, delMsgIDList *[][2]interface{}) (uint32, error) {
|
|
|
|
|
// 从最旧的列表开始找
|
|
|
|
|
msgs, err := db.DB.GetUserMsgListByIndex(ID, index)
|
|
|
|
|
if err != nil {
|
|
|
|
|
return 0, utils.Wrap(err, "GetUserMsgListByIndex failed")
|
|
|
|
|
if err != nil || msgs.UID == "" {
|
|
|
|
|
if err != nil {
|
|
|
|
|
log.NewError(operationID, utils.GetSelfFuncName(), "GetUserMsgListByIndex failed", err.Error(), index, ID)
|
|
|
|
|
}
|
|
|
|
|
return getDelMaxSeqByIDList(*delMsgIDList), delMongoMsgs(operationID, delMsgIDList)
|
|
|
|
|
}
|
|
|
|
|
if len(msgs.Msg) > db.GetSingleGocMsgNum() {
|
|
|
|
|
log.NewWarn(operationID, utils.GetSelfFuncName(), "msgs too large", len(msgs.Msg), msgs.UID)
|
|
|
|
@ -62,15 +79,8 @@ func deleteMongoMsg(operationID string, ID string, index int64, delMsgIDList *[]
|
|
|
|
|
for i, msg := range msgs.Msg {
|
|
|
|
|
// 找到列表中不需要删除的消息了
|
|
|
|
|
if utils.GetCurrentTimestampByMill() < msg.SendTime+int64(config.Config.Mongo.DBRetainChatRecords)*24*60*60*1000 {
|
|
|
|
|
if len(*delMsgIDList) > 0 {
|
|
|
|
|
var IDList []string
|
|
|
|
|
for _, v := range *delMsgIDList {
|
|
|
|
|
IDList = append(IDList, v[0].(string))
|
|
|
|
|
}
|
|
|
|
|
err := db.DB.DelMongoMsgs(IDList)
|
|
|
|
|
if err != nil {
|
|
|
|
|
return 0, utils.Wrap(err, "DelMongoMsgs failed")
|
|
|
|
|
}
|
|
|
|
|
if err := delMongoMsgs(operationID, delMsgIDList); err != nil {
|
|
|
|
|
return 0, err
|
|
|
|
|
}
|
|
|
|
|
minSeq := getDelMaxSeqByIDList(*delMsgIDList)
|
|
|
|
|
if i > 0 {
|
|
|
|
@ -84,19 +94,21 @@ func deleteMongoMsg(operationID string, ID string, index int64, delMsgIDList *[]
|
|
|
|
|
log.NewError(operationID, utils.GetSelfFuncName(), err.Error(), msgs.UID, i)
|
|
|
|
|
return minSeq, nil
|
|
|
|
|
}
|
|
|
|
|
minSeq = msgPb.Seq - 1
|
|
|
|
|
minSeq = msgPb.Seq
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
return minSeq, nil
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
msgPb := &server_api_params.MsgData{}
|
|
|
|
|
err = proto.Unmarshal(msgs.Msg[len(msgs.Msg)-1].Msg, msgPb)
|
|
|
|
|
if err != nil {
|
|
|
|
|
log.NewError(operationID, utils.GetSelfFuncName(), err.Error(), len(msgs.Msg)-1, msgs.UID)
|
|
|
|
|
return 0, utils.Wrap(err, "proto.Unmarshal failed")
|
|
|
|
|
if len(msgs.Msg) > 0 {
|
|
|
|
|
msgPb := &server_api_params.MsgData{}
|
|
|
|
|
err = proto.Unmarshal(msgs.Msg[len(msgs.Msg)-1].Msg, msgPb)
|
|
|
|
|
if err != nil {
|
|
|
|
|
log.NewError(operationID, utils.GetSelfFuncName(), err.Error(), len(msgs.Msg)-1, msgs.UID)
|
|
|
|
|
return 0, utils.Wrap(err, "proto.Unmarshal failed")
|
|
|
|
|
}
|
|
|
|
|
*delMsgIDList = append(*delMsgIDList, [2]interface{}{msgs.UID, msgPb.Seq})
|
|
|
|
|
}
|
|
|
|
|
*delMsgIDList = append(*delMsgIDList, [2]interface{}{msgs.UID, msgPb.Seq})
|
|
|
|
|
// 没有找到 代表需要全部删除掉 继续递归查找下一个比较旧的列表
|
|
|
|
|
seq, err := deleteMongoMsg(operationID, utils.GetSelfFuncName(), index+1, delMsgIDList)
|
|
|
|
|
if err != nil {
|
|
|
|
|