blackteay/open-im-server
1
0
Fork 0
mirror of https://github.com/openimsdk/open-im-server.git synced 2025-12-09 06:02:54 +08:00
Code Issues Packages Projects Releases Wiki Activity

optimize scheduled deletion

Browse Source
This commit is contained in:
withchao 2023-10-30 17:13:26 +08:00
parent 7cdf02a33e
commit cfaf004cd8
2 changed files with 10 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
internal/tools/conversation.go
View File

@ -74,8 +74,7 @@ func (c *MsgTool) ConversationsDestructMsgs() {
return return
} }
const batchNum = 50 const batchNum = 50
seed := time.Now().UnixMilli() log.ZDebug(ctx, "GetAllConversationIDsNumber", "num", num)
log.ZDebug(ctx, "GetAllConversationIDsNumber", "num", num, "seed", seed)
if num == 0 { if num == 0 {
return return
} }
@ -87,9 +86,8 @@ func (c *MsgTool) ConversationsDestructMsgs() {
if num%batchNum != 0 { if num%batchNum != 0 {
maxPage++ maxPage++
} }
r := rand.New(rand.NewSource(seed))
for i := 0; i < count; i++ { for i := 0; i < count; i++ {
pageNumber := r.Int63() % maxPage pageNumber := rand.Int63() % maxPage
conversationIDs, err := c.conversationDatabase.PageConversationIDs(ctx, int32(pageNumber), batchNum) conversationIDs, err := c.conversationDatabase.PageConversationIDs(ctx, int32(pageNumber), batchNum)
if err != nil { if err != nil {
log.ZError(ctx, "PageConversationIDs failed", err, "pageNumber", pageNumber) log.ZError(ctx, "PageConversationIDs failed", err, "pageNumber", pageNumber)

20
internal/tools/msg.go
View File

@ -17,20 +17,18 @@ package tools
import ( import (
"context" "context"
"fmt" "fmt"
"github.com/openimsdk/open-im-server/v3/pkg/common/discovery_register"
"github.com/redis/go-redis/v9"
"google.golang.org/grpc"
"google.golang.org/grpc/credentials/insecure"
"math"
"math/rand"
"time"
"github.com/OpenIMSDK/tools/errs" "github.com/OpenIMSDK/tools/errs"
"github.com/OpenIMSDK/tools/log" "github.com/OpenIMSDK/tools/log"
"github.com/OpenIMSDK/tools/mcontext" "github.com/OpenIMSDK/tools/mcontext"
"github.com/OpenIMSDK/tools/mw" "github.com/OpenIMSDK/tools/mw"
"github.com/OpenIMSDK/tools/tx" "github.com/OpenIMSDK/tools/tx"
"github.com/OpenIMSDK/tools/utils" "github.com/OpenIMSDK/tools/utils"
"github.com/openimsdk/open-im-server/v3/pkg/common/discovery_register"
"github.com/redis/go-redis/v9"
"google.golang.org/grpc"
"google.golang.org/grpc/credentials/insecure"
"math"
"math/rand"
"github.com/openimsdk/open-im-server/v3/pkg/common/config" "github.com/openimsdk/open-im-server/v3/pkg/common/config"
"github.com/openimsdk/open-im-server/v3/pkg/common/db/cache" "github.com/openimsdk/open-im-server/v3/pkg/common/db/cache"
@ -128,8 +126,7 @@ func (c *MsgTool) AllConversationClearMsgAndFixSeq() {
return return
} }
const batchNum = 50 const batchNum = 50
seed := time.Now().UnixMilli() log.ZDebug(ctx, "GetAllConversationIDsNumber", "num", num)
log.ZDebug(ctx, "GetAllConversationIDsNumber", "num", num, "seed", seed)
if num == 0 { if num == 0 {
return return
} }
@ -141,9 +138,8 @@ func (c *MsgTool) AllConversationClearMsgAndFixSeq() {
if num%batchNum != 0 { if num%batchNum != 0 {
maxPage++ maxPage++
} }
r := rand.New(rand.NewSource(seed))
for i := 0; i < count; i++ { for i := 0; i < count; i++ {
pageNumber := r.Int63() % maxPage pageNumber := rand.Int63() % maxPage
conversationIDs, err := c.conversationDatabase.PageConversationIDs(ctx, int32(pageNumber), batchNum) conversationIDs, err := c.conversationDatabase.PageConversationIDs(ctx, int32(pageNumber), batchNum)
if err != nil { if err != nil {
log.ZError(ctx, "PageConversationIDs failed", err, "pageNumber", pageNumber) log.ZError(ctx, "PageConversationIDs failed", err, "pageNumber", pageNumber)