Opentelemetry tweaks

pyproject.toml

@@ -45,7 +45,9 @@ optional = true
 [tool.poetry.group.http.dependencies]
 fastapi = ">=0.99.0"
 jinja2 = "^3.1.2"
-opentelemetry-instrumentation-fastapi = "*"
+# We use the generic ASGI instrumentation, so that if we decide to change
+# Framework it will still work consistently.
+opentelemetry-instrumentation-asgi = "*"
 starlette-prometheus = "^0.10.0"
 strawberry-graphql = { version = ">=0.204.0", extras = ["debug-server"] }
 uvicorn = { version = "^0.31.0", extras = ["standard"] }

src/celery_worker/__init__.py

@@ -4,6 +4,15 @@
 without having to initialise the HTTP framework (or other ones)
 """
 
+from celery.signals import worker_process_init
+from opentelemetry.instrumentation.celery import CeleryInstrumentor
+
 from bootstrap import AppConfig, application_init
 
+
+@worker_process_init.connect(weak=False)
+def init_celery_tracing(*args, **kwargs):
+    CeleryInstrumentor().instrument()  # pragma: nocover
+
+
 app = application_init(AppConfig()).celery_app