119 lines
2.5 KiB
Go
119 lines
2.5 KiB
Go
package nodes
|
|
|
|
import (
|
|
"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
|
|
"github.com/TeaOSLab/EdgeDNS/internal/configs"
|
|
teaconst "github.com/TeaOSLab/EdgeDNS/internal/const"
|
|
"github.com/TeaOSLab/EdgeDNS/internal/events"
|
|
"github.com/TeaOSLab/EdgeDNS/internal/goman"
|
|
"github.com/TeaOSLab/EdgeDNS/internal/remotelogs"
|
|
"github.com/TeaOSLab/EdgeDNS/internal/rpc"
|
|
"github.com/TeaOSLab/EdgeDNS/internal/utils"
|
|
"github.com/iwind/TeaGo/Tea"
|
|
"github.com/iwind/TeaGo/logs"
|
|
"time"
|
|
)
|
|
|
|
func init() {
|
|
if !teaconst.IsMain {
|
|
return
|
|
}
|
|
|
|
events.On(events.EventStart, func() {
|
|
task := NewSyncAPINodesTask()
|
|
goman.New(func() {
|
|
task.Start()
|
|
})
|
|
})
|
|
}
|
|
|
|
// SyncAPINodesTask API节点同步任务
|
|
type SyncAPINodesTask struct {
|
|
}
|
|
|
|
func NewSyncAPINodesTask() *SyncAPINodesTask {
|
|
return &SyncAPINodesTask{}
|
|
}
|
|
|
|
func (this *SyncAPINodesTask) Start() {
|
|
ticker := time.NewTicker(5 * time.Minute)
|
|
if Tea.IsTesting() {
|
|
// 快速测试
|
|
ticker = time.NewTicker(1 * time.Minute)
|
|
}
|
|
events.On(events.EventQuit, func() {
|
|
remotelogs.Println("SYNC_API_NODES_TASK", "quit task")
|
|
ticker.Stop()
|
|
})
|
|
for range ticker.C {
|
|
err := this.Loop()
|
|
if err != nil {
|
|
logs.Println("[TASK][SYNC_API_NODES_TASK]" + err.Error())
|
|
}
|
|
}
|
|
}
|
|
|
|
func (this *SyncAPINodesTask) Loop() error {
|
|
// 如果有节点定制的API节点地址
|
|
var nodeConfig = dnsNodeConfig()
|
|
var hasCustomizedAPINodeAddrs = nodeConfig != nil && len(nodeConfig.APINodeAddrs) > 0
|
|
|
|
config, err := configs.LoadAPIConfig()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// 是否禁止自动升级
|
|
if config.RPCDisableUpdate {
|
|
return nil
|
|
}
|
|
|
|
// 获取所有可用的节点
|
|
rpcClient, err := rpc.SharedRPC()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
resp, err := rpcClient.APINodeRPC.FindAllEnabledAPINodes(rpcClient.Context(), &pb.FindAllEnabledAPINodesRequest{})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
var newEndpoints = []string{}
|
|
for _, node := range resp.ApiNodes {
|
|
if !node.IsOn {
|
|
continue
|
|
}
|
|
newEndpoints = append(newEndpoints, node.AccessAddrs...)
|
|
}
|
|
|
|
// 和现有的对比
|
|
if utils.EqualStrings(newEndpoints, config.RPCEndpoints) {
|
|
return nil
|
|
}
|
|
|
|
// 测试是否有API节点可用
|
|
var hasOk = rpcClient.TestEndpoints(newEndpoints)
|
|
if !hasOk {
|
|
return nil
|
|
}
|
|
|
|
// 修改RPC对象配置
|
|
config.RPCEndpoints = newEndpoints
|
|
|
|
// 更新当前RPC
|
|
if !hasCustomizedAPINodeAddrs {
|
|
err = rpcClient.UpdateConfig(config)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
// 保存到文件
|
|
err = config.WriteFile(Tea.ConfigFile(configs.ConfigFileName))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|