1.4.5.2

EdgeAPI/internal/tasks/node_schedule_task_plus.go

@@ -0,0 +1,77 @@
+// Copyright 2023 Liuxiangchao iwind.liu@gmail.com. All rights reserved. Official site: https://goedge.cn .
+//go:build plus
+
+package tasks
+
+import (
+	"github.com/TeaOSLab/EdgeAPI/internal/db/models"
+	"github.com/TeaOSLab/EdgeAPI/internal/goman"
+	"github.com/iwind/TeaGo/dbs"
+	"time"
+)
+
+func init() {
+	dbs.OnReadyDone(func() {
+		goman.New(func() {
+			NewNodeScheduleTask(1 * time.Minute).Start()
+		})
+	})
+}
+
+// NodeScheduleTask 节点调度任务
+type NodeScheduleTask struct {
+	BaseTask
+
+	ticker *time.Ticker
+}
+
+func NewNodeScheduleTask(duration time.Duration) *NodeScheduleTask {
+	return &NodeScheduleTask{
+		ticker: time.NewTicker(duration),
+	}
+}
+
+func (this *NodeScheduleTask) Start() {
+	for range this.ticker.C {
+		err := this.Loop()
+		if err != nil {
+			this.logErr("NodeScheduleTask", err.Error())
+		}
+	}
+}
+
+func (this *NodeScheduleTask) Loop() error {
+	if !this.IsPrimaryNode() {
+		return nil
+	}
+
+	var tx *dbs.Tx
+
+	// 恢复动作过期的节点
+	err := this.recoverNodes(tx)
+	if err != nil {
+		return err
+	}
+
+	// 下线租期到期的节点
+	err = models.SharedNodeDAO.NotifyNodesWithOfflineDay(tx)
+	if err != nil {
+		return err
+	}
+
+	return nil
+}
+
+func (this *NodeScheduleTask) recoverNodes(tx *dbs.Tx) error {
+	nodeIds, err := models.SharedNodeDAO.FindNodeIdsWithExpiredActions(tx)
+	if err != nil {
+		return err
+	}
+	for _, nodeId := range nodeIds {
+		err = models.SharedNodeDAO.ResetNodeActionStatus(tx, nodeId, true)
+		if err != nil {
+			return err
+		}
+	}
+	return nil
+}