Add cross-instrument parameter for health checks span generation

Author: insani7y

microbootstrap/bootstrappers/faststream.py

@@ -111,7 +111,6 @@ def get_config_type(cls) -> type[FastStreamPrometheusConfig]:
 class FastStreamHealthChecksInstrument(HealthChecksInstrument):
     def bootstrap(self) -> None: ...
     def bootstrap_before(self) -> dict[str, typing.Any]:
-        @tracer.start_as_current_span(f"GET {self.instrument_config.health_checks_path}")
         @handle_get
         async def check_health(scope: typing.Any) -> AsgiResponse:  # noqa: ANN401, ARG001
             return (
@@ -124,6 +123,11 @@ async def check_health(scope: typing.Any) -> AsgiResponse:  # noqa: ANN401, ARG0
                 else AsgiResponse(b"Service is unhealthy", 500, headers={"content-type": "application/json"})
             )
 
+        if self.instrument_config.opentelemetry_generate_health_check_spans:
+            check_health = tracer.start_as_current_span(f"GET {self.instrument_config.health_checks_path}")(
+                check_health,
+            )
+
         return {"asgi_routes": ((self.instrument_config.health_checks_path, check_health),)}
 
     async def define_health_status(self) -> bool:

microbootstrap/instruments/health_checks_instrument.py

@@ -19,6 +19,9 @@ class HealthChecksConfig(BaseInstrumentConfig):
     health_checks_path: str = "/health/"
     health_checks_include_in_schema: bool = False
 
+    # Cross-instrument parameter, comes from opentelemetry
+    opentelemetry_generate_health_check_spans: bool = True
+
 
 class HealthChecksInstrument(Instrument[HealthChecksConfig]):
     instrument_name = "Health checks"