|
|
|
@ -2,14 +2,68 @@ package cronTask
|
|
|
|
|
|
|
|
|
|
import (
|
|
|
|
|
"Open_IM/pkg/common/constant"
|
|
|
|
|
"Open_IM/pkg/common/db"
|
|
|
|
|
"Open_IM/pkg/common/log"
|
|
|
|
|
pbMsg "Open_IM/pkg/proto/msg"
|
|
|
|
|
server_api_params "Open_IM/pkg/proto/sdk_ws"
|
|
|
|
|
"Open_IM/pkg/utils"
|
|
|
|
|
"testing"
|
|
|
|
|
"time"
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
func getMsgListFake(num int) []*pbMsg.MsgDataToMQ {
|
|
|
|
|
var msgList []*pbMsg.MsgDataToMQ
|
|
|
|
|
for i := 1; i < num; i++ {
|
|
|
|
|
msgList = append(msgList, &pbMsg.MsgDataToMQ{
|
|
|
|
|
Token: "tk",
|
|
|
|
|
OperationID: "operationID",
|
|
|
|
|
MsgData: &server_api_params.MsgData{
|
|
|
|
|
SendID: "sendID1",
|
|
|
|
|
RecvID: "recvID1",
|
|
|
|
|
GroupID: "",
|
|
|
|
|
ClientMsgID: "xxx",
|
|
|
|
|
ServerMsgID: "xxx",
|
|
|
|
|
SenderPlatformID: 1,
|
|
|
|
|
SenderNickname: "testNickName",
|
|
|
|
|
SenderFaceURL: "testFaceURL",
|
|
|
|
|
SessionType: 1,
|
|
|
|
|
MsgFrom: 100,
|
|
|
|
|
ContentType: 101,
|
|
|
|
|
Content: []byte("testFaceURL"),
|
|
|
|
|
Seq: uint32(i),
|
|
|
|
|
SendTime: time.Now().Unix(),
|
|
|
|
|
CreateTime: time.Now().Unix(),
|
|
|
|
|
Status: 1,
|
|
|
|
|
},
|
|
|
|
|
})
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func TestDeleteMongoMsgAndResetRedisSeq(t *testing.T) {
|
|
|
|
|
operationID := getCronTaskOperationID()
|
|
|
|
|
testUserIDList := []string{"test_del_id1", "test_del_id2", "test_del_id3", "test_del_id4", "test_del_id5"}
|
|
|
|
|
testUID1 := "test_del_id1"
|
|
|
|
|
//testUID2 := "test_del_id2"
|
|
|
|
|
//testUID3 := "test_del_id3"
|
|
|
|
|
//testUID4 := "test_del_id4"
|
|
|
|
|
//testUID5 := "test_del_id5"
|
|
|
|
|
//testUID6 := "test_del_id6"
|
|
|
|
|
testUserIDList := []string{testUID1}
|
|
|
|
|
|
|
|
|
|
db.DB.SetUserMaxSeq(testUID1, 500)
|
|
|
|
|
db.DB.BatchInsertChat2DB(testUID1, getMsgListFake(500), testUID1+"-"+operationID, 500)
|
|
|
|
|
|
|
|
|
|
//db.DB.SetUserMaxSeq(testUID1, 6000)
|
|
|
|
|
//db.DB.BatchInsertChat2DB()
|
|
|
|
|
//
|
|
|
|
|
//db.DB.SetUserMaxSeq(testUID1, 4999)
|
|
|
|
|
//db.DB.BatchInsertChat2DB()
|
|
|
|
|
//
|
|
|
|
|
//db.DB.SetUserMaxSeq(testUID1, 30000)
|
|
|
|
|
//db.DB.BatchInsertChat2DB()
|
|
|
|
|
//
|
|
|
|
|
//db.DB.SetUserMaxSeq(testUID1, 9999)
|
|
|
|
|
//db.DB.BatchInsertChat2DB()
|
|
|
|
|
|
|
|
|
|
for _, userID := range testUserIDList {
|
|
|
|
|
operationID = userID + "-" + operationID
|
|
|
|
|
if err := DeleteMongoMsgAndResetRedisSeq(operationID, userID); err != nil {
|
|
|
|
|