1.4.5.2

EdgeDNS/internal/rpc/rpc_utils.go

@@ -0,0 +1,60 @@
+package rpc
+
+import (
+	"github.com/TeaOSLab/EdgeDNS/internal/configs"
+	"google.golang.org/grpc/codes"
+	"google.golang.org/grpc/status"
+	"strings"
+	"sync"
+)
+
+var sharedRPC *RPCClient = nil
+var locker = &sync.Mutex{}
+
+// SharedRPC 获取当前的RPC客户端
+func SharedRPC() (*RPCClient, error) {
+	locker.Lock()
+	defer locker.Unlock()
+
+	if sharedRPC != nil {
+		return sharedRPC, nil
+	}
+
+	config, err := configs.SharedAPIConfig()
+	if err != nil {
+		return nil, err
+	}
+	client, err := NewRPCClient(config)
+	if err != nil {
+		return nil, err
+	}
+
+	sharedRPC = client
+	return sharedRPC, nil
+}
+
+// IsConnError 是否为连接错误
+func IsConnError(err error) bool {
+	if err == nil {
+		return false
+	}
+
+	// 检查是否为连接错误
+	statusErr, ok := status.FromError(err)
+	if ok {
+		var errorCode = statusErr.Code()
+		return errorCode == codes.Unavailable || errorCode == codes.Canceled
+	}
+
+	var keywords = []string{
+		"rpc error: code = Unavailable",
+		"rpc error: code = Canceled",
+	}
+	for _, keyword := range keywords {
+		if strings.Contains(err.Error(), keyword) {
+			return true
+		}
+	}
+
+	return false
+}