feat: speed up processing incoming records

bdraco · bdraco · commit 35600fe83f96 · 2023-08-13T15:40:31.000-05:00
add cython defs for set_created_ttl and reset_ttl
diff --git a/src/zeroconf/_dns.pxd b/src/zeroconf/_dns.pxd
@@ -44,6 +44,11 @@ cdef class DNSRecord(DNSEntry):
     )
     cpdef suppressed_by(self, object msg)
 
+    @cython.locals(
+        remaining=cython.float,
+    )
+    cpdef get_remaining_ttl(self, cython.float now)
+
     cpdef get_expiration_time(self, cython.uint percent)
 
     cpdef is_expired(self, cython.float now)
diff --git a/src/zeroconf/_dns.py b/src/zeroconf/_dns.py
@@ -26,7 +26,7 @@
 
 from ._exceptions import AbstractMethodException
 from ._utils.net import _is_v6_address
-from ._utils.time import current_time_millis, millis_to_seconds
+from ._utils.time import current_time_millis
 from .const import _CLASS_MASK, _CLASS_UNIQUE, _CLASSES, _TYPE_ANY, _TYPES
 
 _LEN_BYTE = 1
@@ -193,7 +193,10 @@ def get_expiration_time(self, percent: _int) -> float:
     # TODO: Switch to just int here
     def get_remaining_ttl(self, now: _float) -> Union[int, float]:
         """Returns the remaining TTL in seconds."""
-        return max(0, millis_to_seconds((self.created + (_EXPIRE_FULL_TIME_MS * self.ttl)) - now))
+        remaining = self.created + (_EXPIRE_FULL_TIME_MS * self.ttl) - now
+        if remaining < 0:
+            return 0
+        return remaining / 1000.0
 
     def is_expired(self, now: _float) -> bool:
         """Returns true if this record has expired."""
