feat: cache short pack

bdraco · bdraco · commit 2f043a6699da · 2023-09-11T13:04:06.000-05:00
diff --git a/src/zeroconf/_protocol/outgoing.pxd b/src/zeroconf/_protocol/outgoing.pxd
@@ -20,6 +20,7 @@ cdef object TYPE_CHECKING
 cdef object PACK_BYTE
 cdef object PACK_SHORT
 cdef object PACK_LONG
+cdef object CACHED_PACK_SHORT
 
 cdef class DNSOutgoing:
 
diff --git a/src/zeroconf/_protocol/outgoing.py b/src/zeroconf/_protocol/outgoing.py
@@ -22,6 +22,7 @@
 
 import enum
 import logging
+from functools import lru_cache
 from struct import Struct
 from typing import TYPE_CHECKING, Dict, List, Optional, Sequence, Tuple, Union
 
@@ -53,6 +54,8 @@
 PACK_SHORT = Struct('>H').pack
 PACK_LONG = Struct('>L').pack
 
+CACHED_PACK_SHORT = lru_cache(maxsize=None)(PACK_SHORT)
+
 
 class State(enum.Enum):
     init = 0
@@ -213,15 +216,15 @@ def _write_byte(self, value: int_) -> None:
 
     def _insert_short_at_start(self, value: int_) -> None:
         """Inserts an unsigned short at the start of the packet"""
-        self.data.insert(0, PACK_SHORT(value))
+        self.data.insert(0, CACHED_PACK_SHORT(value))
 
     def _replace_short(self, index: int_, value: int_) -> None:
         """Replaces an unsigned short in a certain position in the packet"""
-        self.data[index] = PACK_SHORT(value)
+        self.data[index] = CACHED_PACK_SHORT(value)
 
     def write_short(self, value: int_) -> None:
         """Writes an unsigned short to the packet"""
-        self.data.append(PACK_SHORT(value))
+        self.data.append(CACHED_PACK_SHORT(value))
         self.size += 2
 
     def _write_int(self, value: Union[float, int]) -> None:
