blackteay/open-im-server
1
0
Fork 0
mirror of https://github.com/openimsdk/open-im-server.git synced 2025-11-04 19:32:17 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add etcd as a service discovery mechanism

Browse Source
This commit is contained in:
skiffer-git 2024-05-13 11:10:25 +08:00
parent 3074604c86
commit f51cdadb57
1 changed files with 18 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
pkg/common/discoveryregister/etcd/etcd.go
View File

@ -87,12 +87,27 @@ func (r *SvcDiscoveryRegistryImpl) GetUserIdHashGatewayHost(ctx context.Context,
// GetConns returns gRPC client connections for a given service name
func (r *SvcDiscoveryRegistryImpl) GetConns(ctx context.Context, serviceName string, opts ...grpc.DialOption) ([]*grpc.ClientConn, error) {
target := fmt.Sprintf("etcd:///%s/%s", r.rootDirectory, serviceName)
conn, err := grpc.DialContext(ctx, target, append(append(r.dialOptions, opts...), grpc.WithResolvers(r.resolver))...)
var conns []*grpc.ClientConn
// Construct the full key for the service
fullServiceKey := fmt.Sprintf("%s/%s", r.rootDirectory, serviceName)
// List all endpoints for the service
resp, err := r.client.Get(ctx, fullServiceKey, clientv3.WithPrefix())
if err != nil {
return nil, err
}
return []*grpc.ClientConn{conn}, nil
for _, kv := range resp.Kvs {
endpoint := string(kv.Key[len(fullServiceKey)+1:]) // Extract the endpoint address
target := fmt.Sprintf("etcd://%s", endpoint)
conn, err := grpc.DialContext(ctx, target, append(append(r.dialOptions, opts...), grpc.WithResolvers(r.resolver))...)
if err != nil {
return nil, err
}
conns = append(conns, conn)
}
return conns, nil
}
// GetConn returns a single gRPC client connection for a given service name