Add GraphQL OpenTelemetry integration

Author: shangyian

datajunction-server/datajunction_server/api/graphql/main.py

@@ -5,6 +5,7 @@
 
 import strawberry
 from fastapi import Depends
+from strawberry.extensions.tracing import OpenTelemetryExtension
 from strawberry.fastapi import GraphQLRouter
 from strawberry.types import Info
 from datajunction_server.api.graphql.queries.catalogs import list_catalogs
@@ -128,6 +129,11 @@ class Query:
     )
 
 
-schema = strawberry.Schema(query=Query)
+schema = strawberry.Schema(
+    query=Query,
+    extensions=[
+        OpenTelemetryExtension,
+    ],
+)
 
 graphql_app = GraphQLRouter(schema, context_getter=get_context)

datajunction-server/datajunction_server/api/main.py

@@ -2,6 +2,7 @@
 Main DJ server app.
 """
 
+from contextlib import asynccontextmanager
 import logging
 from datajunction_server.api import setup_logging  # noqa
 
@@ -54,6 +55,17 @@
 
 dependencies = [Depends(default_attribute_types), Depends(default_catalog)]
 
+
+@asynccontextmanager
+async def lifespan(app: FastAPI):
+    """
+    Lifespan context for initializing and tearing down app-wide resources, like the FastAPI cache
+    """
+    FastAPICache.init(InMemoryBackend(), prefix="inmemory-cache")  # pragma: no cover
+
+    yield
+
+
 app = FastAPI(
     title=settings.name,
     description=settings.description,
@@ -63,6 +75,7 @@
         "url": "https://mit-license.org/",
     },
     dependencies=dependencies,
+    lifespan=lifespan,
 )
 
 app.add_middleware(
@@ -98,14 +111,6 @@
 app.include_router(notifications.router)
 
 
-@app.on_event("startup")
-async def startup():
-    """
-    Initialize FastAPI cache when the server starts up
-    """
-    FastAPICache.init(InMemoryBackend(), prefix="inmemory-cache")  # pragma: no cover
-
-
 @app.exception_handler(DJException)
 async def dj_exception_handler(
     request: Request,