aboutsummaryrefslogtreecommitdiffstats
path: root/crates/pseudonyms/src/server
diff options
context:
space:
mode:
Diffstat (limited to 'crates/pseudonyms/src/server')
-rw-r--r--crates/pseudonyms/src/server/interceptor.rs23
1 files changed, 23 insertions, 0 deletions
diff --git a/crates/pseudonyms/src/server/interceptor.rs b/crates/pseudonyms/src/server/interceptor.rs
new file mode 100644
index 0000000..f30f140
--- /dev/null
+++ b/crates/pseudonyms/src/server/interceptor.rs
@@ -0,0 +1,23 @@
+use warden_stack::{
+ opentelemetry::global, tracing::telemetry::tonic::extractor,
+ tracing_opentelemetry::OpenTelemetrySpanExt,
+};
+use tonic::{Status, service::Interceptor};
+use tracing::Span;
+
+#[derive(Clone, Copy)]
+pub struct MyInterceptor;
+
+impl Interceptor for MyInterceptor {
+ fn call(&mut self, request: tonic::Request<()>) -> Result<tonic::Request<()>, Status> {
+ let span = Span::current();
+
+ let cx = global::get_text_map_propagator(|propagator| {
+ propagator.extract(&extractor::MetadataMap(request.metadata()))
+ });
+
+ span.set_parent(cx);
+
+ Ok(request)
+ }
+}