libdd-profiling 1.0.0

Continuous profiling library supporting pprof format collection and export to Datadog
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
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76

// Copyright 2021-Present Datadog, Inc. https://www.datadoghq.com/
// SPDX-License-Identifier: Apache-2.0

#[cfg(unix)]
use libdd_common::connector::uds;
use libdd_common::Endpoint;

#[cfg(windows)]
use libdd_common::connector::named_pipe;

use http::Uri;
use std::borrow::Cow;
use std::str::FromStr;

/// Creates an Endpoint for talking to the Datadog agent.
///
/// # Arguments
/// * `base_url` - has protocol, host, and port e.g. http://localhost:8126/
pub fn agent(base_url: Uri) -> anyhow::Result<Endpoint> {
    let mut parts = base_url.into_parts();
    let p_q = match parts.path_and_query {
        None => None,
        Some(pq) => {
            let path = pq.path();
            let path = path.strip_suffix('/').unwrap_or(path);
            Some(format!("{path}/profiling/v1/input").parse()?)
        }
    };
    parts.path_and_query = p_q;
    let url = Uri::from_parts(parts)?;
    Ok(Endpoint::from_url(url))
}

/// Creates an Endpoint for talking to the Datadog agent though a unix socket.
///
/// # Arguments
/// * `socket_path` - file system path to the socket
#[cfg(unix)]
pub fn agent_uds(path: &std::path::Path) -> anyhow::Result<Endpoint> {
    let base_url = uds::socket_path_to_uri(path)?;
    agent(base_url)
}

/// Creates an Endpoint for talking to the Datadog agent though a windows named pipe.
///
/// # Arguments
/// * `path` - file system path to the named pipe
#[cfg(windows)]
pub fn agent_named_pipe(path: &std::path::Path) -> anyhow::Result<Endpoint> {
    let base_url = named_pipe::named_pipe_path_to_uri(path)?;
    agent(base_url)
}

/// Creates an Endpoint for talking to Datadog intake without using the agent.
/// This is an experimental feature.
///
/// # Arguments
/// * `site` - e.g. "datadoghq.com".
/// * `api_key`
pub fn agentless<AsStrRef: AsRef<str>, IntoCow: Into<Cow<'static, str>>>(
    site: AsStrRef,
    api_key: IntoCow,
) -> anyhow::Result<Endpoint> {
    let intake_url: String = format!("https://intake.profile.{}/api/v2/profile", site.as_ref());

    Ok(Endpoint {
        url: Uri::from_str(intake_url.as_str())?,
        api_key: Some(api_key.into()),
        ..Default::default()
    })
}

pub fn file(path: impl AsRef<str>) -> anyhow::Result<Endpoint> {
    let url: String = format!("file://{}", path.as_ref());
    Ok(Endpoint::from_slice(&url))
}