stubr 0.6.2

Wiremock implemented in Rust
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53

use std::str::FromStr;

use crate::wiremock::{Request, ResponseTemplate};
use http_types::headers::HeaderName;

pub struct OpenTracing<'a>(pub &'a Request);

impl<'a> OpenTracing<'a> {
    const TRACE_ID_KEY: &'static str = "x-b3-traceid";
    const SPAN_ID_KEY: &'static str = "x-b3-spanid";
    const PARENT_SPAN_ID_KEY: &'static str = "x-b3-parentspanid";
    const SAMPLED_KEY: &'static str = "x-b3-sampled";
    const B3_KEY: &'static str = "b3";
    const OPEN_TRACING_HEADERS: [&'static str; 5] = [
        Self::TRACE_ID_KEY,
        Self::SPAN_ID_KEY,
        Self::PARENT_SPAN_ID_KEY,
        Self::SAMPLED_KEY,
        Self::B3_KEY,
    ];

    pub fn add_opentracing_header(
        &'a self, mut resp: ResponseTemplate, stub_headers: Option<impl Iterator<Item = &'a str>>,
    ) -> ResponseTemplate {
        if let Some(h) = stub_headers {
            for (k, v) in self.mixed_tracing_headers(h) {
                resp = resp.insert_header(k, v);
            }
        } else {
            for (k, v) in self.tracing_headers() {
                resp = resp.insert_header(k, v);
            }
        }
        resp
    }

    /// considering headers defined in stubs
    fn mixed_tracing_headers(&'a self, stub_headers: impl Iterator<Item = &'a str>) -> impl Iterator<Item = (&'a str, &'a str)> {
        stub_headers
            .filter(|it| !Self::OPEN_TRACING_HEADERS.contains(it))
            .filter_map(|k| self.req_header(k).map(|v| (k, v)))
    }

    fn tracing_headers(&'a self) -> impl Iterator<Item = (&'a str, &'a str)> {
        Self::OPEN_TRACING_HEADERS
            .into_iter()
            .filter_map(|k| self.req_header(k).map(|v| (k, v)))
    }

    fn req_header(&'a self, key: &'a str) -> Option<&'a str> {
        self.0.headers.get(HeaderName::from_str(key).ok().as_ref()?).map(|v| v.as_str())
    }
}