2525
2626
2727from .._exceptions import ServiceNameAlreadyRegistered
28- from ..services import ServiceInfo
28+ from .._services import ServiceInfo
2929
3030
3131class ServiceRegistry :
@@ -40,7 +40,7 @@ def __init__(
4040 self ,
4141 ) -> None :
4242 """Create the ServiceRegistry class."""
43- self .services = {} # type: Dict[str, ServiceInfo]
43+ self ._services = {} # type: Dict[str, ServiceInfo]
4444 self .types = {} # type: Dict[str, List]
4545 self .servers = {} # type: Dict[str, List]
4646 self ._lock = threading .Lock () # add and remove services thread safe
@@ -66,11 +66,11 @@ def update(self, info: ServiceInfo) -> None:
6666
6767 def get_service_infos (self ) -> List [ServiceInfo ]:
6868 """Return all ServiceInfo."""
69- return list (self .services .values ())
69+ return list (self ._services .values ())
7070
7171 def get_info_name (self , name : str ) -> Optional [ServiceInfo ]:
7272 """Return all ServiceInfo for the name."""
73- return self .services .get (name )
73+ return self ._services .get (name )
7474
7575 def get_types (self ) -> List [str ]:
7676 """Return all types."""
@@ -89,7 +89,7 @@ def _get_by_index(self, attr: str, key: str) -> List[ServiceInfo]:
8989 service_infos = []
9090
9191 for name in getattr (self , attr ).get (key , [])[:]:
92- info = self .services .get (name )
92+ info = self ._services .get (name )
9393 # Since we do not get under a lock since it would be
9494 # a performance issue, its possible
9595 # the service can be unregistered during the get
@@ -102,17 +102,17 @@ def _get_by_index(self, attr: str, key: str) -> List[ServiceInfo]:
102102 def _add (self , info : ServiceInfo ) -> None :
103103 """Add a new service under the lock."""
104104 lower_name = info .name .lower ()
105- if lower_name in self .services :
105+ if lower_name in self ._services :
106106 raise ServiceNameAlreadyRegistered
107107
108- self .services [lower_name ] = info
108+ self ._services [lower_name ] = info
109109 self .types .setdefault (info .type , []).append (lower_name )
110110 self .servers .setdefault (info .server , []).append (lower_name )
111111
112112 def _remove (self , info : ServiceInfo ) -> None :
113113 """Remove a service under the lock."""
114114 lower_name = info .name .lower ()
115- old_service_info = self .services [lower_name ]
115+ old_service_info = self ._services [lower_name ]
116116 self .types [old_service_info .type ].remove (lower_name )
117117 self .servers [old_service_info .server ].remove (lower_name )
118- del self .services [lower_name ]
118+ del self ._services [lower_name ]
0 commit comments