darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2025-12-16 10:12:32 -05:00
Code Issues Packages Projects Releases Wiki Activity

This migration will bring all environments in line with the prod.

Browse Source 
There are a few indexes that we still need to drop from prod notification_history. Indexes on prod can take too long to run in a migration so we need to run them manually.
This commit is contained in:
Rebecca Law
2020-07-28 08:30:52 +01:00
parent 1adb460f5c
commit 180ed86a27
1 changed files with 73 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

73
migrations/versions/0327_idx_notification_history.py Normal file
View File

@@ -0,0 +1,73 @@
"""
Revision ID: 0327_idx_notification_history
Revises: 0326_broadcast_event
Create Date: 2020-07-28 08:11:07.666708
"""
import os
from alembic import op
revision = '0327_idx_notification_history'
down_revision = '0326_broadcast_event'
environment = os.environ['NOTIFY_ENVIRONMENT']
def upgrade():
if environment not in ["live", "production"]:
op.execute('DROP INDEX IF EXISTS ix_notifications_service_id_created_at')
op.execute('DROP INDEX IF EXISTS ix_notification_history_created_at')
op.execute('DROP INDEX IF EXISTS ix_notification_history_service_id_created_at')
index = """
CREATE INDEX IF NOT EXISTS ix_notification_history_service_id_composite
on notification_history(service_id, key_type, notification_type, created_at)
"""
op.execute(index)
composite_index = """
CREATE INDEX IF NOT EXISTS ix_notifications_notification_type_composite
on notifications(notification_type, notification_status, created_at)
"""
op.execute(composite_index)
# need to run on PROD
op.execute('DROP INDEX IF EXISTS ix_notification_history_service_id')
op.execute('DROP INDEX IF EXISTS ix_notification_history_template_id')
op.execute('DROP INDEX IF EXISTS ix_notification_history_notification_status')
op.execute('DROP INDEX IF EXISTS ix_notification_history_notification_type')
def downgrade():
if environment not in ["live", "production"]:
op.execute("""
CREATE INDEX IF NOT EXISTS ix_notifications_service_id_created_at
ON notifications(service_id, date(created_at))
""")
op.execute("""
CREATE INDEX IF NOT EXISTS ix_notification_history_created_at
on notification_history(created_at)
""")
op.execute("""
CREATE INDEX IF NOT EXISTS ix_notification_history_service_id_created_at
on notification_history(created_at)
""")
op.execute('DROP INDEX IF EXISTS ix_notification_history_service_id_composite')
# need to run on PROD
op.execute('DROP INDEX IF EXISTS ix_notifications_notification_type_composite')
op.execute('CREATE INDEX IF NOT EXISTS ix_notification_history_service_id on notification_history (service_id)')
op.execute("""
CREATE INDEX IF NOT EXISTS ix_notification_history_template_id on notification_history (template_id)
""")
op.execute("""
CREATE INDEX IF NOT EXISTS ix_notification_history_notification_status
on notification_history (notification_status)
""")
op.execute("""
CREATE INDEX IF NOT EXISTS ix_notification_history_notification_type
on notification_history (notification_type)
""")