diff --git a/snuba/snuba_migrations/events_analytics_platform/0002_eap_spans_project_id_index.py b/snuba/snuba_migrations/events_analytics_platform/0002_eap_spans_project_id_index.py new file mode 100644 index 0000000000..5eb892c269 --- /dev/null +++ b/snuba/snuba_migrations/events_analytics_platform/0002_eap_spans_project_id_index.py @@ -0,0 +1,31 @@ +from typing import Sequence + +from snuba.clusters.storage_sets import StorageSetKey +from snuba.migrations import migration, operations + + +class Migration(migration.ClickhouseNodeMigration): + blocking = False + + def forwards_ops(self) -> Sequence[operations.SqlOperation]: + return [ + operations.AddIndex( + storage_set=StorageSetKey.EVENTS_ANALYTICS_PLATFORM, + table_name="eap_spans_local", + index_name="bf_project_id", + index_expression="project_id", + index_type="bloom_filter", + granularity=1, + target=operations.OperationTarget.LOCAL, + ), + ] + + def backwards_ops(self) -> Sequence[operations.SqlOperation]: + return [ + operations.DropIndex( + storage_set=StorageSetKey.EVENTS_ANALYTICS_PLATFORM, + table_name="eap_spans_local", + index_name="bf_project_id", + target=operations.OperationTarget.LOCAL, + ), + ]