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 }