78 lines
1.4 KiB
Go
78 lines
1.4 KiB
Go
// 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
|
|
}
|