EdgeAdmin/internal/rpc/rpc_utils.go

package rpc

import (
	"strings"
	"sync"

	"github.com/TeaOSLab/EdgeAdmin/internal/configs"
	"google.golang.org/grpc/codes"
	"google.golang.org/grpc/status"
)

var sharedRPC *RPCClient = nil
var locker = &sync.Mutex{}

func SharedRPC() (*RPCClient, error) {
	locker.Lock()
	defer locker.Unlock()

	if sharedRPC != nil {
		return sharedRPC, nil
	}

	config, err := configs.LoadAPIConfig()
	if err != nil {
		return nil, err
	}
	client, err := NewRPCClient(config, true)
	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
	}

	if strings.Contains(err.Error(), "code = Canceled") {
		return true
	}

	return false
}

// IsUnimplementedError 检查是否为未实现错误
func IsUnimplementedError(err error) bool {
	if err == nil {
		return false
	}

	statusErr, ok := status.FromError(err)
	if ok {
		if statusErr.Code() == codes.Unimplemented {
			return true
		}
	}

	return false
}