mirror of
https://github.com/TeaOSLab/EdgeAPI.git
synced 2025-11-05 01:20:25 +08:00
59 lines
1.1 KiB
Go
59 lines
1.1 KiB
Go
package tasks
|
||
|
||
import (
|
||
"time"
|
||
|
||
"github.com/TeaOSLab/EdgeAPI/internal/db/models"
|
||
"github.com/TeaOSLab/EdgeAPI/internal/goman"
|
||
"github.com/TeaOSLab/EdgeCommon/pkg/nodeconfigs"
|
||
"github.com/iwind/TeaGo/dbs"
|
||
)
|
||
|
||
func init() {
|
||
dbs.OnReadyDone(func() {
|
||
goman.New(func() {
|
||
NewNodeTaskExtractor(10 * time.Second).Start()
|
||
})
|
||
})
|
||
}
|
||
|
||
// NodeTaskExtractor 节点任务
|
||
type NodeTaskExtractor struct {
|
||
BaseTask
|
||
|
||
ticker *time.Ticker
|
||
}
|
||
|
||
func NewNodeTaskExtractor(duration time.Duration) *NodeTaskExtractor {
|
||
return &NodeTaskExtractor{
|
||
ticker: time.NewTicker(duration),
|
||
}
|
||
}
|
||
|
||
func (this *NodeTaskExtractor) Start() {
|
||
for range this.ticker.C {
|
||
err := this.Loop()
|
||
if err != nil {
|
||
this.logErr("NodeTaskExtractor", err.Error())
|
||
}
|
||
}
|
||
}
|
||
|
||
func (this *NodeTaskExtractor) Loop() error {
|
||
// 检查是否为主节点
|
||
if !this.IsPrimaryNode() {
|
||
return nil
|
||
}
|
||
|
||
// 这里不解锁，是为了让任务N秒钟之内只运行一次
|
||
|
||
for _, role := range []string{nodeconfigs.NodeRoleNode, nodeconfigs.NodeRoleDNS} {
|
||
err := models.SharedNodeTaskDAO.ExtractAllClusterTasks(nil, role)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
}
|
||
|
||
return nil
|
||
}
|