Skip to content

Commit 8555097

Browse files
Merge branch 'master' into add-support_standard_group_id-to-statics
2 parents 23bac44 + 127a619 commit 8555097

File tree

3 files changed

+8
-2
lines changed

3 files changed

+8
-2
lines changed

packages/service-library/src/servicelib/fastapi/client_session.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,20 +2,25 @@
22

33
import httpx
44
from fastapi import FastAPI
5+
from servicelib.fastapi.tracing import setup_httpx_client_tracing
6+
from settings_library.tracing import TracingSettings
57

68

79
def setup_client_session(
810
app: FastAPI,
911
*,
1012
default_timeout: datetime.timedelta = datetime.timedelta(seconds=20),
11-
max_keepalive_connections: int = 20
13+
max_keepalive_connections: int = 20,
14+
tracing_settings: TracingSettings | None,
1215
) -> None:
1316
async def on_startup() -> None:
1417
session = httpx.AsyncClient(
1518
transport=httpx.AsyncHTTPTransport(http2=True),
1619
limits=httpx.Limits(max_keepalive_connections=max_keepalive_connections),
1720
timeout=default_timeout.total_seconds(),
1821
)
22+
if tracing_settings:
23+
setup_httpx_client_tracing(session)
1924
app.state.aiohttp_client_session = session
2025

2126
async def on_shutdown() -> None:

services/director/src/simcore_service_director/core/application.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -50,6 +50,7 @@ def create_app(settings: ApplicationSettings) -> FastAPI:
5050
app,
5151
max_keepalive_connections=settings.DIRECTOR_REGISTRY_CLIENT_MAX_KEEPALIVE_CONNECTIONS,
5252
default_timeout=settings.DIRECTOR_REGISTRY_CLIENT_TIMEOUT,
53+
tracing_settings=settings.DIRECTOR_TRACING,
5354
)
5455
setup_registry(app)
5556

services/storage/src/simcore_service_storage/core/application.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -69,7 +69,7 @@ def create_app(settings: ApplicationSettings) -> FastAPI: # noqa: C901
6969

7070
setup_db(app)
7171
setup_s3(app)
72-
setup_client_session(app)
72+
setup_client_session(app, tracing_settings=settings.STORAGE_TRACING)
7373

7474
if settings.STORAGE_CELERY and not settings.STORAGE_WORKER_MODE:
7575
setup_rabbitmq(app)

0 commit comments

Comments
 (0)