mirror of
https://github.com/TeaOSLab/EdgeAPI.git
synced 2025-11-06 10:00:24 +08:00
52 lines
968 B
Go
52 lines
968 B
Go
package tasks
|
||
|
||
import (
|
||
"github.com/TeaOSLab/EdgeAPI/internal/db/models"
|
||
"github.com/iwind/TeaGo/dbs"
|
||
"github.com/iwind/TeaGo/logs"
|
||
"time"
|
||
)
|
||
|
||
func init() {
|
||
dbs.OnReadyDone(func() {
|
||
go NewNodeTaskExtractor().Start()
|
||
})
|
||
}
|
||
|
||
// 节点任务
|
||
type NodeTaskExtractor struct {
|
||
}
|
||
|
||
func NewNodeTaskExtractor() *NodeTaskExtractor {
|
||
return &NodeTaskExtractor{}
|
||
}
|
||
|
||
func (this *NodeTaskExtractor) Start() {
|
||
ticker := time.NewTicker(10 * time.Second)
|
||
for range ticker.C {
|
||
err := this.Loop()
|
||
if err != nil {
|
||
logs.Println("[TASK][NODE_TASK_EXTRACTOR]" + err.Error())
|
||
}
|
||
}
|
||
}
|
||
|
||
func (this *NodeTaskExtractor) Loop() error {
|
||
ok, err := models.SharedSysLockerDAO.Lock(nil, "node_task_extractor", 10-1) // 假设执行时间为1秒
|
||
if err != nil {
|
||
return err
|
||
}
|
||
if !ok {
|
||
return nil
|
||
}
|
||
|
||
// 这里不解锁，是为了让任务N秒钟之内只运行一次
|
||
|
||
err = models.SharedNodeTaskDAO.ExtractAllClusterTasks(nil)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
return nil
|
||
}
|