فهرست منبع

新增clientName

dcsunny 2 سال پیش
والد
کامیت
818e1a3232
1فایلهای تغییر یافته به همراه4 افزوده شده و 3 حذف شده
  1. 4 3
      grpc/client.go

+ 4 - 3
grpc/client.go

@@ -24,11 +24,12 @@ type DialOption struct {
 func GetDialInsecure(
 	r *registry.Registry,
 	logger log.Logger,
+	clientName string,
 	endpointNameKey string, option *DialOption) (*grpc2.ClientConn, error) {
 	option = checkOption(option)
 
-	if _, ok := connMap[endpointNameKey]; ok {
-		return connMap[endpointNameKey], nil
+	if _, ok := connMap[clientName]; ok {
+		return connMap[clientName], nil
 	}
 	endpoint := grpc.WithEndpoint(endpointNameKey)
 	option.Middlewares = append(option.Middlewares, logging.Client(logger))
@@ -46,7 +47,7 @@ func GetDialInsecure(
 	if err != nil {
 		return nil, err
 	}
-	connMap[endpointNameKey] = conn
+	connMap[clientName] = conn
 	return conn, nil
 }