handle payments service

bisgaard-itis · Nov 13, 2024 · 285bfc1 · 285bfc1
1 parent 46d0b2a
commit 285bfc1
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 0 deletions.
diff --git a/services/payments/src/simcore_service_payments/services/payments_gateway.py b/services/payments/src/simcore_service_payments/services/payments_gateway.py
@@ -25,6 +25,7 @@
     HealthMixinMixin,
 )
 from servicelib.fastapi.httpx_utils import to_curl_command
+from servicelib.fastapi.tracing import setup_httpx_client_tracing
 from simcore_service_payments.models.schemas.acknowledgements import (
     AckPaymentWithPaymentMethod,
 )
@@ -216,5 +217,7 @@ def setup_payments_gateway(app: FastAPI):
             secret=settings.PAYMENTS_GATEWAY_API_SECRET.get_secret_value()
         ),
     )
+    if settings.PAYMENTS_TRACING:
+        setup_httpx_client_tracing(api.client)
     api.attach_lifespan_to(app)
     api.set_to_app_state(app)
diff --git a/services/payments/src/simcore_service_payments/services/resource_usage_tracker.py b/services/payments/src/simcore_service_payments/services/resource_usage_tracker.py
@@ -25,6 +25,7 @@
     BaseHTTPApi,
     HealthMixinMixin,
 )
+from servicelib.fastapi.tracing import setup_httpx_client_tracing
 
 from ..core.settings import ApplicationSettings
 
@@ -73,5 +74,7 @@ def setup_resource_usage_tracker(app: FastAPI):
     api = ResourceUsageTrackerApi.from_client_kwargs(
         base_url=settings.PAYMENTS_RESOURCE_USAGE_TRACKER.base_url,
     )
+    if settings.PAYMENTS_TRACING:
+        setup_httpx_client_tracing(api.client)
     api.set_to_app_state(app)
     api.attach_lifespan_to(app)
diff --git a/services/payments/src/simcore_service_payments/services/stripe.py b/services/payments/src/simcore_service_payments/services/stripe.py
@@ -19,6 +19,7 @@
     BaseHTTPApi,
     HealthMixinMixin,
 )
+from servicelib.fastapi.tracing import setup_httpx_client_tracing
 
 from ..core.errors import StripeRuntimeError
 from ..core.settings import ApplicationSettings
@@ -91,6 +92,8 @@ def setup_stripe(app: FastAPI):
         base_url=settings.PAYMENTS_STRIPE_URL,
         auth=_StripeBearerAuth(settings.PAYMENTS_STRIPE_API_SECRET.get_secret_value()),
     )
+    if settings.PAYMENTS_TRACING:
+        setup_httpx_client_tracing(api.client)
 
     api.set_to_app_state(app)
     api.attach_lifespan_to(app)