apunkt/SurfSense
1
0
Fork 0
mirror of https://github.com/MODSetter/SurfSense.git synced 2026-05-25 19:15:18 +02:00
Code Issues Projects Releases Packages Wiki Activity

feat(connectors): add connector sync spans

Browse source
This commit is contained in:
Anish Sarkar 2026-05-22 13:49:59 +05:30
parent 4e3a6dff46
commit 7c07c220fc
1 changed files with 7 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

9
surfsense_backend/app/tasks/celery_tasks/connector_tasks.py
View file

@ -8,7 +8,7 @@ from collections.abc import Awaitable, Callable
from celery import current_task from celery import current_task
from app.celery_app import celery_app from app.celery_app import celery_app
from app.observability import metrics as ot_metrics from app.observability import metrics as ot_metrics, otel as ot
from app.tasks.celery_tasks import ( from app.tasks.celery_tasks import (
get_celery_session_maker, get_celery_session_maker,
run_async_celery_task as _run_async_celery_task, run_async_celery_task as _run_async_celery_task,
@ -23,9 +23,14 @@ def run_async_celery_task[T](coro_factory: Callable[[], Awaitable[T]]) -> T:
t0 = time.perf_counter() t0 = time.perf_counter()
status = "failed" status = "failed"
try: try:
result = _run_async_celery_task(coro_factory) with ot.connector_sync_span(connector_type=task_name) as sp:
result = _run_async_celery_task(coro_factory)
sp.set_attribute("connector.status", "success")
status = "success" status = "success"
return result return result
except Exception:
status = "failed"
raise
finally: finally:
elapsed_s = time.perf_counter() - t0 elapsed_s = time.perf_counter() - t0
ot_metrics.record_connector_sync_duration( ot_metrics.record_connector_sync_duration(