增加健康检查定时任务/健康检查可以发送消息

internal/tasks/message_task.go

@@ -0,0 +1,41 @@
+package tasks
+
+import (
+	"github.com/TeaOSLab/EdgeAPI/internal/db/models"
+	"github.com/TeaOSLab/EdgeAPI/internal/utils"
+	"github.com/iwind/TeaGo/dbs"
+	"github.com/iwind/TeaGo/logs"
+	"time"
+)
+
+func init() {
+	dbs.OnReady(func() {
+		go NewMessageTask().Run()
+	})
+}
+
+// 消息相关任务
+type MessageTask struct {
+}
+
+// 获取新对象
+func NewMessageTask() *MessageTask {
+	return &MessageTask{}
+}
+
+// 运行
+func (this *MessageTask) Run() {
+	ticker := utils.NewTicker(24 * time.Hour)
+	for ticker.Wait() {
+		err := this.loop()
+		if err != nil {
+			logs.Println("[TASK][MESSAGE]" + err.Error())
+		}
+	}
+}
+
+// 单次运行
+func (this *MessageTask) loop() error {
+	dayTime := time.Now().AddDate(0, 0, -30) // TODO 这个30天应该可以在界面上设置
+	return models.NewMessageDAO().DeleteMessagesBeforeDay(dayTime)
+}