Add src/middleware/lambda/value.rs

Author: lbeschastny

src/middleware/lambda/context.rs

@@ -1,44 +1,19 @@
 use crate::semconv;
 use lambda_runtime::LambdaInvocation;
-use opentelemetry::{trace::SpanKind, StringValue};
+use opentelemetry::trace::SpanKind;
 use opentelemetry_sdk::trace::TracerProvider;
 use tracing::Span;
 use tracing_opentelemetry_instrumentation_sdk::TRACING_TARGET;
 
-use super::OtelLambdaLayer;
+use super::{
+    value::{OptionalValue, Value},
+    OtelLambdaLayer,
+};
 
 pub trait LambdaServiceContext {
     fn create_span(&self, req: &LambdaInvocation, coldstart: bool) -> Span;
 }
 
-pub struct Value(StringValue);
-
-impl Value {
-    pub fn as_str(&self) -> &str {
-        self.0.as_str()
-    }
-}
-
-impl<T: Into<StringValue>> From<T> for Value {
-    fn from(inner: T) -> Self {
-        Self(inner.into())
-    }
-}
-
-pub struct OptionalValue(Option<StringValue>);
-
-impl OptionalValue {
-    pub fn as_str(&self) -> Option<&str> {
-        self.0.as_ref().map(|value| value.as_str())
-    }
-}
-
-impl<T: Into<StringValue>> From<Option<T>> for OptionalValue {
-    fn from(inner: Option<T>) -> Self {
-        Self(inner.map(|value| value.into()))
-    }
-}
-
 macro_rules! lambda_service {
     ($trigger: ident, $kind: ident, $service:ident { $($prop:ident: $type:ty,)* }$(, $($field:tt)*)?) => {
         #[allow(non_snake_case)]

src/middleware/lambda/mod.rs

@@ -1,4 +1,5 @@
 pub mod context;
 pub mod layer;
+pub mod value;
 
 pub use layer::OtelLambdaLayer;

src/middleware/lambda/value.rs

@@ -0,0 +1,23 @@
+use opentelemetry::StringValue;
+
+pub struct Value(StringValue);
+
+impl Value {
+    pub fn as_str(&self) -> &str {
+        self.0.as_str()
+    }
+}
+
+impl<T: Into<StringValue>> From<T> for Value {
+    fn from(inner: T) -> Self {
+        Self(inner.into())
+    }
+}
+
+pub struct OptionalValue(Option<StringValue>);
+
+impl OptionalValue {
+    pub fn as_str(&self) -> Option<&str> {
+        self.0.as_ref().map(|value| value.as_str())
+    }
+}