feat: complete settings

service/cron_service/cron_service.go

@@ -0,0 +1,78 @@
+package cron_service
+
+import (
+	"fmt"
+	"github.com/robfig/cron/v3"
+	"github.com/unknwon/com"
+	"message-nest/models"
+	"message-nest/pkg/constant"
+	"message-nest/pkg/logging"
+	"message-nest/service/send_message_service"
+)
+
+type CronService struct {
+}
+
+var ClearLogsTaskId cron.EntryID
+
+// 清除日志的定时任务
+func ClearLogs() {
+	var logOutput []string
+	var errStr string
+	status := 1
+
+	sm := send_message_service.SendMessageService{TaskID: constant.CleanLogsTaskId}
+	logging.Logger.Error("开始清除日志")
+	logOutput = append(logOutput, "开始清除日志")
+
+	setting, err := models.GetSettingByKey(constant.LogsCleanSectionName, constant.LogsCleanKeepKeyName)
+	if err != nil {
+		errStr = fmt.Sprintf("获取日志的保留数失败，原因：%s", err)
+		logging.Logger.Error(errStr)
+		sm.MarkStatus(errStr, &status)
+		logOutput = append(logOutput, errStr)
+	}
+
+	keepNum := com.StrTo(setting.Value).MustInt()
+	err = models.DeleteOutDateLogs(keepNum)
+	if err != nil {
+		errStr = fmt.Sprintf("删除日志失败，原因：%s", err)
+		logging.Logger.Error(errStr)
+		sm.MarkStatus(errStr, &status)
+		logOutput = append(logOutput, errStr)
+	} else {
+		errStr = fmt.Sprintf("删除日志成功，保留数目：%d", keepNum)
+		logging.Logger.Error(errStr)
+		logOutput = append(logOutput, errStr)
+	}
+	sm.RecordSendLog(logOutput, status)
+}
+
+// 启动注册清除任务定时任务
+func (cs *CronService) InitLogsCronRun() {
+	setting, err := models.GetSettingByKey(constant.LogsCleanSectionName, constant.LogsCleanCronKeyName)
+	if err != nil {
+		logging.Logger.Error(fmt.Sprintf("获取日志的cron失败，原因：%s", err))
+	}
+	ClearLogsTaskId = AddTask(ScheduledTask{
+		Schedule: setting.Value,
+		Job:      ClearLogs,
+	})
+}
+
+// 更新清除任务定时任务
+func (cs *CronService) UpdateLogsCronRun(cron string) {
+	RemoveTask(ClearLogsTaskId)
+	ClearLogsTaskId = AddTask(ScheduledTask{
+		Schedule: cron,
+		Job:      ClearLogs,
+	})
+	logging.Logger.Error(fmt.Sprintf("更新日志的cron成功，%s", cron))
+	logging.Logger.Info(fmt.Sprintf("所有的定时任务： %s", TaskList))
+
+}
+
+func Setup() {
+	cs := CronService{}
+	cs.InitLogsCronRun()
+}

service/cron_service/tasks.go

@@ -0,0 +1,52 @@
+package cron_service
+
+import (
+	"fmt"
+	"github.com/robfig/cron/v3"
+	"message-nest/pkg/logging"
+	"sync"
+)
+
+type ScheduledTask struct {
+	//ID       int
+	Schedule string
+	Job      func()
+}
+
+var (
+	CronInstance *cron.Cron
+	TaskList     map[cron.EntryID]*ScheduledTask
+	mutex        sync.Mutex
+)
+
+func init() {
+	CronInstance = cron.New()
+	CronInstance.Start()
+	TaskList = make(map[cron.EntryID]*ScheduledTask)
+}
+
+// 添加定时任务
+func AddTask(task ScheduledTask) cron.EntryID {
+	mutex.Lock()
+	defer mutex.Unlock()
+
+	taskId, err := CronInstance.AddFunc(task.Schedule, task.Job)
+	if err != nil {
+		// 处理错误
+		logging.Logger.Error(fmt.Sprintf("添加定时任务失败，原因：%s", err))
+	} else {
+		TaskList[taskId] = &task
+		logging.Logger.Error(fmt.Sprintf("添加日志清除任务成功，entryID： %d", taskId))
+	}
+	return taskId
+}
+
+// 删除定时任务
+func RemoveTask(taskID cron.EntryID) {
+	mutex.Lock()
+	defer mutex.Unlock()
+	if _, ok := TaskList[taskID]; ok {
+		CronInstance.Remove(taskID)
+		delete(TaskList, taskID)
+	}
+}