zk conn

pkg/rpcclient/auth.go

@@ -19,7 +19,7 @@ func NewAuth(discov discoveryregistry.SvcDiscoveryRegistry) *Auth {
 }
 
 type Auth struct {
-	conn   *grpc.ClientConn
+	conn   grpc.ClientConnInterface
 	Client auth.AuthClient
 	discov discoveryregistry.SvcDiscoveryRegistry
 }

pkg/rpcclient/conversation.go

@@ -14,7 +14,7 @@ import (
 
 type Conversation struct {
 	Client conversation.ConversationClient
-	conn   *grpc.ClientConn
+	conn   grpc.ClientConnInterface
 	discov discoveryregistry.SvcDiscoveryRegistry
 }
 

pkg/rpcclient/friend.go

@@ -11,7 +11,7 @@ import (
 )
 
 type Friend struct {
-	conn   *grpc.ClientConn
+	conn   grpc.ClientConnInterface
 	Client friend.FriendClient
 	discov discoveryregistry.SvcDiscoveryRegistry
 }

pkg/rpcclient/group.go

@@ -15,7 +15,7 @@ import (
 )
 
 type Group struct {
-	conn   *grpc.ClientConn
+	conn   grpc.ClientConnInterface
 	Client group.GroupClient
 	discov discoveryregistry.SvcDiscoveryRegistry
 }

pkg/rpcclient/msg.go

@@ -104,7 +104,7 @@ func newSessionTypeConf() map[int32]int32 {
 }
 
 type Message struct {
-	conn   *grpc.ClientConn
+	conn   grpc.ClientConnInterface
 	Client msg.MsgClient
 	discov discoveryregistry.SvcDiscoveryRegistry
 }

pkg/rpcclient/push.go

@@ -10,7 +10,7 @@ import (
 )
 
 type Push struct {
-	conn   *grpc.ClientConn
+	conn   grpc.ClientConnInterface
 	Client push.PushMsgServiceClient
 	discov discoveryregistry.SvcDiscoveryRegistry
 }

pkg/rpcclient/third.go

@@ -10,7 +10,7 @@ import (
 )
 
 type Third struct {
-	conn   *grpc.ClientConn
+	conn   grpc.ClientConnInterface
 	Client third.ThirdClient
 	discov discoveryregistry.SvcDiscoveryRegistry
 }

pkg/rpcclient/user.go

@@ -15,7 +15,7 @@ import (
 )
 
 type User struct {
-	conn   *grpc.ClientConn
+	conn   grpc.ClientConnInterface
 	Client user.UserClient
 	discov discoveryregistry.SvcDiscoveryRegistry
 }