Update startup activities

Author: marvinbuss

code/function/fastapp/api/v1/endpoints/sample.py

@@ -23,9 +23,9 @@ async def post_predict(
         response = await client.get("https://www.bing.com")
     logger.info(f"Received response status code: {response.status_code}")
 
-    with httpx.Client() as client:
-        response = client.get("https://www.google.com")
-    response = httpx.get("https://www.google.de")
+    # with httpx.Client() as client:
+    #     response = client.get("https://www.google.com")
+    # response = httpx.get("https://www.google.de")
 
     # tracer_attributes = {"http.client_ip": x_forwarded_for}
     # with tracer.start_as_current_span(

code/function/fastapp/main.py

@@ -1,34 +1,34 @@
+from contextlib import asynccontextmanager
+
 from fastapi import FastAPI
 from fastapp.api.v1.api_v1 import api_v1_router
 from fastapp.core.config import settings
 from fastapp.utils import setup_opentelemetry
 
 
-def get_app() -> FastAPI:
+def get_app(lifespan) -> FastAPI:
     """Setup the Fast API server.
 
     RETURNS (FastAPI): The FastAPI object to start the server.
     """
     app = FastAPI(
         title=settings.PROJECT_NAME,
+        description="",
         version=settings.APP_VERSION,
         openapi_url="/openapi.json",
         debug=settings.DEBUG,
+        lifespan=lifespan,
     )
     app.include_router(api_v1_router, prefix=settings.API_V1_STR)
     return app
 
 
-app = get_app()
-
-
-@app.on_event("startup")
-async def startup_event():
+@asynccontextmanager
+async def lifespan(app: FastAPI) -> None:
     """Gracefully start the application before the server reports readiness."""
     setup_opentelemetry(app=app)
+    yield
+    pass
 
 
-@app.on_event("shutdown")
-async def shutdown_event():
-    """Gracefully close connections before shutdown of the server."""
-    pass
+app = get_app(lifespan=lifespan)

code/function/wrapper/__init__.py

@@ -4,7 +4,7 @@
 from opentelemetry.context import attach, detach
 from opentelemetry.trace.propagation.tracecontext import TraceContextTextMapPropagator
 
-setup_opentelemetry(app=app)
+# setup_opentelemetry(app=app)
 tracer = setup_tracer(__name__)