瀏覽代碼

feat: 新增不使用服务发现,指定endpoint获取grpc连接的方法

lihf 1 年之前
父節點
當前提交
4376c1cec6
共有 1 個文件被更改,包括 20 次插入0 次删除
  1. 20 0
      grpc/client.go

+ 20 - 0
grpc/client.go

@@ -50,6 +50,26 @@ func GetDialInsecure(
 	return conn, nil
 }
 
+func GetDialSimple(clientName string, endpoint string, option *DialOption) (*grpc2.ClientConn, error) {
+	option = checkOption(option)
+
+	if _, ok := connMap[clientName]; ok {
+		return connMap[clientName], nil
+	}
+
+	conn, err := grpc.DialInsecure(context.Background(),
+		grpc.WithEndpoint(endpoint),
+		grpc.WithMiddleware(option.Middlewares...),
+		grpc.WithTimeout(option.Timeout),
+		grpc.WithOptions(option.GrpcOptions...),
+	)
+	if err != nil {
+		return nil, err
+	}
+	connMap[clientName] = conn
+	return conn, nil
+}
+
 func NewGrpcClientOption(logger log.Logger) *DialOption {
 	option := &DialOption{
 		Middlewares: make([]middleware.Middleware, 0),