openstack/deb-python-taskflow
Code Issues Proposed changes

Merge "Make it possible to see the queries executed (in BLATHER mode)"

Browse Source
This commit is contained in:
Jenkins
2015-07-08 23:41:55 +00:00
committed by Gerrit Code Review
parent b77eac140f 9604703c59
commit f42b9d49e4
1 changed files with 13 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
taskflow/persistence/backends/impl_sqlalchemy.py
View File

@@ -108,6 +108,12 @@ DEFAULT_TXN_ISOLATION_LEVELS = {
} }
def _log_statements(log_level, conn, cursor, statement, parameters, *args):
if LOG.isEnabledFor(log_level):
LOG.log(log_level, "Running statement '%s' with parameters %s",
statement, parameters)
def _in_any(reason, err_haystack): def _in_any(reason, err_haystack):
"""Checks if any elements of the haystack are in the given reason.""" """Checks if any elements of the haystack are in the given reason."""
for err in err_haystack: for err in err_haystack:
@@ -290,6 +296,13 @@ class SQLAlchemyBackend(base.Backend):
# or engine arg overrides make sure we merge them in. # or engine arg overrides make sure we merge them in.
engine_args.update(conf.pop('engine_args', {})) engine_args.update(conf.pop('engine_args', {}))
engine = sa.create_engine(sql_connection, **engine_args) engine = sa.create_engine(sql_connection, **engine_args)
log_statements = conf.pop('log_statements', False)
if _as_bool(log_statements):
log_statements_level = conf.pop("log_statements_level",
logging.BLATHER)
sa.event.listen(engine, "before_cursor_execute",
functools.partial(_log_statements,
log_statements_level))
checkin_yield = conf.pop('checkin_yield', checkin_yield = conf.pop('checkin_yield',
eventlet_utils.EVENTLET_AVAILABLE) eventlet_utils.EVENTLET_AVAILABLE)
if _as_bool(checkin_yield): if _as_bool(checkin_yield):