From 3eb14b53f11ac550fb2e7f99e45eeb58ee75a2a9 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Wed, 17 Jan 2024 07:34:18 -0300 Subject: [PATCH] Add indexes to tables apikey, flow, and user --- .../alembic/versions/0b8757876a7c_.py | 52 +++++++++++++++++++ 1 file changed, 52 insertions(+) create mode 100644 src/backend/langflow/alembic/versions/0b8757876a7c_.py diff --git a/src/backend/langflow/alembic/versions/0b8757876a7c_.py b/src/backend/langflow/alembic/versions/0b8757876a7c_.py new file mode 100644 index 000000000..6d6c22575 --- /dev/null +++ b/src/backend/langflow/alembic/versions/0b8757876a7c_.py @@ -0,0 +1,52 @@ +"""empty message + +Revision ID: 0b8757876a7c +Revises: 006b3990db50 +Create Date: 2024-01-17 10:32:56.686287 + +""" +from typing import Sequence, Union + +from alembic import op + +# revision identifiers, used by Alembic. +revision: str = '0b8757876a7c' +down_revision: Union[str, None] = '006b3990db50' +branch_labels: Union[str, Sequence[str], None] = None +depends_on: Union[str, Sequence[str], None] = None + + +def upgrade() -> None: + # ### commands auto generated by Alembic - please adjust! ### + with op.batch_alter_table('apikey', schema=None) as batch_op: + batch_op.create_index(batch_op.f('ix_apikey_api_key'), ['api_key'], unique=True) + batch_op.create_index(batch_op.f('ix_apikey_name'), ['name'], unique=False) + batch_op.create_index(batch_op.f('ix_apikey_user_id'), ['user_id'], unique=False) + + with op.batch_alter_table('flow', schema=None) as batch_op: + batch_op.create_index(batch_op.f('ix_flow_description'), ['description'], unique=False) + batch_op.create_index(batch_op.f('ix_flow_name'), ['name'], unique=False) + batch_op.create_index(batch_op.f('ix_flow_user_id'), ['user_id'], unique=False) + + with op.batch_alter_table('user', schema=None) as batch_op: + batch_op.create_index(batch_op.f('ix_user_username'), ['username'], unique=True) + + # ### end Alembic commands ### + + +def downgrade() -> None: + # ### commands auto generated by Alembic - please adjust! ### + with op.batch_alter_table('user', schema=None) as batch_op: + batch_op.drop_index(batch_op.f('ix_user_username')) + + with op.batch_alter_table('flow', schema=None) as batch_op: + batch_op.drop_index(batch_op.f('ix_flow_user_id')) + batch_op.drop_index(batch_op.f('ix_flow_name')) + batch_op.drop_index(batch_op.f('ix_flow_description')) + + with op.batch_alter_table('apikey', schema=None) as batch_op: + batch_op.drop_index(batch_op.f('ix_apikey_user_id')) + batch_op.drop_index(batch_op.f('ix_apikey_name')) + batch_op.drop_index(batch_op.f('ix_apikey_api_key')) + + # ### end Alembic commands ### \ No newline at end of file