Instrument your Rust application with OpenTelemetry

1opentelemetry = { version = "[VERSION]", features = ["trace", "metrics"] }
2opentelemetry_sdk = { version = "[VERSION]" , features = ["rt-tokio"] }
3opentelemetry-otlp = { version = "[VERSION]", features = ["http-proto", "reqwest-client", "reqwest-rustls"] }
4opentelemetry-http = { version =  "[VERSION]" }
5opentelemetry-semantic-conventions = { version = "[VERSION]" }

1use std::collections::HashMap;
2use std::io::{BufRead, BufReader, Read};
3
4use opentelemetry::{
5    global,
6    trace::{Span, TraceContextExt, TraceError, Tracer},
7    Context, KeyValue,
8};
9
10use opentelemetry_sdk::{runtime, trace as sdktrace, Resource};
11use opentelemetry_http::{HeaderExtractor, HeaderInjector};
12use opentelemetry_otlp::WithExportConfig;
13use opentelemetry_semantic_conventions as semcov;

1fn init_opentelemetry() -> Result<sdktrace::Tracer, TraceError>  {
2    // Helper function to read potentially available OneAgent data
3    fn read_dt_metadata() -> Resource {
4        fn read_single(path: &str, metadata: &mut Vec<KeyValue>) -> std::io::Result<()> {
5            let mut file = std::fs::File::open(path)?;
6            if path.starts_with("dt_metadata") {
7                let mut name = String::new();
8                file.read_to_string(&mut name)?;
9                file = std::fs::File::open(name)?;
10            }
11            for line in BufReader::new(file).lines() {
12                if let Some((k, v)) = line?.split_once('=') {
13                    metadata.push(KeyValue::new(k.to_string(), v.to_string()))
14                }
15            }
16            Ok(())
17        }
18        let mut metadata = Vec::new();
19        for name in [
20            "dt_metadata_e617c525669e072eebe3d0f08212e8f2.properties",
21            "/var/lib/dynatrace/enrichment/dt_metadata.properties",
22            "/var/lib/dynatrace/enrichment/dt_host_metadata.properties"
23        ] {
24            let _ = read_single(name, &mut metadata);
25        }
26        Resource::new(metadata)
27    }
28
29    // ===== GENERAL SETUP =====
30    let DT_API_TOKEN = env::var("DT_API_TOKEN").unwrap(); // TODO: change
31    let DT_API_URL = env::var("DT_API_URL").unwrap();
32
33    let mut map = HashMap::new();
34    map.insert("Authorization".to_string(), format!("Api-Token {}", DT_API_TOKEN));
35    let mut resource = Resource::new([
36    KeyValue::new(semcov::resource::SERVICE_NAME, "rust-app") //TODO Replace with the name of your application
37    ]);
38    resource = resource.merge(&read_dt_metadata());
39
40    // ===== TRACING SETUP =====
41    global::set_text_map_propagator(TraceContextPropagator::new());
42
43    opentelemetry_otlp::new_pipeline()
44        .tracing()
45        .with_exporter(
46            opentelemetry_otlp::new_exporter()
47                .http()
48                .with_endpoint(DT_API_URL)
49                .with_headers(map),
50        )
51        .with_trace_config(
52            sdktrace::config()
53                .with_resource(resource)
54                .with_sampler(sdktrace::Sampler::AlwaysOn),
55        )
56        .install_batch(runtime::Tokio)
57}

1let tracer = global::tracer("my-tracer");

1let mut span = tracer.start("Call to /myendpoint");
2span.set_attribute(KeyValue::new("http.method", "GET"));
3span.set_attribute(KeyValue::new("net.protocol.version", "1.1"));
4
5// TODO: Your code goes here
6
7span.end();