Applying env-filter to OpenTelemetry span exporter (in Rust microservices) | Adding error handling for Elasticsearch response parsing

Archisman-Mridha · Archisman-Mridha · commit 07db74b3dd48 · 2024-02-07T16:55:18.000+05:30
diff --git a/Cargo.toml b/Cargo.toml
@@ -45,7 +45,7 @@ futures = "0.3.30"
 jsonwebtoken = "9.2.0"
 kafka = { version = "0.10.0", default-features = false }
 lazy_static = "1.4.0"
-opentelemetry = "0.21.0"
+opentelemetry = { version = "0.21.0", default-features = false, features = ["trace"] }
 opentelemetry-otlp = { version = "0.14.0", default-features = false, features = ["gzip-tonic", "trace", "grpc-tonic"] }
 opentelemetry_sdk = { version = "0.21.2", features = ["rt-tokio"] }
 postgres-types = { version = "0.2.6", features = ["with-time-0_3"] }
diff --git a/backend/lib.rs b/backend/lib.rs
@@ -87,11 +87,13 @@ pub mod utils {
     };
     use tokio::spawn;
     use tracing::{
-      debug, debug_span, level_filters::LevelFilter, subscriber::set_global_default, warn, Span,
+      debug, info_span, level_filters::LevelFilter, subscriber::set_global_default, warn, Span,
     };
     use tracing_opentelemetry::{OpenTelemetryLayer, OpenTelemetrySpanExt};
-    use tracing_subscriber::fmt;
-    use tracing_subscriber::{layer::SubscriberExt, EnvFilter, Layer, Registry};
+    use tracing_subscriber::{
+      layer::{Layered, SubscriberExt},
+      EnvFilter, Registry,
+    };
     use warp::Filter;
 
     pub fn setupObservability(serviceName: &'static str) {
@@ -102,8 +104,9 @@ pub mod utils {
 
       set_global_default(
         Registry::default()
+          .with(envFilter)
           .with(startTracer(serviceName))
-          .with(fmt::layer().with_filter(envFilter)),
+          .with(tracing_subscriber::fmt::layer()),
       )
       .expect("ERROR : Setting up global default tracing registry");
 
@@ -127,7 +130,9 @@ pub mod utils {
     }
 
     // startTracer creates an OpenTelemetry tracing pipeline and sets the global tracer.
-    pub fn startTracer(serviceName: &'static str) -> OpenTelemetryLayer<Registry, Tracer> {
+    pub fn startTracer(
+      serviceName: &'static str,
+    ) -> OpenTelemetryLayer<Layered<EnvFilter, Registry>, Tracer> {
       opentelemetry::global::set_text_map_propagator(TraceContextPropagator::new());
 
       let tracer = new_pipeline()
@@ -155,7 +160,7 @@ pub mod utils {
 
     pub fn makeSpan(request: &Request<Body>) -> Span {
       let headers = request.headers();
-      debug_span!(
+      info_span!(
         "Incoming Request",
         ?headers,
         trace_id = tracing::field::Empty
diff --git a/backend/microservices/profiles/adapters/elasticsearch.rs b/backend/microservices/profiles/adapters/elasticsearch.rs
@@ -108,17 +108,17 @@ impl ElasticsearchAdapter {
 // deserializeQueryProfilesResponse takes in the response returned from Elasticsearch for the query
 // done in ElasticsearchAdapter.searchProfiles and tries to deserialize it to Vec<ProfilePreview>.
 async fn deserializeQueryProfilesResponse(response: Response) -> Result<Vec<ProfilePreview>> {
-  #[derive(Deserialize)]
+  #[derive(Debug, Deserialize)]
   struct Response {
     hits: Hits,
   }
 
-  #[derive(Deserialize)]
+  #[derive(Debug, Deserialize)]
   struct Hits {
     hits: Vec<Hit>,
   }
 
-  #[derive(Deserialize)]
+  #[derive(Debug, Deserialize)]
   struct Hit {
     _id: String,
     _source: Source,
@@ -133,7 +133,7 @@ async fn deserializeQueryProfilesResponse(response: Response) -> Result<Vec<Prof
   let response = response.bytes().await?.to_vec();
   let response = from_utf8(&response)?;
 
-  let response: Response = serde_json::from_str(response).unwrap();
+  let response: Response = serde_json::from_str(response)?;
 
   let profiles = response
     .hits
