🔊 v0.3.0 计划任务（定期备份等场景）

cmd/dashboard/main.go

@@ -1,6 +1,7 @@
 package main
 
 import (
+	"fmt"
 	"time"
 
 	"github.com/patrickmn/go-cache"
@@ -10,13 +11,13 @@ import (
 	"github.com/naiba/nezha/cmd/dashboard/controller"
 	"github.com/naiba/nezha/cmd/dashboard/rpc"
 	"github.com/naiba/nezha/model"
+	pb "github.com/naiba/nezha/proto"
 	"github.com/naiba/nezha/service/alertmanager"
 	"github.com/naiba/nezha/service/dao"
 )
 
 func init() {
 	var err error
-	dao.ServerList = make(map[uint64]*model.Server)
 	dao.Conf = &model.Config{}
 	err = dao.Conf.Read("data/config.yaml")
 	if err != nil {
@@ -30,14 +31,26 @@ func init() {
 		dao.DB = dao.DB.Debug()
 	}
 	dao.Cache = cache.New(5*time.Minute, 10*time.Minute)
-	initDB()
+	initSystem()
 }
 
-func initDB() {
+func initSystem() {
 	dao.DB.AutoMigrate(model.Server{}, model.User{},
 		model.Notification{}, model.AlertRule{}, model.Monitor{},
-		model.MonitorHistory{})
-	// load cache
+		model.MonitorHistory{}, model.Cron{})
+
+	loadServers() //加载服务器列表
+	loadCrons()   //加载计划任务
+
+	// 清理旧数据
+	dao.Cron.AddFunc("* 3 * * *", cleanMonitorHistory)
+}
+
+func cleanMonitorHistory() {
+	dao.DB.Delete(&model.MonitorHistory{}, "created_at < ?", time.Now().AddDate(0, -1, 0))
+}
+
+func loadServers() {
 	var servers []model.Server
 	dao.DB.Find(&servers)
 	for _, s := range servers {
@@ -49,6 +62,35 @@ func initDB() {
 	dao.ReSortServer()
 }
 
+func loadCrons() {
+	var crons []model.Cron
+	dao.DB.Find(&crons)
+	var err error
+	for i := 0; i < len(crons); i++ {
+		cr := crons[i]
+		cr.CronID, err = dao.Cron.AddFunc(cr.Scheduler, func() {
+			dao.ServerLock.RLock()
+			defer dao.ServerLock.RUnlock()
+			for j := 0; j < len(cr.Servers); j++ {
+				if dao.ServerList[cr.Servers[j]].TaskStream != nil {
+					dao.ServerList[cr.Servers[j]].TaskStream.Send(&pb.Task{
+						Id:   cr.ID,
+						Data: cr.Command,
+						Type: model.TaskTypeCommand,
+					})
+				} else {
+					alertmanager.SendNotification(fmt.Sprintf("计划任务：%s，服务器：%d 离线，无法执行。", cr.Name, cr.Servers[j]))
+				}
+			}
+		})
+		if err != nil {
+			panic(err)
+		}
+		dao.Crons[cr.ID] = &cr
+	}
+	dao.Cron.Start()
+}
+
 func main() {
 	go controller.ServeWeb(dao.Conf.HTTPPort)
 	go rpc.ServeRPC(5555)