feat(replays): add replay_id column to merged discover table (#5777)

* feat(replays): add replay_id column to merged discover table

* dont use legacy migration

snuba/migrations/group_loader.py

@@ -136,6 +136,7 @@ def get_migrations(self) -> Sequence[str]:
             "0005_discover_fix_transaction_name",
             "0006_discover_add_trace_id",
             "0007_discover_add_span_id",
+            "0008_discover_add_replay_id",
         ]
 
 

snuba/snuba_migrations/discover/0008_discover_add_replay_id.py

@@ -0,0 +1,56 @@
+from typing import Sequence
+
+from snuba.clickhouse.columns import UUID, Column
+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
+
+
+class Migration(migration.ClickhouseNodeMigration):
+    """
+    Add replay_id to merge table
+    """
+
+    blocking = False
+
+    def forwards_ops(self) -> Sequence[operations.SqlOperation]:
+        return [
+            operations.AddColumn(
+                storage_set=StorageSetKey.DISCOVER,
+                table_name=table_name,
+                column=Column("replay_id", UUID(Modifiers(nullable=True))),
+                after="span_id",
+                target=target,
+            )
+            for table_name, target in [
+                ("discover_local", OperationTarget.LOCAL),
+                ("discover_dist", OperationTarget.DISTRIBUTED),
+            ]
+        ]
+
+    def backwards_ops(self) -> Sequence[operations.SqlOperation]:
+        return [
+            operations.DropColumn(
+                storage_set=StorageSetKey.DISCOVER,
+                table_name=table_name,
+                column_name="replay_id",
+                target=target,
+            )
+            for table_name, target in [
+                ("discover_dist", OperationTarget.DISTRIBUTED),
+                ("discover_local", OperationTarget.LOCAL),
+            ]
+        ]
+
+    def forwards_local(self) -> Sequence[operations.SqlOperation]:
+        return self.forwards_ops()
+
+    def backwards_local(self) -> Sequence[operations.SqlOperation]:
+        return self.backwards_ops()
+
+    def forwards_dist(self) -> Sequence[operations.SqlOperation]:
+        return self.forwards_ops()
+
+    def backwards_dist(self) -> Sequence[operations.SqlOperation]:
+        return self.backwards_ops()