swarmd_instruments 0.1.2

Declarative instrumentation
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

use std::{borrow::Cow, str::FromStr};

use anyhow::Context;
use sentry::{types::Dsn, ClientInitGuard};
use tracing_subscriber::{self, layer::SubscriberExt, registry::Registry};

/// Instrument layer for internal traces
#[must_use]
pub struct Instruments {
    _client_guard: Option<ClientInitGuard>,
}

impl Instruments {
    /// Create a new `Instruments` stack and register it globally.
    pub fn new(release_name: Option<Cow<'static, str>>) -> anyhow::Result<Self> {
        // NOTE(@miaxos): The dsn here is linked to swarmd-cli, if we have another bin, we should move that
        // part to be an argument.

        let should_activate = std::env::var("SWARMD_TELEMETRY_LEVEL")
            .ok()
            .and_then(|x| x.parse::<u8>().ok())
            .unwrap_or(1);

        let guard = if should_activate > 0 {
            Some(sentry::init(sentry::ClientOptions {
                dsn: Some(Dsn::from_str("https://3ca89d7f605a902ddd39375c2a5b1509@o4506304538345472.ingest.sentry.io/4506304540311552")?),
                debug: false,
                release: release_name,
                // Enable capturing of traces; set this a to lower value in production:
                traces_sample_rate: 1.0,
                ..sentry::ClientOptions::default()
            }))
        } else {
            None
        };

        let subscriber = Registry::default()
            .with(tracing_subscriber::EnvFilter::from_default_env())
            .with(sentry_tracing::layer())
            .with(tracing_subscriber::fmt::layer());

        tracing::subscriber::set_global_default(subscriber)
            .with_context(|| "cannot set global default subscriber")?;

        Ok(Self {
            _client_guard: guard,
        })
    }
}