TRKJ/EdgeAPI
2
0
Fork 0
mirror of https://github.com/TeaOSLab/EdgeAPI.git synced 2025-11-12 14:50:25 +08:00
Code Issues Packages Projects Releases Wiki Activity

多个API节点时选择一个作为主节点/优化任务相关代码

Browse Source
This commit is contained in:
刘祥超
2022-04-23 12:32:30 +08:00
parent 773f3e1a7e
commit 89c1edc9ee
35 changed files with 467 additions and 350 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
internal/tasks/health_check_task.go
View File

@@ -5,50 +5,52 @@ import (
"encoding/json"
"github.com/TeaOSLab/EdgeAPI/internal/db/models"
"github.com/TeaOSLab/EdgeAPI/internal/goman"
"github.com/TeaOSLab/EdgeAPI/internal/utils"
"github.com/TeaOSLab/EdgeAPI/internal/remotelogs"
"github.com/TeaOSLab/EdgeAPI/internal/utils/numberutils"
"github.com/TeaOSLab/EdgeCommon/pkg/serverconfigs"
"github.com/iwind/TeaGo/dbs"
"github.com/iwind/TeaGo/lists"
"github.com/iwind/TeaGo/logs"
"time"
)
func init() {
dbs.OnReadyDone(func() {
goman.New(func() {
NewHealthCheckTask().Run()
NewHealthCheckTask(1 * time.Minute).Start()
})
})
}
// HealthCheckTask 节点健康检查任务
type HealthCheckTask struct {
BaseTask
ticker *time.Ticker
tasksMap map[int64]*HealthCheckClusterTask // taskId => task
}
func NewHealthCheckTask() *HealthCheckTask {
func NewHealthCheckTask(duration time.Duration) *HealthCheckTask {
return &HealthCheckTask{
ticker: time.NewTicker(duration),
tasksMap: map[int64]*HealthCheckClusterTask{},
}
}
func (this *HealthCheckTask) Run() {
err := this.loop()
func (this *HealthCheckTask) Start() {
err := this.Loop()
if err != nil {
logs.Println("[TASK][HEALTH_CHECK]" + err.Error())
this.logErr("HealthCheckTask", err.Error())
}
ticker := utils.NewTicker(60 * time.Second)
for ticker.Wait() {
err := this.loop()
for range this.ticker.C {
err := this.Loop()
if err != nil {
logs.Println("[TASK][HEALTH_CHECK]" + err.Error())
this.logErr("HealthCheckTask", err.Error())
}
}
}
func (this *HealthCheckTask) loop() error {
func (this *HealthCheckTask) Loop() error {
clusters, err := models.NewNodeClusterDAO().FindAllEnableClusters(nil)
if err != nil {
return err
@@ -74,7 +76,7 @@ func (this *HealthCheckTask) loop() error {
if len(cluster.HealthCheck) > 0 {
err = json.Unmarshal(cluster.HealthCheck, config)
if err != nil {
logs.Println("[TASK][HEALTH_CHECK]" + err.Error())
this.logErr("HealthCheckTask", err.Error())
continue
}
}
@@ -85,7 +87,7 @@ func (this *HealthCheckTask) loop() error {
newJSON, _ := json.Marshal(config)
oldJSON, _ := json.Marshal(task.Config())
if bytes.Compare(oldJSON, newJSON) != 0 {
logs.Println("[TASK][HEALTH_CHECK]update cluster '" + numberutils.FormatInt64(clusterId) + "'")
remotelogs.Println("TASK", "[HealthCheckTask]update cluster '"+numberutils.FormatInt64(clusterId)+"'")
goman.New(func() {
task.Reset(config)
})