discours.io/core
5
0
Fork 0
Code Pull Requests Actions Projects Activity

logger-query-id
All checks were successful
Deploy to core / deploy (push) Successful in 1m13s
Details

Browse Source
This commit is contained in:
Untone 2024-02-21 13:44:36 +03:00
parent 4ca884f257
commit 9118ae9268
1 changed files with 12 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
services/db.py
View File

@ -25,15 +25,21 @@ Base = declarative_base()
# Перехватчики для журнала запросов SQLAlchemy # Перехватчики для журнала запросов SQLAlchemy
@event.listens_for(Engine, 'before_cursor_execute') @event.listens_for(Engine, 'before_cursor_execute')
def before_cursor_execute(conn, cursor, statement, parameters, context, executemany): def before_cursor_execute(conn, cursor, statement, parameters, context, executemany):
conn.info.setdefault(f'query_start_time:{statement}', time.time()) query_id = id(cursor) # Уникальный идентификатор запроса
conn.info.setdefault(f'query_start_time:{query_id}', time.time())
@event.listens_for(Engine, 'after_cursor_execute') @event.listens_for(Engine, 'after_cursor_execute')
def after_cursor_execute(conn, cursor, statement, parameters, context, executemany): def after_cursor_execute(conn, cursor, statement, parameters, context, executemany):
total = time.time() - conn.info[f'query_start_time:{statement}'] query_id = id(cursor) # Уникальный идентификатор запроса
del conn.info[f'query_start_time:{statement}'] if f'query_start_time:{query_id}' in conn.info:
stars = '*' * math.floor(total*1000) total = time.time() - conn.info.get(f'query_start_time:{query_id}', time.time())
if stars: del conn.info[f'query_start_time:{query_id}']
logger.debug(f'\n{statement}\n {stars} {total*1000} s\n') stars = '*' * math.floor(total*1000)
if stars:
logger.debug(f'\n{statement}\n {stars} {total*1000} s\n')
else:
logger.error(f"Не удалось найти информацию о времени начала запроса с идентификатором {query_id}.")
def local_session(src=""): def local_session(src=""):