mirror of
https://github.com/GSA/notifications-api.git
synced 2025-12-17 10:42:25 -05:00
7489cda6f0
Dropping unnecessary indexes on Notification and Notication_History Create new composite indexes that should give queries better performance Drop and create new indexes. We are not running this migration on production because we want to run each one at a time. Run this on staging then start run some load tests
80 lines
3.7 KiB
Python
80 lines
3.7 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_service_id_created_at')
|
|
op.execute(
|
|
'CREATE INDEX ix_notifications_service_id_created_at ON NOTIFICATIONS (service_id, created_at)'
|
|
)
|
|
|
|
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_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
|