feat(eap): Add a column to store downsampled retention days (#7318)

Author: phacops

snuba/snuba_migrations/events_analytics_platform/0046_add_downsampled_retention_days.py

@@ -0,0 +1,68 @@
+from snuba.clickhouse.columns import 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
+
+storage_set = StorageSetKey.EVENTS_ANALYTICS_PLATFORM
+table_name_prefix = "eap_items_1"
+new_column_name = "downsampled_retention_days"
+after = "retention_days"
+
+
+class Migration(migration.ClickhouseNodeMigration):
+    blocking = False
+
+    def forwards_ops(self) -> list[operations.SqlOperation]:
+        return [
+            operations.AddColumn(
+                storage_set=storage_set,
+                table_name=f"{table_name_prefix}_local",
+                column=Column(
+                    new_column_name,
+                    UInt(
+                        16,
+                        modifiers=Modifiers(
+                            codecs=[
+                                "ZSTD(1)",
+                            ]
+                        ),
+                    ),
+                ),
+                after=after,
+                target=OperationTarget.LOCAL,
+            ),
+            operations.AddColumn(
+                storage_set=storage_set,
+                table_name=f"{table_name_prefix}_dist",
+                column=Column(
+                    new_column_name,
+                    UInt(
+                        16,
+                        modifiers=Modifiers(
+                            codecs=[
+                                "ZSTD(1)",
+                            ]
+                        ),
+                    ),
+                ),
+                after=after,
+                target=OperationTarget.DISTRIBUTED,
+            ),
+        ]
+
+    def backwards_ops(self) -> list[operations.SqlOperation]:
+        return [
+            operations.DropColumn(
+                storage_set=storage_set,
+                table_name=f"{table_name_prefix}_dist",
+                column_name=new_column_name,
+                target=OperationTarget.DISTRIBUTED,
+            ),
+            operations.DropColumn(
+                storage_set=storage_set,
+                table_name=f"{table_name_prefix}_local",
+                column_name=new_column_name,
+                target=OperationTarget.LOCAL,
+            ),
+        ]