feat: speed up the service registry with a cython pxd (#1226)

build_ext.py

@@ -28,6 +28,7 @@ def build(setup_kwargs: Any) -> None:
                         "src/zeroconf/_listener.py",
                         "src/zeroconf/_protocol/incoming.py",
                         "src/zeroconf/_protocol/outgoing.py",
+                        "src/zeroconf/_services/registry.py",
                     ],
                     compiler_directives={"language_level": "3"},  # Python 3
                 ),

src/zeroconf/_services/registry.pxd

@@ -0,0 +1,18 @@
+
+import cython
+
+
+cdef class ServiceRegistry:
+
+    cdef cython.dict _services
+    cdef public cython.dict types
+    cdef public cython.dict servers
+
+    @cython.locals(
+        record_list=cython.list,
+    )
+    cdef _async_get_by_index(self, cython.dict records, str key)
+
+    cdef _add(self, object info)
+
+    cdef _remove(self, cython.list infos)

src/zeroconf/_services/registry.py

@@ -78,7 +78,10 @@ def async_get_infos_server(self, server: str) -> List[ServiceInfo]:
 
     def _async_get_by_index(self, records: Dict[str, List], key: str) -> List[ServiceInfo]:
         """Return all ServiceInfo matching the index."""
-        return [self._services[name] for name in records.get(key, [])]
+        record_list = records.get(key)
+        if record_list is None:
+            return []
+        return [self._services[name] for name in record_list]
 
     def _add(self, info: ServiceInfo) -> None:
         """Add a new service under the lock."""