Переписать запрос SQLAlchemy перед отправкой в ​​движок

Question

Можно ли переписать окончательные скомпилированные необработанные байты запроса перед их отправкой в ​​базовый механизм базы данных?Я посмотрел на API событий, но, похоже, он не раскрывает такие низкоуровневые события.

Вариант использования: я хочу добавить строку в начало всех запросов, содержащую контекстные метаданные:

/* {"trace_id":"1234","client_ip":"1.2.3.4", ...} */ SELECT ...

Данные извлекаются из журналов запросов базы данных и используются для корреляции событий с другими службами в стеке.

Answer 1

Просто используйте событие before_cursor_execute с retval=True, как показано в https://github.com/sqlalchemy/sqlalchemy/wiki/Profiling

@event.listens_for(Engine, "before_cursor_execute", retval=True)
def before_cursor_execute(conn, cursor, statement,
    parameters, context, executemany):
  statement = "/* {} */ {}".format(
    json.dumps({
      "trace_id": "1234",
      "client_ip": "1.2.3.4",
    }),
    statement
  )
  return statement, parameters