61 lines
1.1 KiB
Go
61 lines
1.1 KiB
Go
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
|
|
}
|