feat: now 22% speed up

bdraco · bdraco · commit 6d69ad8efae4 · 2023-11-13T13:12:35.000-06:00
diff --git a/src/zeroconf/_protocol/outgoing.pxd b/src/zeroconf/_protocol/outgoing.pxd
@@ -15,8 +15,11 @@ cdef cython.uint _FLAGS_TC
 cdef cython.uint _MAX_MSG_ABSOLUTE
 cdef cython.uint _MAX_MSG_TYPICAL
 
+
 cdef bint TYPE_CHECKING
 
+cdef unsigned int SHORT_CACHE_MAX
+
 cdef object PACK_BYTE
 cdef object PACK_SHORT
 cdef object PACK_LONG
diff --git a/src/zeroconf/_protocol/outgoing.py b/src/zeroconf/_protocol/outgoing.py
@@ -53,8 +53,10 @@
 PACK_SHORT = Struct('>H').pack
 PACK_LONG = Struct('>L').pack
 
+SHORT_CACHE_MAX = 128
+
 BYTE_TABLE = tuple(PACK_BYTE(i) for i in range(256))
-SHORT_LOOKUP = tuple(PACK_SHORT(i) for i in range(32))
+SHORT_LOOKUP = tuple(PACK_SHORT(i) for i in range(SHORT_CACHE_MAX))
 
 
 class State(enum.Enum):
@@ -223,7 +225,7 @@ def _write_byte(self, value: int_) -> None:
 
     def _get_short(self, value: int_) -> bytes:
         """Gets an unsigned short from a certain position in the packet"""
-        if value < 32:
+        if value < SHORT_CACHE_MAX:
             return SHORT_LOOKUP[value]
         return PACK_SHORT(value)
 
