fix #2758, findMappedServices panic (#2768)

* fix #2758, findMappedServices panic * update: run go fmt --------- Co-authored-by: zhangcunli <[email protected]>
apache · Dec 27, 2024 · 8f9a47a · 8f9a47a
1 parent cade298
commit 8f9a47a
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/registry/servicediscovery/service_discovery_registry.go b/registry/servicediscovery/service_discovery_registry.go
@@ -346,16 +346,20 @@ func (s *serviceDiscoveryRegistry) findMappedServices(url *common.URL, listener
 	}
 	if listener != nil {
 		protocolServiceKey := url.ServiceKey() + ":" + url.Protocol
+		s.lock.Lock()
 		s.serviceMappingListeners[protocolServiceKey] = listener
+		s.lock.Unlock()
 	}
 	return serviceNames
 }
 
 func (s *serviceDiscoveryRegistry) stopListen(url *common.URL) {
 	protocolServiceKey := url.ServiceKey() + ":" + url.Protocol
+	s.lock.Lock()
 	listener := s.serviceMappingListeners[protocolServiceKey]
 	if listener != nil {
 		delete(s.serviceMappingListeners, protocolServiceKey)
 		listener.Stop()
 	}
+	s.lock.Unlock()
 }