blackteay/open-im-server
1
0
Fork 0
mirror of https://github.com/openimsdk/open-im-server.git synced 2025-05-29 09:19:18 +08:00
Code Issues Packages Projects Releases Wiki Activity

close conn

Browse Source
This commit is contained in:
wangchuxiao 2023-06-21 10:49:26 +08:00
parent d1347c3db0
commit 37aab966df
2 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
pkg/discoveryregistry/discovery_register.go
View File

@ -11,6 +11,7 @@ type Conn interface {
GetConns(ctx context.Context, serviceName string, opts ...grpc.DialOption) ([]*grpc.ClientConn, error) GetConns(ctx context.Context, serviceName string, opts ...grpc.DialOption) ([]*grpc.ClientConn, error)
GetConn(ctx context.Context, serviceName string, opts ...grpc.DialOption) (*grpc.ClientConn, error) GetConn(ctx context.Context, serviceName string, opts ...grpc.DialOption) (*grpc.ClientConn, error)
AddOption(opts ...grpc.DialOption) AddOption(opts ...grpc.DialOption)
CloseConn(conn *grpc.ClientConn)
// do not use this method for call rpc // do not use this method for call rpc
GetClientLocalConns() map[string][]resolver.Address GetClientLocalConns() map[string][]resolver.Address
} }

4
pkg/discoveryregistry/zookeeper/discover.go
View File

@ -104,3 +104,7 @@ func (s *ZkClient) GetConn(ctx context.Context, serviceName string, opts ...grpc
newOpts := append(s.options, grpc.WithDefaultServiceConfig(fmt.Sprintf(`{"LoadBalancingPolicy": "%s"}`, s.balancerName))) newOpts := append(s.options, grpc.WithDefaultServiceConfig(fmt.Sprintf(`{"LoadBalancingPolicy": "%s"}`, s.balancerName)))
return grpc.DialContext(ctx, fmt.Sprintf("%s:///%s", s.scheme, serviceName), append(newOpts, opts...)...) return grpc.DialContext(ctx, fmt.Sprintf("%s:///%s", s.scheme, serviceName), append(newOpts, opts...)...)
} }
func (s *ZkClient) CloseConn(conn *grpc.ClientConn) {
conn.Close()
}