87 lines
2.5 KiB
Go
87 lines
2.5 KiB
Go
package cron_service
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/robfig/cron/v3"
|
|
"github.com/sirupsen/logrus"
|
|
"github.com/unknwon/com"
|
|
"message-nest/models"
|
|
"message-nest/pkg/constant"
|
|
"message-nest/service/send_message_service"
|
|
)
|
|
|
|
var ClearLogsTaskId cron.EntryID
|
|
|
|
// ClearLogs 清除日志的定时任务
|
|
func ClearLogs() {
|
|
var errStr string
|
|
sm := send_message_service.SendMessageService{
|
|
TaskID: constant.CleanLogsTaskId,
|
|
DefaultLogger: logrus.WithFields(logrus.Fields{
|
|
"prefix": "[Cron Clear Logs]",
|
|
}),
|
|
}
|
|
sm.Status = send_message_service.SendSuccess
|
|
|
|
sm.LogsAndStatusMark("开始清除日志", sm.Status)
|
|
|
|
setting, err := models.GetSettingByKey(constant.LogsCleanSectionName, constant.LogsCleanKeepKeyName)
|
|
if err != nil {
|
|
errStr = fmt.Sprintf("获取日志的保留数失败，原因：%s", err)
|
|
sm.LogsAndStatusMark(errStr, send_message_service.SendFail)
|
|
}
|
|
|
|
keepNum := com.StrTo(setting.Value).MustInt()
|
|
affectedRows, err := models.DeleteOutDateLogs(keepNum)
|
|
if err != nil {
|
|
errStr = fmt.Sprintf("删除日志失败，原因：%s", err)
|
|
sm.LogsAndStatusMark(errStr, send_message_service.SendFail)
|
|
} else {
|
|
errStr = fmt.Sprintf("删除日志成功，删除条数：%d，保留数目：%d", affectedRows, keepNum)
|
|
sm.LogsAndStatusMark(errStr, sm.Status)
|
|
}
|
|
|
|
sm.RecordSendLog()
|
|
}
|
|
|
|
type CronService struct {
|
|
}
|
|
|
|
// StartLogsCronRun 启动注册清除任务定时任务
|
|
func (cs *CronService) StartLogsCronRun() {
|
|
// 注册任务
|
|
setting, err := models.GetSettingByKey(constant.LogsCleanSectionName, constant.LogsCleanCronKeyName)
|
|
if err != nil {
|
|
logrus.Error(fmt.Sprintf("获取日志的cron失败，原因：%s", err))
|
|
}
|
|
ClearLogsTaskId = AddTask(ScheduledTask{
|
|
Schedule: setting.Value,
|
|
Job: ClearLogs,
|
|
})
|
|
|
|
// 添加任务
|
|
err = models.AddSendTaskWithID("日志定时清除", constant.CleanLogsTaskId, "admin")
|
|
if err != nil {
|
|
logrus.Error(fmt.Sprintf("添加日志定时清除任务失败，原因：%s", err))
|
|
}
|
|
}
|
|
|
|
// UpdateLogsCronRun 更新清除任务定时任务
|
|
func (cs *CronService) UpdateLogsCronRun(cron string) {
|
|
RemoveTask(ClearLogsTaskId)
|
|
ClearLogsTaskId = AddTask(ScheduledTask{
|
|
Schedule: cron,
|
|
Job: ClearLogs,
|
|
})
|
|
logrus.Error(fmt.Sprintf("更新日志的cron成功，%s", cron))
|
|
logrus.Error(fmt.Sprintf("所有的定时任务： %s", TaskList))
|
|
|
|
}
|
|
|
|
// StartLogsCronRun 启动的时候开启定时任务
|
|
func StartLogsCronRun() {
|
|
logrus.Infof("开始注册定时清除日志任务...")
|
|
cs := CronService{}
|
|
cs.StartLogsCronRun()
|
|
}
|