mirror of
https://github.com/GSA/notifications-api.git
synced 2025-12-12 00:02:36 -05:00
72 lines
3.3 KiB
Python
72 lines
3.3 KiB
Python
"""
|
|
|
|
Revision ID: 0320_optimise_notifications
|
|
Revises: 0319_contact_list_archived
|
|
Create Date: 2020-03-26 11:16:12.389524
|
|
|
|
"""
|
|
import os
|
|
|
|
from alembic import op
|
|
|
|
revision = '0320_optimise_notifications'
|
|
down_revision = '0319_contact_list_archived'
|
|
environment = os.environ['NOTIFY_ENVIRONMENT']
|
|
|
|
|
|
def upgrade():
|
|
# We like to run this operation on live via the command prompt, to watch the progress and stop if necessary
|
|
if environment not in ["live", "production"]:
|
|
# Drop indexes notifications - no need to recreate these indexes
|
|
op.execute('DROP INDEX IF EXISTS ix_notifications_key_type')
|
|
op.execute('DROP INDEX IF EXISTS ix_notifications_api_key_id')
|
|
op.execute('DROP INDEX IF EXISTS ix_notifications_notification_status')
|
|
op.execute('DROP INDEX IF EXISTS ix_notifications_notification_type')
|
|
op.execute('DROP INDEX IF EXISTS ix_notifications_service_id')
|
|
|
|
# Create new composite indexes
|
|
# PLEASE NOTE: that if you create index on production you need to add concurrently to the create statement,
|
|
# however we are unable to do that inside a transaction like this upgrade method
|
|
ix_notifications_service_id_composite = """
|
|
CREATE INDEX ix_notifications_service_id_composite
|
|
on notifications (service_id, notification_type, notification_status, created_at)
|
|
"""
|
|
op.execute(ix_notifications_service_id_composite)
|
|
|
|
ix_notifications_notification_type_composite = """
|
|
CREATE INDEX ix_notifications_notification_type_composite
|
|
on notifications (notification_type, notification_status, created_at)
|
|
"""
|
|
op.execute(ix_notifications_notification_type_composite)
|
|
# DROP and CREATE all other indexes
|
|
op.execute('DROP INDEX IF EXISTS ix_notifications_client_reference')
|
|
op.execute('CREATE INDEX ix_notifications_client_reference ON notifications (client_reference)')
|
|
|
|
op.execute('DROP INDEX IF EXISTS ix_notifications_created_at')
|
|
op.execute('CREATE INDEX ix_notifications_created_at ON notifications (created_at)')
|
|
|
|
op.execute('DROP INDEX IF EXISTS ix_notifications_job_id')
|
|
op.execute('CREATE INDEX ix_notifications_job_id ON notifications (job_id)')
|
|
|
|
op.execute('DROP INDEX IF EXISTS ix_notifications_reference')
|
|
op.execute('CREATE INDEX ix_notifications_reference ON notifications (reference)')
|
|
|
|
op.execute('DROP INDEX IF EXISTS ix_notifications_service_created_at')
|
|
op.execute(
|
|
'CREATE INDEX ix_notifications_service_created_at ON notifications (service_id, created_at)')
|
|
|
|
op.execute('DROP INDEX IF EXISTS ix_notifications_template_id')
|
|
op.execute('CREATE INDEX ix_notifications_template_id ON notifications (template_id)')
|
|
|
|
# Drop indexes notification_history
|
|
op.execute('DROP INDEX IF EXISTS ix_notification_history_key_type')
|
|
op.execute('DROP INDEX IF EXISTS ix_notification_history_api_key_id')
|
|
op.execute('DROP INDEX IF EXISTS ix_notification_history_notification_status')
|
|
op.execute('DROP INDEX IF EXISTS ix_notification_history_notification_type')
|
|
op.execute('DROP INDEX IF EXISTS ix_notification_history_service_id')
|
|
op.execute('DROP INDEX IF EXISTS ix_notification_history_week_created')
|
|
|
|
|
|
def downgrade():
|
|
pass
|