Skip to content
Snippets Groups Projects
Commit 80405667 authored by shen's avatar shen
Browse files

optimized code for consul service discovery

parent 0ddcd72f
No related branches found
No related tags found
No related merge requests found
...@@ -136,7 +136,7 @@ type consulServiceDiscovery struct { ...@@ -136,7 +136,7 @@ type consulServiceDiscovery struct {
*consul.Config *consul.Config
} }
func (csd *consulServiceDiscovery) Initialize(registryURL common.URL) error { func (csd *consulServiceDiscovery) Init(registryURL common.URL) error {
csd.serviceUrl = registryURL csd.serviceUrl = registryURL
csd.checkPassInterval = registryURL.GetParamInt(CHECK_PASS_INTERVAL, DEFAULT_CHECK_PASS_INTERVAL) csd.checkPassInterval = registryURL.GetParamInt(CHECK_PASS_INTERVAL, DEFAULT_CHECK_PASS_INTERVAL)
csd.tag = registryURL.GetParam(QUERY_TAG, "") csd.tag = registryURL.GetParam(QUERY_TAG, "")
......
...@@ -81,7 +81,7 @@ func TestConsulServiceDiscovery_Destroy(t *testing.T) { ...@@ -81,7 +81,7 @@ func TestConsulServiceDiscovery_Destroy(t *testing.T) {
prepareData() prepareData()
serviceDiscovery, err := extension.GetServiceDiscovery(constant.CONSUL_KEY, testName) serviceDiscovery, err := extension.GetServiceDiscovery(constant.CONSUL_KEY, testName)
_, registryUrl := prepareService() _, registryUrl := prepareService()
serviceDiscovery.Initialize(registryUrl) serviceDiscovery.Init(registryUrl)
assert.Nil(t, err) assert.Nil(t, err)
assert.NotNil(t, serviceDiscovery) assert.NotNil(t, serviceDiscovery)
err = serviceDiscovery.Destroy() err = serviceDiscovery.Destroy()
...@@ -109,7 +109,7 @@ func TestConsulServiceDiscovery_CRUD(t *testing.T) { ...@@ -109,7 +109,7 @@ func TestConsulServiceDiscovery_CRUD(t *testing.T) {
serviceDiscovery, err := extension.GetServiceDiscovery(constant.CONSUL_KEY, testName) serviceDiscovery, err := extension.GetServiceDiscovery(constant.CONSUL_KEY, testName)
assert.Nil(t, err) assert.Nil(t, err)
err = serviceDiscovery.Initialize(registryUrl) err = serviceDiscovery.Init(registryUrl)
assert.Nil(t, err) assert.Nil(t, err)
err = serviceDiscovery.Unregister(instance) err = serviceDiscovery.Unregister(instance)
......
...@@ -67,7 +67,7 @@ type etcdV3ServiceDiscovery struct { ...@@ -67,7 +67,7 @@ type etcdV3ServiceDiscovery struct {
childListenerMap map[string]*etcdv3.EventListener childListenerMap map[string]*etcdv3.EventListener
} }
func (e *etcdV3ServiceDiscovery) Initialize(registryURL common.URL) error { func (e *etcdV3ServiceDiscovery) Init(registryURL common.URL) error {
return nil return nil
} }
......
...@@ -117,7 +117,7 @@ func (tel *TestServiceInstancePreRegisteredEventListener) GetEventType() reflect ...@@ -117,7 +117,7 @@ func (tel *TestServiceInstancePreRegisteredEventListener) GetEventType() reflect
type ServiceDiscoveryA struct { type ServiceDiscoveryA struct {
} }
func (msd *ServiceDiscoveryA) Initialize(registryURL common.URL) error { func (msd *ServiceDiscoveryA) Init(registryURL common.URL) error {
return nil return nil
} }
......
...@@ -38,8 +38,8 @@ import ( ...@@ -38,8 +38,8 @@ import (
// EventPublishingServiceDiscovery will enhance Service Discovery // EventPublishingServiceDiscovery will enhance Service Discovery
// Publish some event about service discovery // Publish some event about service discovery
type EventPublishingServiceDiscovery struct { type EventPublishingServiceDiscovery struct {
serviceDiscovery registry.ServiceDiscovery serviceDiscovery registry.ServiceDiscovery
once sync.Once serviceDiscoveryInitOnce sync.Once
} }
// NewEventPublishingServiceDiscovery is a constructor // NewEventPublishingServiceDiscovery is a constructor
...@@ -54,10 +54,10 @@ func (epsd *EventPublishingServiceDiscovery) String() string { ...@@ -54,10 +54,10 @@ func (epsd *EventPublishingServiceDiscovery) String() string {
return epsd.serviceDiscovery.String() return epsd.serviceDiscovery.String()
} }
func (epsd *EventPublishingServiceDiscovery) Initialize(registryURL common.URL) error { func (epsd *EventPublishingServiceDiscovery) Init(registryURL common.URL) error {
var err error var err error
epsd.once.Do(func() { epsd.serviceDiscoveryInitOnce.Do(func() {
err = epsd.serviceDiscovery.Initialize(registryURL) err = epsd.serviceDiscovery.Init(registryURL)
}) })
return err return err
} }
......
...@@ -63,7 +63,7 @@ type nacosServiceDiscovery struct { ...@@ -63,7 +63,7 @@ type nacosServiceDiscovery struct {
namingClient naming_client.INamingClient namingClient naming_client.INamingClient
} }
func (n *nacosServiceDiscovery) Initialize(registryURL common.URL) error { func (n *nacosServiceDiscovery) Init(registryURL common.URL) error {
return nil return nil
} }
......
...@@ -42,7 +42,7 @@ type ServiceDiscovery interface { ...@@ -42,7 +42,7 @@ type ServiceDiscovery interface {
* Initializes the ServiceDiscovery * Initializes the ServiceDiscovery
* *
*/ */
Initialize(registryURL common.URL) error Init(registryURL common.URL) error
// Destroy will destroy the service discovery. // Destroy will destroy the service discovery.
// If the discovery cannot be destroy, it will return an error. // If the discovery cannot be destroy, it will return an error.
......
...@@ -126,7 +126,7 @@ func creatServiceDiscovery(url *common.URL) (registry.ServiceDiscovery, error) { ...@@ -126,7 +126,7 @@ func creatServiceDiscovery(url *common.URL) (registry.ServiceDiscovery, error) {
return nil, perrors.WithMessage(err, "Create service discovery fialed") return nil, perrors.WithMessage(err, "Create service discovery fialed")
} }
serviceDiscovery := event.NewEventPublishingServiceDiscovery(originServiceDiscovery) serviceDiscovery := event.NewEventPublishingServiceDiscovery(originServiceDiscovery)
serviceDiscovery.Initialize(*url) serviceDiscovery.Init(*url)
return serviceDiscovery, nil return serviceDiscovery, nil
} }
......
...@@ -126,7 +126,7 @@ func (m *mockServiceNameMapping) Get(serviceInterface string, group string, vers ...@@ -126,7 +126,7 @@ func (m *mockServiceNameMapping) Get(serviceInterface string, group string, vers
type mockServiceDiscovery struct { type mockServiceDiscovery struct {
} }
func (m *mockServiceDiscovery) Initialize(registryURL common.URL) error { func (m *mockServiceDiscovery) Init(registryURL common.URL) error {
return nil return nil
} }
......
...@@ -163,7 +163,7 @@ func (zksd *zookeeperServiceDiscovery) String() string { ...@@ -163,7 +163,7 @@ func (zksd *zookeeperServiceDiscovery) String() string {
return fmt.Sprintf("zookeeper-service-discovery[%s]", zksd.url) return fmt.Sprintf("zookeeper-service-discovery[%s]", zksd.url)
} }
func (zksd *zookeeperServiceDiscovery) Initialize(registryURL common.URL) error { func (zksd *zookeeperServiceDiscovery) Init(registryURL common.URL) error {
return nil return nil
} }
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment