added req-id correlation to subscriptions

Author: dotansimha

bin/router/src/lib.rs

@@ -66,8 +66,10 @@ pub use hive_router_config::humantime_serde;
 use hive_router_config::{load_config, subscriptions::CallbackConfig, HiveRouterConfig};
 pub use hive_router_internal::background_tasks;
 use hive_router_internal::telemetry::{
-    logging::targets, otel::tracing_opentelemetry::OpenTelemetrySpanExt,
-    traces::spans::http_request::HttpServerRequestSpan, TelemetryContext,
+    logging::targets,
+    otel::{opentelemetry, tracing_opentelemetry::OpenTelemetrySpanExt},
+    traces::spans::http_request::HttpServerRequestSpan,
+    TelemetryContext,
 };
 pub use hive_router_internal::BoxError;
 use hive_router_internal::{
@@ -186,9 +188,14 @@ async fn graphql_endpoint_handler(
             "http request attributes",
         );
 
-        let inner_res =
-            graphql_endpoint_dispatch(&mut request, body_stream, schema_state, app_state.clone())
-                .await;
+        let inner_res = graphql_endpoint_dispatch(
+            &mut request,
+            body_stream,
+            schema_state,
+            app_state.clone(),
+            parent_ctx,
+        )
+        .await;
 
         let payload_bytes = match inner_res.body().size() {
             BodySize::Empty | BodySize::None => 0,
@@ -234,10 +241,8 @@ async fn graphql_endpoint_dispatch(
     body_stream: web::types::Payload,
     schema_state: web::types::State<Arc<SchemaState>>,
     app_state: web::types::State<Arc<RouterSharedState>>,
+    parent_ctx: opentelemetry::Context,
 ) -> web::HttpResponse {
-    let parent_ctx = app_state
-        .telemetry_context
-        .extract_context(&HeaderExtractor(request.headers()));
     let root_http_request_span = HttpServerRequestSpan::from_request(
         request,
         &app_state