Skip to content

Commit

Permalink
fix(migrations): correct parent_message_id for service-api records
Browse files Browse the repository at this point in the history
- Set default parent_message_id to UUID_NIL when it's null for service-api entries post v0.9.0 release.
- Provide downgrade path to reset parent_message_id back to null in case of rollback.
  • Loading branch information
laipz8200 committed Oct 9, 2024
1 parent b279d19 commit ee3b228
Showing 1 changed file with 48 additions and 0 deletions.
Original file line number Diff line number Diff line change
@@ -0,0 +1,48 @@
"""fix wrong service-api history
Revision ID: d8e744d88ed6
Revises: 33f5fac87f29
Create Date: 2024-10-09 13:29:23.548498
"""
from alembic import op
from constants import UUID_NIL
import models as models
import sqlalchemy as sa


# revision identifiers, used by Alembic.
revision = 'd8e744d88ed6'
down_revision = '33f5fac87f29'
branch_labels = None
depends_on = None

# (UTC) release date of v0.9.0
v0_9_0_release_date= '2024-09-29 12:00:00'

def upgrade():
# ### commands auto generated by Alembic - please adjust! ###
sql = f"""UPDATE
public.messages
SET
parent_message_id = '{UUID_NIL}'
WHERE
invoke_from = 'service-api'
AND parent_message_id IS NULL
AND created_at >= '{v0_9_0_release_date}';"""
op.execute(sql)
# ### end Alembic commands ###


def downgrade():
# ### commands auto generated by Alembic - please adjust! ###
sql = f"""UPDATE
public.messages
SET
parent_message_id = NULL
WHERE
invoke_from = 'service-api'
AND parent_message_id = '{UUID_NIL}'
AND created_at >= '{v0_9_0_release_date}';"""
op.execute(sql)
# ### end Alembic commands ###

0 comments on commit ee3b228

Please sign in to comment.