datadog_apm_sync/
api.rs

1use crate::{client::ApmConfig, model::Span};
2use serde::Serialize;
3use std::collections::HashMap;
4
5const SAMPLING_PRIORITY_KEY: &str = "_sampling_priority_v1";
6const ANALYTICS_SAMPLE_RATE_KEY: &str = "_dd1.sr.eausr";
7const _SAMPLE_RATE_METRIC_KEY: &str = "_sample_rate";
8const _SAMPLING_AGENT_DECISION: &str = "_dd.agent_psr";
9const _SAMPLING_RULE_DECISION: &str = "_dd.rule_psr";
10const _SAMPLING_LIMIT_DECISION: &str = "_dd.limit_psr";
11
12fn fill_meta(span: &Span, env: Option<String>) -> HashMap<String, String> {
13    let mut meta = HashMap::new();
14    if let Some(env) = env {
15        meta.insert("env".to_string(), env);
16    }
17    if let Some(sql) = &span.sql {
18        meta.insert("sql.query".to_string(), sql.query.clone());
19        meta.insert("sql.rows".to_string(), sql.rows.clone());
20        meta.insert("sql.db".to_string(), sql.db.clone());
21    }
22    for (key, value) in &span.tags {
23        meta.insert(key.to_string(), value.to_string());
24    }
25    meta
26}
27
28fn fill_metrics(apm_config: &ApmConfig) -> HashMap<String, f64> {
29    let mut metrics = HashMap::new();
30    if apm_config.apm_enabled {
31        metrics.insert(
32            SAMPLING_PRIORITY_KEY.to_string(),
33            apm_config.sample_priority,
34        );
35        metrics.insert(
36            ANALYTICS_SAMPLE_RATE_KEY.to_string(),
37            apm_config.sample_rate,
38        );
39    }
40    metrics
41}
42
43#[derive(Debug, Serialize, Clone, PartialEq)]
44pub struct RawSpan {
45    service: String,
46    name: String,
47    resource: String,
48    trace_id: u64,
49    span_id: u64,
50    parent_id: Option<u64>,
51    start: u64,
52    duration: u64,
53    error: i32,
54    meta: HashMap<String, String>,
55    metrics: HashMap<String, f64>,
56    r#type: String,
57}
58
59impl RawSpan {
60    pub fn from_span(span: &Span, service: &str, env: &Option<String>, cfg: &ApmConfig) -> RawSpan {
61        let http_enabled = span.tags.contains_key("http.url");
62        let is_error = span.tags.contains_key("error.message");
63        RawSpan {
64            service: service.to_string(),
65            trace_id: span.trace_id,
66            span_id: span.id,
67            name: span.name.clone(),
68            resource: span.resource.clone(),
69            parent_id: span.parent_id,
70            start: span.start.timestamp_nanos_opt().unwrap_or(0) as u64,
71            duration: span.duration.num_nanoseconds().unwrap_or(0) as u64,
72            error: if is_error { 1 } else { 0 },
73            r#type: if http_enabled { "web" } else { "custom" }.to_string(),
74            meta: fill_meta(span, env.clone()),
75            metrics: fill_metrics(cfg),
76        }
77    }
78}
datadog_apm_sync/api.rs

datadog_apm_sync/
api.rs
