From 1968002d656383069a386bd874c9f0cc83e3116e Mon Sep 17 00:00:00 2001 From: rtkay123 Date: Fri, 15 Aug 2025 19:36:22 +0200 Subject: feat(rule-exec): receive messages --- lib/warden-middleware/src/grpc.rs | 28 ++++++++++++++++++++++++++++ 1 file changed, 28 insertions(+) create mode 100644 lib/warden-middleware/src/grpc.rs (limited to 'lib/warden-middleware/src/grpc.rs') diff --git a/lib/warden-middleware/src/grpc.rs b/lib/warden-middleware/src/grpc.rs new file mode 100644 index 0000000..f239ddb --- /dev/null +++ b/lib/warden-middleware/src/grpc.rs @@ -0,0 +1,28 @@ +pub mod interceptor { + use tonic::{ + Status, + service::{Interceptor, interceptor::InterceptedService}, + transport::Channel, + }; + use tracing::Span; + use warden_stack::{ + opentelemetry::global, tracing::telemetry::tonic::injector, + tracing_opentelemetry::OpenTelemetrySpanExt, + }; + + pub type Intercepted = InterceptedService; + + #[derive(Clone, Copy)] + pub struct MyInterceptor; + + impl Interceptor for MyInterceptor { + fn call(&mut self, mut request: tonic::Request<()>) -> Result, Status> { + let cx = Span::current().context(); + global::get_text_map_propagator(|propagator| { + propagator.inject_context(&cx, &mut injector::MetadataMap(request.metadata_mut())) + }); + + Ok(request) + } + } +} -- cgit v1.2.3