Adding migrations for JIT liquidity tables

This commit is contained in:
Eli Barbieri 2022-05-10 12:43:06 -07:00
parent 3c9fe3dca9
commit add06d11dd
2 changed files with 77 additions and 0 deletions

View File

@ -0,0 +1,44 @@
"""add_jit_liquidity_table
Revision ID: a46974a623b3
Revises: 5c5375de15fd
Create Date: 2022-05-10 12:36:57.139209
"""
from alembic import op
import sqlalchemy as sa
# revision identifiers, used by Alembic.
revision = 'a46974a623b3'
down_revision = '5c5375de15fd'
branch_labels = None
depends_on = None
def upgrade():
op.create_table(
"jit_liquidity",
sa.Column("id", sa.String, primary_key=True),
sa.Column("block_number", sa.Numeric(), nullable=False),
sa.Column("bot_address", sa.String(42), nullable=True),
sa.Column("pool_address", sa.String(42), nullable=False),
sa.Column("token0_address", sa.String(42), nullable=True),
sa.Column("token1_address", sa.String(42), nullable=True),
sa.Column("mint_transaction_hash", sa.String(66), nullable=False),
sa.Column("mint_transaction_trace", sa.ARRAY(sa.Integer)),
sa.Column("burn_transaction_hash", sa.String(66), nullable=False),
sa.Column("burn_transaction_trace", sa.ARRAY(sa.Integer)),
sa.Column("mint_token0_amount", sa.Numeric),
sa.Column("mint_token1_amount", sa.Numeric),
sa.Column("burn_token0_amount", sa.Numeric),
sa.Column("burn_token1_amount", sa.Numeric),
sa.Column("token0_swap_volume", sa.Numeric),
sa.Column("token1_swap_volume", sa.Numeric),
)
op.create_index("ix_jit_liquidity_block_number", "jit_liquidity", ["block_number"])
def downgrade():
op.drop_index("ix_jit_liquidity_block_number")
op.drop_table("jit_liquidity")

View File

@ -0,0 +1,33 @@
"""add_jit_liquidity_swaps_join_table
Revision ID: c77f5db6105e
Revises: a46974a623b3
Create Date: 2022-05-10 12:37:25.275799
"""
from alembic import op
import sqlalchemy as sa
# revision identifiers, used by Alembic.
revision = 'c77f5db6105e'
down_revision = 'a46974a623b3'
branch_labels = None
depends_on = None
def upgrade():
op.create_table(
"jit_liquidity_swaps",
sa.Column("created_at", sa.TIMESTAMP, server_default=sa.func.now()),
sa.Column("jit_liquidity_id", sa.String(1024), primary_key=True),
sa.Column("swap_transaction_hash", sa.String(66), primary_key=True),
sa.Column("swap_trace_address", sa.ARRAY(sa.Integer), primary_key=True),
sa.ForeignKeyConstraint(
["jit_liquidity_id"], ["jit_liquidity.id"], ondelete="CASCADE"
),
)
def downgrade():
op.drop_table("jit_liquidity_swaps")