소스 검색

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),