Преглед на файлове

fix: GetDialInsecure 使用endpointNameKey作为key

dcsunny преди 1 година
родител
ревизия
42b469e99c
променени са 1 файла, в които са добавени 5 реда и са изтрити 5 реда
  1. 5 5
      grpc/client.go

+ 5 - 5
grpc/client.go

@@ -36,7 +36,7 @@ func GetDialInsecure(
 		return nil, nil
 	}
 	ConnLock.Lock()
-	conn, ok := ConnMap[clientName]
+	conn, ok := ConnMap[endpointNameKey]
 	ConnLock.Unlock()
 	if ok {
 		return conn, nil
@@ -53,15 +53,15 @@ func GetDialInsecure(
 		return nil, err
 	}
 	ConnLock.Lock()
-	ConnMap[clientName] = conn
+	ConnMap[endpointNameKey] = conn
 	ConnLock.Unlock()
 	return conn, nil
 }
 
-func GetDialSimple(clientName string, endpoint string, option *DialOption) (*grpc2.ClientConn, error) {
+func GetDialSimple(endpointNameKey string, endpoint string, option *DialOption) (*grpc2.ClientConn, error) {
 	option = checkOption(option)
 	ConnLock.Lock()
-	conn, ok := ConnMap[clientName]
+	conn, ok := ConnMap[endpointNameKey]
 	ConnLock.Unlock()
 	if ok {
 		return conn, nil
@@ -77,7 +77,7 @@ func GetDialSimple(clientName string, endpoint string, option *DialOption) (*grp
 		return nil, err
 	}
 	ConnLock.Lock()
-	ConnMap[clientName] = conn
+	ConnMap[endpointNameKey] = conn
 	ConnLock.Unlock()
 	return conn, nil
 }