fix(extension): migrate discovery selector logger and otel registries

Fixes: #3247

Author: xxs588

common/extension/logger.go

@@ -26,15 +26,15 @@ import (
 	"dubbo.apache.org/dubbo-go/v3/logger"
 )
 
-var logs = make(map[string]func(config *common.URL) (logger.Logger, error))
+var logs = NewRegistry[func(config *common.URL) (logger.Logger, error)]("logger")
 
 func SetLogger(driver string, log func(config *common.URL) (logger.Logger, error)) {
-	logs[driver] = log
+	logs.Register(driver, log)
 }
 
 func GetLogger(driver string, config *common.URL) (logger.Logger, error) {
-	if logs[driver] != nil {
-		return logs[driver](config)
+	if logCreator, ok := logs.Get(driver); ok {
+		return logCreator(config)
 	} else {
 		return nil, errors.Errorf("logger for %s does not exist. "+
 			"please make sure that you have imported the package "+

common/extension/otel_trace.go

@@ -29,14 +29,14 @@ import (
 	"dubbo.apache.org/dubbo-go/v3/otel/trace"
 )
 
-var traceExporterMap = make(map[string]func(config *trace.ExporterConfig) (trace.Exporter, error), 4)
+var traceExporterMap = NewRegistry[func(config *trace.ExporterConfig) (trace.Exporter, error)]("trace exporter")
 
 func SetTraceExporter(name string, createFunc func(config *trace.ExporterConfig) (trace.Exporter, error)) {
-	traceExporterMap[name] = createFunc
+	traceExporterMap.Register(name, createFunc)
 }
 
 func GetTraceExporter(name string, config *trace.ExporterConfig) (trace.Exporter, error) {
-	createFunc, ok := traceExporterMap[name]
+	createFunc, ok := traceExporterMap.Get(name)
 	if !ok {
 		panic("Cannot find the trace provider with name " + name)
 	}
@@ -45,7 +45,7 @@ func GetTraceExporter(name string, config *trace.ExporterConfig) (trace.Exporter
 
 func GetTraceShutdownCallback() func() {
 	return func() {
-		for name, createFunc := range traceExporterMap {
+		for name, createFunc := range traceExporterMap.Snapshot() {
 			if exporter, err := createFunc(nil); err == nil {
 				if err := exporter.GetTracerProvider().Shutdown(context.Background()); err != nil {
 					logger.Errorf("Graceful shutdown --- Failed to shutdown trace provider %s, error: %s", name, err.Error())

common/extension/service_discovery.go

@@ -27,13 +27,13 @@ import (
 	"dubbo.apache.org/dubbo-go/v3/registry"
 )
 
-var discoveryCreatorMap = make(map[string]func(url *common.URL) (registry.ServiceDiscovery, error), 4)
+var discoveryCreatorMap = NewRegistry[func(url *common.URL) (registry.ServiceDiscovery, error)]("service discovery")
 
 // SetServiceDiscovery will store the @creator and @name
 // protocol indicate the implementation, like nacos
 // the name like nacos-1...
 func SetServiceDiscovery(protocol string, creator func(url *common.URL) (registry.ServiceDiscovery, error)) {
-	discoveryCreatorMap[protocol] = creator
+	discoveryCreatorMap.Register(protocol, creator)
 }
 
 // GetServiceDiscovery will return the registry.ServiceDiscovery
@@ -42,7 +42,7 @@ func SetServiceDiscovery(protocol string, creator func(url *common.URL) (registr
 // if not found, or initialize instance failed, it will return error.
 func GetServiceDiscovery(url *common.URL) (registry.ServiceDiscovery, error) {
 	protocol := url.GetParam(constant.RegistryKey, "")
-	creator, ok := discoveryCreatorMap[protocol]
+	creator, ok := discoveryCreatorMap.Get(protocol)
 	if !ok {
 		return nil, perrors.New("Could not find the service discovery with discovery protocol: " + protocol)
 	}

common/extension/service_instance_selector_factory.go

@@ -25,17 +25,17 @@ import (
 	"dubbo.apache.org/dubbo-go/v3/registry/servicediscovery/instance"
 )
 
-var serviceInstanceSelectorMappings = make(map[string]func() instance.ServiceInstanceSelector, 2)
+var serviceInstanceSelectorMappings = NewRegistry[func() instance.ServiceInstanceSelector]("service instance selector")
 
 // SetServiceInstanceSelector registers a factory for ServiceInstanceSelector.
 func SetServiceInstanceSelector(name string, f func() instance.ServiceInstanceSelector) {
-	serviceInstanceSelectorMappings[name] = f
+	serviceInstanceSelectorMappings.Register(name, f)
 }
 
 // GetServiceInstanceSelector will create an instance
 // it will panic if selector with the @name not found
 func GetServiceInstanceSelector(name string) (instance.ServiceInstanceSelector, error) {
-	serviceInstanceSelector, ok := serviceInstanceSelectorMappings[name]
+	serviceInstanceSelector, ok := serviceInstanceSelectorMappings.Get(name)
 	if !ok {
 		return nil, perrors.New("Could not find service instance selector with" +
 			"name:" + name)