Skip to content
Snippets Groups Projects
Commit f1d1a0cb authored by zhangshen023's avatar zhangshen023
Browse files

optimized some code

parent ad20922b
No related branches found
No related tags found
No related merge requests found
...@@ -42,7 +42,12 @@ import ( ...@@ -42,7 +42,12 @@ import (
) )
const ( const (
enable = "enable" enable = "enable"
watch_type = "type"
watch_type_service = "service"
watch_service = "service"
watch_passingonly = "passingonly"
watch_passingonly_true = true
) )
// init will put the service discovery into extension // init will put the service discovery into extension
...@@ -198,10 +203,7 @@ func (csd *consulServiceDiscovery) GetServices() *gxset.HashSet { ...@@ -198,10 +203,7 @@ func (csd *consulServiceDiscovery) GetServices() *gxset.HashSet {
// encodeConsulMetadata because consul validate key strictly. // encodeConsulMetadata because consul validate key strictly.
func encodeConsulMetadata(metadata map[string]string) map[string]string { func encodeConsulMetadata(metadata map[string]string) map[string]string {
consulMetadata := make(map[string]string, 8) consulMetadata := make(map[string]string, len(metadata))
if metadata == nil {
return consulMetadata
}
encoder := base64.RawStdEncoding encoder := base64.RawStdEncoding
for k, v := range metadata { for k, v := range metadata {
consulMetadata[encoder.EncodeToString([]byte(k))] = v consulMetadata[encoder.EncodeToString([]byte(k))] = v
...@@ -211,10 +213,7 @@ func encodeConsulMetadata(metadata map[string]string) map[string]string { ...@@ -211,10 +213,7 @@ func encodeConsulMetadata(metadata map[string]string) map[string]string {
// nolint // nolint
func decodeConsulMetadata(metadata map[string]string) map[string]string { func decodeConsulMetadata(metadata map[string]string) map[string]string {
dubboMetadata := make(map[string]string, 8) dubboMetadata := make(map[string]string, len(metadata))
if metadata == nil {
return dubboMetadata
}
encoder := base64.RawStdEncoding encoder := base64.RawStdEncoding
for k, v := range metadata { for k, v := range metadata {
kBytes, err := encoder.DecodeString(k) kBytes, err := encoder.DecodeString(k)
...@@ -305,9 +304,9 @@ func (csd *consulServiceDiscovery) GetRequestInstances(serviceNames []string, of ...@@ -305,9 +304,9 @@ func (csd *consulServiceDiscovery) GetRequestInstances(serviceNames []string, of
func (csd *consulServiceDiscovery) AddListener(listener *registry.ServiceInstancesChangedListener) error { func (csd *consulServiceDiscovery) AddListener(listener *registry.ServiceInstancesChangedListener) error {
params := make(map[string]interface{}, 8) params := make(map[string]interface{}, 8)
params["type"] = "service" params[watch_type] = watch_type_service
params["service"] = listener.ServiceName params[watch_service] = listener.ServiceName
params["passingonly"] = true params[watch_passingonly] = watch_passingonly_true
plan, err := watch.Parse(params) plan, err := watch.Parse(params)
if err != nil { if err != nil {
logger.Errorf("add listener for service %s,error:%v", listener.ServiceName, err) logger.Errorf("add listener for service %s,error:%v", listener.ServiceName, err)
......
...@@ -124,8 +124,7 @@ func creatServiceDiscovery(url *common.URL) (registry.ServiceDiscovery, error) { ...@@ -124,8 +124,7 @@ func creatServiceDiscovery(url *common.URL) (registry.ServiceDiscovery, error) {
if err != nil { if err != nil {
return nil, perrors.WithMessage(err, "Create service discovery fialed") return nil, perrors.WithMessage(err, "Create service discovery fialed")
} }
serviceDiscovery := event.NewEventPublishingServiceDiscovery(originServiceDiscovery) return event.NewEventPublishingServiceDiscovery(originServiceDiscovery), nil
return serviceDiscovery, nil
} }
func parseServices(literalServices string) *gxset.HashSet { func parseServices(literalServices string) *gxset.HashSet {
......
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