[api] Select encode map before lookup

2026-09-01 10:36:01 +00:00 · 2026-04-03 12:31:01 -10:00
parent 02b424388e
commit f3238a857e
1 changed files with 4 additions and 3 deletions
@@ -259,11 +259,12 @@ class TypeInfo(ABC):
        if tag >= 128:
            return None
        max_val = self.max_value
-        raw_expr = (
-            self.RAW_ENCODE_SMALL_MAP.get(self.encode_func)
+        encode_map = (
+            self.RAW_ENCODE_SMALL_MAP
            if max_val is not None and max_val < 128
-            else self.RAW_ENCODE_MAP.get(self.encode_func)
+            else self.RAW_ENCODE_MAP
        )
+        raw_expr = encode_map.get(self.encode_func)
        if raw_expr is None:
            return None
        return f"buffer.write_raw_byte({tag});\n{raw_expr.format(value=value_expr)}"