habsi/authentik
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

root: include task_id in events and logs (#9749)

Browse Source 
Signed-off-by: Jens Langhammer <jens@goauthentik.io>
This commit is contained in:
Jens L
2024-05-16 13:33:00 +02:00
committed by GitHub
parent 9589e04072
commit f28209548b
1 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
authentik/root/celery.py
View File

@ -63,7 +63,7 @@ def task_prerun_hook(task_id: str, task, *args, **kwargs):
@task_postrun.connect @task_postrun.connect
def task_postrun_hook(task_id, task, *args, retval=None, state=None, **kwargs): def task_postrun_hook(task_id: str, task, *args, retval=None, state=None, **kwargs):
"""Log task_id on worker""" """Log task_id on worker"""
CTX_TASK_ID.set(...) CTX_TASK_ID.set(...)
LOGGER.info( LOGGER.info(
@ -73,14 +73,16 @@ def task_postrun_hook(task_id, task, *args, retval=None, state=None, **kwargs):
@task_failure.connect @task_failure.connect
@task_internal_error.connect @task_internal_error.connect
def task_error_hook(task_id, exception: Exception, traceback, *args, **kwargs): def task_error_hook(task_id: str, exception: Exception, traceback, *args, **kwargs):
"""Create system event for failed task""" """Create system event for failed task"""
from authentik.events.models import Event, EventAction from authentik.events.models import Event, EventAction
LOGGER.warning("Task failure", exc=exception) LOGGER.warning("Task failure", task_id=task_id.replace("-", ""), exc=exception)
CTX_TASK_ID.set(...) CTX_TASK_ID.set(...)
if before_send({}, {"exc_info": (None, exception, None)}) is not None: if before_send({}, {"exc_info": (None, exception, None)}) is not None:
Event.new(EventAction.SYSTEM_EXCEPTION, message=exception_to_string(exception)).save() Event.new(
EventAction.SYSTEM_EXCEPTION, message=exception_to_string(exception), task_id=task_id
).save()
def _get_startup_tasks_default_tenant() -> list[Callable]: def _get_startup_tasks_default_tenant() -> list[Callable]: