openstack/taskflow
Code Issues Proposed changes

Merge "db: Don't rely on branched connections"

Browse Source
This commit is contained in:
Zuul 2024-04-15 17:09:44 +00:00 committed by Gerrit Code Review
commit 3ca2d4fdc8
1 changed files with 9 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
taskflow/persistence/backends/sqlalchemy/alembic/env.py
View File

@ -63,13 +63,18 @@ def run_migrations_online():
connectable = engine_from_config( connectable = engine_from_config(
config.get_section(config.config_ini_section), config.get_section(config.config_ini_section),
prefix='sqlalchemy.', poolclass=pool.NullPool) prefix='sqlalchemy.', poolclass=pool.NullPool)
with connectable.connect() as connection: with connectable.connect() as connection:
context.configure(connection=connection, context.configure(connection=connection,
target_metadata=target_metadata) target_metadata=target_metadata)
with context.begin_transaction():
context.run_migrations()
else:
context.configure(
connection=connectable,
target_metadata=target_metadata)
with context.begin_transaction(): with context.begin_transaction():
context.run_migrations() context.run_migrations()
if context.is_offline_mode(): if context.is_offline_mode():
run_migrations_offline() run_migrations_offline()
else: else: