feat: update traces

traces/Cargo.toml

@@ -10,4 +10,4 @@ opentelemetry-otlp = { version = "0.11.0", features = ["tonic", "tls", "tls-root
 tracing = { version = "0.1.37" }
 serde = { version = "1.0.152", features = ["derive"] }
 tonic = { version = "0.8.1", features = ["tls"] }
-tokio = { version = "1.24.2", features = ["default"] }
+tokio = { version = "1.25.0", features = ["default"] }

traces/src/grpc.rs

@@ -0,0 +1,53 @@
+use opentelemetry::{
+    global::{self, BoxedSpan, BoxedTracer},
+    propagation::{Extractor, Injector},
+    trace::Tracer,
+    Context,
+};
+
+pub struct ExMetadataMap<'a>(&'a tonic::metadata::MetadataMap);
+
+impl<'a> Extractor for ExMetadataMap<'a> {
+    /// Get a value for a key from the MetadataMap.  If the value can't be converted to &str, returns None
+    fn get(&self, key: &str) -> Option<&str> {
+        self.0.get(key).and_then(|metadata| metadata.to_str().ok())
+    }
+
+    /// Collect all the keys from the MetadataMap.
+    fn keys(&self) -> Vec<&str> {
+        self.0
+            .keys()
+            .map(|key| match key {
+                tonic::metadata::KeyRef::Ascii(v) => v.as_str(),
+                tonic::metadata::KeyRef::Binary(v) => v.as_str(),
+            })
+            .collect::<Vec<_>>()
+    }
+}
+
+pub struct InjMetadataMap<'a>(&'a mut tonic::metadata::MetadataMap);
+
+impl<'a> Injector for InjMetadataMap<'a> {
+    /// Set a key and value in the MetadataMap.  Does nothing if the key or value are not valid inputs
+    fn set(&mut self, key: &str, value: String) {
+        if let Ok(key) = tonic::metadata::MetadataKey::from_bytes(key.as_bytes()) {
+            if let Ok(val) = tonic::metadata::MetadataValue::try_from(&value) {
+                self.0.insert(key, val);
+            }
+        }
+    }
+}
+
+pub fn span(meta: &tonic::metadata::MetadataMap, tracer: &BoxedTracer) -> (Context, BoxedSpan) {
+    let ctx = global::get_text_map_propagator(|prop| prop.extract(&ExMetadataMap(meta)));
+
+    let span = tracer.start_with_context("Processing reply", &ctx);
+
+    (ctx, span)
+}
+
+pub fn inject(ctx: &Context, meta: &mut tonic::metadata::MetadataMap) {
+    global::get_text_map_propagator(|propagator| {
+        propagator.inject_context(&ctx, &mut InjMetadataMap(meta))
+    });
+}

traces/src/lib.rs

@@ -1,6 +1,4 @@
-///deprecated
-pub mod amqp;
-
+pub mod grpc;
 pub mod jaeger;
 pub mod otlp;