feat: Allow to use 128 bits integers (#6862)

Author: phacops

snuba/migrations/parse_schema.py

@@ -38,7 +38,7 @@
     basic_type       = "DateTime" / "Date" / "IPv4" / "IPv6" / "String" / "UUID"
     uint             = "UInt" uint_size
     int              = "Int" uint_size
-    uint_size        = "8" / "16" / "32" / "64"
+    uint_size        = "8" / "16" / "32" / "64" / "128"
     float            = "Float" float_size
     float_size       = "32" / "64"
     fixedstring      = "FixedString" open_paren space* fixedstring_size space* close_paren

snuba/utils/schemas.py

@@ -175,7 +175,9 @@ def for_schema(self) -> str:
 
     @staticmethod
     def to_columns(
-        columns: Sequence[Union[Column[TModifiers], tuple[str, ColumnType[TModifiers]]]]
+        columns: Sequence[
+            Union[Column[TModifiers], tuple[str, ColumnType[TModifiers]]]
+        ],
     ) -> Sequence[Column[TModifiers]]:
         return [Column(*col) if not isinstance(col, Column) else col for col in columns]
 
@@ -582,7 +584,7 @@ def get_raw(self) -> FixedString[TModifiers]:
 class UInt(ColumnType[TModifiers]):
     def __init__(self, size: int, modifiers: Optional[TModifiers] = None) -> None:
         super().__init__(modifiers)
-        assert size in (8, 16, 32, 64)
+        assert size in (8, 16, 32, 64, 128)
         self.size = size
 
     def _repr_content(self) -> str:
@@ -608,7 +610,7 @@ def get_raw(self) -> UInt[TModifiers]:
 class Int(ColumnType[TModifiers]):
     def __init__(self, size: int, modifiers: Optional[TModifiers] = None) -> None:
         super().__init__(modifiers)
-        assert size in (8, 16, 32, 64)
+        assert size in (8, 16, 32, 64, 128)
         self.size = size
 
     def _repr_content(self) -> str:

tests/migrations/test_parse_schema.py

@@ -36,6 +36,7 @@
     (("IPv6", "", "", ""), IPv6()),
     (("String", "", "", ""), String()),
     (("UInt32", "", "", ""), UInt(32)),
+    (("UInt128", "", "", ""), UInt(128)),
     (("UUID", "", "", ""), UUID()),
     # Aggregate functions
     (