pull/2318/head
skiffer-git 6 months ago
parent 8dc6c490d9
commit 140d9625fe

@ -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 {

Loading…
Cancel
Save