blackteay/open-im-server
1
0
Fork 0
mirror of https://github.com/openimsdk/open-im-server.git synced 2025-04-24 10:22:36 +08:00
Code Issues Packages Projects Releases Wiki Activity

add etcd

Browse Source
This commit is contained in:
skiffer-git 2024-05-11 10:55:57 +08:00
parent 8dc6c490d9
commit 140d9625fe
1 changed files with 11 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -76,7 +76,7 @@ func (r *SvcDiscoveryRegistryImpl) CloseConn(conn *grpc.ClientConn) {
} }
func (r *SvcDiscoveryRegistryImpl) Register(serviceName, host string, port int, opts ...grpc.DialOption) error { func (r *SvcDiscoveryRegistryImpl) Register(serviceName, host string, port int, opts ...grpc.DialOption) error {
r.serviceKey = fmt.Sprintf("%s/%s/%d", serviceName, host, port) r.serviceKey = fmt.Sprintf("%s/%s-%d", serviceName, host, port)
em, err := endpoints.NewManager(r.client, serviceName) em, err := endpoints.NewManager(r.client, serviceName)
if err != nil { if err != nil {
return err return err
@ -92,6 +92,16 @@ func (r *SvcDiscoveryRegistryImpl) Register(serviceName, host string, port int,
endpoint := endpoints.Endpoint{Addr: fmt.Sprintf("%s:%d", host, port)} endpoint := endpoints.Endpoint{Addr: fmt.Sprintf("%s:%d", host, port)}
err = em.AddEndpoint(context.TODO(), r.serviceKey, endpoint, clientv3.WithLease(leaseResp.ID)) err = em.AddEndpoint(context.TODO(), r.serviceKey, endpoint, clientv3.WithLease(leaseResp.ID))
return err return err
lease, _ := r.client.Grant(context.TODO(), 30)
em, err = endpoints.NewManager(r.client, "foo/bar/my-service")
if err != nil {
return err
}
err := em.AddEndpoint(context.TODO(), "foo/bar/my-service/e1", endpoints.Endpoint{Addr: "1.2.3.4"}, clientv3.WithLease(lease.ID))
} }
func (r *SvcDiscoveryRegistryImpl) UnRegister() error { func (r *SvcDiscoveryRegistryImpl) UnRegister() error {