如何使用Alembic创建PostgreSQL并发索引？

问题核心在于CREATE INDEX CONCURRENTLY不允许在事务块内执行，而Alembic默认会将整个迁移脚本包裹在事务中。以下是两种可行的解决方式：

方法1：原生SQL配合自动提交块 #

直接通过op.execute()执行原生SQL，并用Alembic的autocommit_block()上下文管理器包裹，让SQL在无事务的环境下运行：

from alembic import op

def upgrade():
    # 启用自动提交模式，规避事务包裹
    with op.get_context().autocommit_block():
        op.execute("CREATE INDEX CONCURRENTLY idx_mytable_mycol ON mytable (mycol);")

def downgrade():
    # 删除并发创建的索引时，同样需要CONCURRENTLY并在自动提交块中执行
    with op.get_context().autocommit_block():
        op.execute("DROP INDEX CONCURRENTLY IF EXISTS idx_mytable_mycol;")

方法2：使用SQLAlchemy封装API #

如果偏好使用SQLAlchemy的封装方法而非原生SQL，可通过postgresql_concurrently=True参数启用并发创建，同时配合autocommit_block()：

from alembic import op
import sqlalchemy as sa

def upgrade():
    with op.get_context().autocommit_block():
        op.create_index(
            index_name='idx_mytable_mycol',
            table_name='mytable',
            columns=['mycol'],
            postgresql_concurrently=True
        )

def downgrade():
    with op.get_context().autocommit_block():
        op.drop_index(
            index_name='idx_mytable_mycol',
            table_name='mytable',
            postgresql_concurrently=True
        )

注意事项 #

• 并发创建索引速度慢于普通创建，但不会对大表施加长时间写锁，适合业务低峰期或需避免阻塞的场景。
• 若创建过程中断，索引会处于invalid状态，需手动删除后重新创建。
• 不要在同一个autocommit_block()中执行多个CONCURRENTLY操作，建议分开执行降低风险。
• 确保执行迁移的数据库用户拥有创建/删除索引的权限。

内容的提问来源于stack exchange，提问作者Natim