From 6c8b7de7861f07a6ccd8f210d4b2b4c733281d0a Mon Sep 17 00:00:00 2001
From: Gordon <46924906+FGadvancer@users.noreply.github.com>
Date: Tue, 19 Mar 2024 16:44:17 +0800
Subject: [PATCH] refactor: unified naming for module startup functions.

---
 internal/tools/cron_task.go | 2 +-
 internal/tools/msg.go       | 6 +++---
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/internal/tools/cron_task.go b/internal/tools/cron_task.go
index 82a4c0344..cd92b6725 100644
--- a/internal/tools/cron_task.go
+++ b/internal/tools/cron_task.go
@@ -33,7 +33,7 @@ func StartTask(ctx context.Context, config *config.GlobalConfig) error {
 
 	log.CInfo(ctx, "cron task server starting", "chatRecordsClearTime", config.ChatRecordsClearTime, "msgDestructTime", config.MsgDestructTime)
 
-	msgTool, err := InitMsgTool(config)
+	msgTool, err := InitMsgTool(ctx, config)
 	if err != nil {
 		return err
 	}
diff --git a/internal/tools/msg.go b/internal/tools/msg.go
index b5adffbba..e9fb7e4f0 100644
--- a/internal/tools/msg.go
+++ b/internal/tools/msg.go
@@ -63,12 +63,12 @@ func NewMsgTool(msgDatabase controller.CommonMsgDatabase, userDatabase controlle
 	}
 }
 
-func InitMsgTool(config *config.GlobalConfig) (*MsgTool, error) {
-	rdb, err := cache.NewRedis(&config.Redis)
+func InitMsgTool(ctx context.Context, config *config.GlobalConfig) (*MsgTool, error) {
+	rdb, err := cache.NewRedis(ctx, &config.Redis)
 	if err != nil {
 		return nil, err
 	}
-	mongo, err := unrelation.NewMongoDB(&config.Mongo)
+	mongo, err := unrelation.NewMongoDB(ctx, &config.Mongo)
 	if err != nil {
 		return nil, err
 	}