sg notification

internal/tools/cron_task.go

@@ -2,10 +2,11 @@ package tools
 
 import (
 	"fmt"
+	"sync"
+
 	"github.com/OpenIMSDK/Open-IM-Server/pkg/common/config"
 	"github.com/OpenIMSDK/Open-IM-Server/pkg/common/log"
 	"github.com/robfig/cron/v3"
-	"sync"
 )
 
 const cronTaskOperationID = "cronTaskOperationID-"
@@ -22,7 +23,7 @@ func StartCronTask() error {
 	c := cron.New()
 	var wg sync.WaitGroup
 	wg.Add(1)
-	_, err = c.AddFunc(config.Config.Mongo.ChatRecordsClearTime, msgTool.AllUserClearMsgAndFixSeq)
+	_, err = c.AddFunc(config.Config.Mongo.ChatRecordsClearTime, msgTool.AllConversationClearMsgAndFixSeq)
 	if err != nil {
 		fmt.Println("start cron failed", err.Error(), config.Config.Mongo.ChatRecordsClearTime)
 		return err

internal/tools/msg.go

@@ -20,9 +20,10 @@ import (
 )
 
 type MsgTool struct {
-	msgDatabase   controller.CommonMsgDatabase
-	userDatabase  controller.UserDatabase
-	groupDatabase controller.GroupDatabase
+	msgDatabase          controller.CommonMsgDatabase
+	conversationDatabase controller.ConversationDataBase
+	userDatabase         controller.UserDatabase
+	groupDatabase        controller.GroupDatabase
 }
 
 var errSeq = errors.New("cache max seq and mongo max seq is diff > 10")
@@ -56,23 +57,15 @@ func InitMsgTool() (*MsgTool, error) {
 	return msgTool, nil
 }
 
-func (c *MsgTool) AllUserClearMsgAndFixSeq() {
+func (c *MsgTool) AllConversationClearMsgAndFixSeq() {
 	ctx := mcontext.NewCtx(utils.GetSelfFuncName())
 	log.ZInfo(ctx, "============================ start del cron task ============================")
-	var err error
-	userIDs, err := c.userDatabase.GetAllUserID(ctx)
-	if err == nil {
-		c.ClearUsersMsg(ctx, userIDs)
-	} else {
-		log.ZError(ctx, "ClearUsersMsg failed", err)
-	}
-	// working group msg clear
-	superGroupIDs, err := c.groupDatabase.GetGroupIDsByGroupType(ctx, constant.WorkingGroup)
-	if err == nil {
-		c.ClearSuperGroupMsg(ctx, superGroupIDs)
-	} else {
-		log.ZError(ctx, "ClearSuperGroupMsg failed", err)
+	conversationIDs, err := c.conversationDatabase.GetAllConversationIDs(ctx)
+	if err != nil {
+		log.ZError(ctx, "GetAllConversationIDs failed", err)
+		return
 	}
+	c.ClearSuperGroupMsg(ctx, conversationIDs)
 	log.ZInfo(ctx, "============================ start del cron finished ============================")
 }