fix(group-attributes): Modify the column type for `group_first_releas…

…e_id` to bigint (#6040)

* Modify the column type to bigint

* Fix types

* Add migration to group_loader

* Add OperationTarget

snuba/migrations/group_loader.py

@@ -400,6 +400,7 @@ def get_migrations(self) -> Sequence[str]:
             "0001_group_attributes",
             "0002_add_priority_to_group_attributes",
             "0003_add_first_release_id_to_group_attributes",
+            "0004_fix_group_first_release_id_type",
         ]
 
 

snuba/snuba_migrations/group_attributes/0004_fix_group_first_release_id_type.py

@@ -0,0 +1,47 @@
+from typing import Sequence
+
+from snuba.clickhouse.columns import UUID, Column, UInt
+from snuba.clusters.storage_sets import StorageSetKey
+from snuba.migrations import migration, operations
+from snuba.migrations.columns import MigrationModifiers as Modifiers
+from snuba.migrations.operations import OperationTarget, SqlOperation
+
+
+class Migration(migration.ClickhouseNodeMigration):
+    blocking = False
+
+    def forwards_ops(self) -> Sequence[SqlOperation]:
+        return [
+            operations.ModifyColumn(
+                storage_set=StorageSetKey.GROUP_ATTRIBUTES,
+                table_name="group_attributes_local",
+                column=Column(
+                    "group_first_release_id", UInt(64, Modifiers(nullable=True))
+                ),
+                target=OperationTarget.LOCAL,
+            ),
+            operations.ModifyColumn(
+                storage_set=StorageSetKey.GROUP_ATTRIBUTES,
+                table_name="group_attributes_dist",
+                column=Column(
+                    "group_first_release_id", UInt(64, Modifiers(nullable=True))
+                ),
+                target=OperationTarget.DISTRIBUTED,
+            ),
+        ]
+
+    def backwards_ops(self) -> Sequence[SqlOperation]:
+        return [
+            operations.ModifyColumn(
+                storage_set=StorageSetKey.GROUP_ATTRIBUTES,
+                table_name="group_attributes_dist",
+                column=Column("group_first_release_id", UUID(Modifiers(nullable=True))),
+                target=OperationTarget.DISTRIBUTED,
+            ),
+            operations.ModifyColumn(
+                storage_set=StorageSetKey.GROUP_ATTRIBUTES,
+                table_name="group_attributes_local",
+                column=Column("group_first_release_id", UUID(Modifiers(nullable=True))),
+                target=OperationTarget.LOCAL,
+            ),
+        ]