节点可以单独设置所使用的API节点地址

internal/nodes/task_sync_api_nodes.go

@@ -1,23 +1,15 @@
 package nodes
 
 import (
-	"context"
-	"crypto/tls"
 	"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
 	"github.com/TeaOSLab/EdgeNode/internal/configs"
 	"github.com/TeaOSLab/EdgeNode/internal/events"
 	"github.com/TeaOSLab/EdgeNode/internal/goman"
 	"github.com/TeaOSLab/EdgeNode/internal/rpc"
 	"github.com/TeaOSLab/EdgeNode/internal/trackers"
+	"github.com/TeaOSLab/EdgeNode/internal/utils"
 	"github.com/iwind/TeaGo/Tea"
 	"github.com/iwind/TeaGo/logs"
-	"google.golang.org/grpc"
-	"google.golang.org/grpc/credentials"
-	"google.golang.org/grpc/credentials/insecure"
-	"net/url"
-	"sort"
-	"strings"
-	"sync"
 	"time"
 )
 
@@ -64,6 +56,9 @@ func (this *SyncAPINodesTask) Stop() {
 }
 
 func (this *SyncAPINodesTask) Loop() error {
+	// 如果有节点定制的API节点地址
+	var hasCustomizedAPINodeAddrs = sharedNodeConfig != nil && len(sharedNodeConfig.APINodeAddrs) > 0
+
 	config, err := configs.LoadAPIConfig()
 	if err != nil {
 		return err
@@ -96,21 +91,25 @@ func (this *SyncAPINodesTask) Loop() error {
 	}
 
 	// 和现有的对比
-	if this.isSame(newEndpoints, config.RPC.Endpoints) {
+	if utils.EqualStrings(newEndpoints, config.RPC.Endpoints) {
 		return nil
 	}
 
 	// 测试是否有API节点可用
-	var hasOk = this.testEndpoints(newEndpoints)
+	var hasOk = rpcClient.TestEndpoints(newEndpoints)
 	if !hasOk {
 		return nil
 	}
 
 	// 修改RPC对象配置
 	config.RPC.Endpoints = newEndpoints
-	err = rpcClient.UpdateConfig(config)
-	if err != nil {
-		return err
+
+	// 更新当前RPC
+	if !hasCustomizedAPINodeAddrs {
+		err = rpcClient.UpdateConfig(config)
+		if err != nil {
+			return err
+		}
 	}
 
 	// 保存到文件
@@ -121,53 +120,3 @@ func (this *SyncAPINodesTask) Loop() error {
 
 	return nil
 }
-
-func (this *SyncAPINodesTask) isSame(endpoints1 []string, endpoints2 []string) bool {
-	sort.Strings(endpoints1)
-	sort.Strings(endpoints2)
-	return strings.Join(endpoints1, "&") == strings.Join(endpoints2, "&")
-}
-
-func (this *SyncAPINodesTask) testEndpoints(endpoints []string) bool {
-	if len(endpoints) == 0 {
-		return false
-	}
-
-	var wg = sync.WaitGroup{}
-	wg.Add(len(endpoints))
-
-	var ok = false
-
-	for _, endpoint := range endpoints {
-		go func(endpoint string) {
-			defer wg.Done()
-
-			u, err := url.Parse(endpoint)
-			if err != nil {
-				return
-			}
-
-			ctx, cancelFunc := context.WithTimeout(context.Background(), 5*time.Second)
-			defer func() {
-				cancelFunc()
-			}()
-			var conn *grpc.ClientConn
-			if u.Scheme == "http" {
-				conn, err = grpc.DialContext(ctx, u.Host, grpc.WithTransportCredentials(insecure.NewCredentials()), grpc.WithBlock())
-			} else if u.Scheme == "https" {
-				conn, err = grpc.DialContext(ctx, u.Host, grpc.WithTransportCredentials(credentials.NewTLS(&tls.Config{
-					InsecureSkipVerify: true,
-				})), grpc.WithBlock())
-			}
-			if err != nil {
-				return
-			}
-			_ = conn.Close()
-
-			ok = true
-		}(endpoint)
-	}
-	wg.Wait()
-
-	return ok
-}