Skip to content

Commit

Permalink
Propagate opentelemetry tracing headers in requests to limitador
Browse files Browse the repository at this point in the history
  • Loading branch information
adam-cattermole committed Mar 19, 2024
1 parent f4e38b2 commit c37de84
Show file tree
Hide file tree
Showing 2 changed files with 18 additions and 1 deletion.
18 changes: 17 additions & 1 deletion src/filter/http_context.rs
Original file line number Diff line number Diff line change
Expand Up @@ -21,6 +21,7 @@ pub struct Filter {
pub context_id: u32,
pub config: Rc<FilterConfig>,
pub response_headers_to_add: Vec<(String, String)>,
pub tracing_headers: Vec<(String, String)>,
}

impl Filter {
Expand Down Expand Up @@ -51,11 +52,17 @@ impl Filter {

let rl_req_serialized = Message::write_to_bytes(&rl_req).unwrap(); // TODO(rahulanand16nov): Error Handling

let rl_tracing_headers = self
.tracing_headers
.iter()
.map(|(header, value)| (header.as_str(), value.as_bytes()))
.collect();

match self.dispatch_grpc_call(
rlp.service.as_str(),
RATELIMIT_SERVICE_NAME,
RATELIMIT_METHOD_NAME,
Vec::new(),
rl_tracing_headers,
Some(&rl_req_serialized),
Duration::from_secs(5),
) {
Expand Down Expand Up @@ -207,6 +214,15 @@ impl HttpContext for Filter {
fn on_http_request_headers(&mut self, _: usize, _: bool) -> Action {
info!("on_http_request_headers #{}", self.context_id);

let req_headers = self.get_http_request_headers();
for (header, value) in req_headers.iter() {
match header.as_str() {
"traceparent" | "tracestate" | "baggage" => {
self.tracing_headers.push((header.clone(), value.clone()))
}
_ => (),
}
}
match self
.config
.index
Expand Down
1 change: 1 addition & 0 deletions src/filter/root_context.rs
Original file line number Diff line number Diff line change
Expand Up @@ -22,6 +22,7 @@ impl RootContext for FilterRoot {
context_id,
config: Rc::clone(&self.config),
response_headers_to_add: Vec::default(),
tracing_headers: Vec::default(),
}))
}

Expand Down

0 comments on commit c37de84

Please sign in to comment.