feat: reduce overhead to enumerate ip addresses in ServiceInfo

bdraco · bdraco · commit bbc5c7bc4ca9 · 2023-06-12T11:43:05.000-10:00
Most of the reply overhead was because of the enums
diff --git a/src/zeroconf/_services/info.py b/src/zeroconf/_services/info.py
@@ -245,14 +245,24 @@ def addresses_by_version(self, version: IPVersion) -> List[bytes]:
         This means the first address will always be the most recently added
         address of the given IP version.
         """
+        if version == IPVersion.All:
+            return [
+                *(addr.packed for addr in self._ipv4_addresses),
+                *(addr.packed for addr in self._ipv6_addresses),
+            ]
         if version == IPVersion.V4Only:
             return [addr.packed for addr in self._ipv4_addresses]
-        if version == IPVersion.V6Only:
-            return [addr.packed for addr in self._ipv6_addresses]
-        return [
-            *(addr.packed for addr in self._ipv4_addresses),
-            *(addr.packed for addr in self._ipv6_addresses),
-        ]
+        return [addr.packed for addr in self._ipv6_addresses]
+
+    def _ip_addresses_by_version_value(
+        self, version_value: int
+    ) -> Union[List[ipaddress.IPv4Address], List[ipaddress.IPv6Address], List[ipaddress._BaseAddress]]:
+        """Backend for addresses_by_version that uses the raw value."""
+        if version_value == IPVersion.All.value:
+            return [*self._ipv4_addresses, *self._ipv6_addresses]
+        if version_value == IPVersion.V4Only.value:
+            return self._ipv4_addresses
+        return self._ipv6_addresses
 
     def ip_addresses_by_version(
         self, version: IPVersion
@@ -265,11 +275,7 @@ def ip_addresses_by_version(
         This means the first address will always be the most recently added
         address of the given IP version.
         """
-        if version == IPVersion.V4Only:
-            return self._ipv4_addresses
-        if version == IPVersion.V6Only:
-            return self._ipv6_addresses
-        return [*self._ipv4_addresses, *self._ipv6_addresses]
+        return self._ip_addresses_by_version_value(version.value)
 
     def parsed_addresses(self, version: IPVersion = IPVersion.All) -> List[str]:
         """List addresses in their parsed string form.
@@ -280,7 +286,7 @@ def parsed_addresses(self, version: IPVersion = IPVersion.All) -> List[str]:
         This means the first address will always be the most recently added
         address of the given IP version.
         """
-        return [str(addr) for addr in self.ip_addresses_by_version(version)]
+        return [str(addr) for addr in self._ip_addresses_by_version_value(version.value)]
 
     def parsed_scoped_addresses(self, version: IPVersion = IPVersion.All) -> List[str]:
         """Equivalent to parsed_addresses, with the exception that IPv6 Link-Local
@@ -296,7 +302,7 @@ def parsed_scoped_addresses(self, version: IPVersion = IPVersion.All) -> List[st
             return self.parsed_addresses(version)
         return [
             f"{addr}%{self.interface_index}" if addr.version == 6 and addr.is_link_local else str(addr)
-            for addr in self.ip_addresses_by_version(version)
+            for addr in self._ip_addresses_by_version_value(version.value)
         ]
 
     def _set_properties(self, properties: Dict) -> None:
@@ -494,7 +500,7 @@ def dns_addresses(
                 address.packed,
                 created=created,
             )
-            for address in self.ip_addresses_by_version(version)
+            for address in self._ip_addresses_by_version_value(version.value)
         ]
 
     def dns_pointer(self, override_ttl: Optional[int] = None, created: Optional[float] = None) -> DNSPointer:
