91 lines
2.2 KiB
Rust
91 lines
2.2 KiB
Rust
#![cfg(feature = "sentry_telemetry")]
|
|
|
|
use std::{
|
|
str::FromStr,
|
|
sync::{Arc, OnceLock},
|
|
};
|
|
|
|
use conduit::{config::Config, debug, trace};
|
|
use sentry::{
|
|
types::{
|
|
protocol::v7::{Context, Event},
|
|
Dsn,
|
|
},
|
|
Breadcrumb, ClientOptions, Level,
|
|
};
|
|
|
|
static SEND_PANIC: OnceLock<bool> = OnceLock::new();
|
|
static SEND_ERROR: OnceLock<bool> = OnceLock::new();
|
|
|
|
pub(crate) fn init(config: &Config) -> Option<sentry::ClientInitGuard> {
|
|
config.sentry.then(|| sentry::init(options(config)))
|
|
}
|
|
|
|
fn options(config: &Config) -> ClientOptions {
|
|
SEND_PANIC
|
|
.set(config.sentry_send_panic)
|
|
.expect("SEND_PANIC was not previously set");
|
|
SEND_ERROR
|
|
.set(config.sentry_send_error)
|
|
.expect("SEND_ERROR was not previously set");
|
|
|
|
let dsn = config
|
|
.sentry_endpoint
|
|
.as_ref()
|
|
.expect("init_sentry should only be called if sentry is enabled and this is not None")
|
|
.as_str();
|
|
|
|
ClientOptions {
|
|
dsn: Some(Dsn::from_str(dsn).expect("sentry_endpoint must be a valid URL")),
|
|
server_name: config
|
|
.sentry_send_server_name
|
|
.then(|| config.server_name.to_string().into()),
|
|
traces_sample_rate: config.sentry_traces_sample_rate,
|
|
debug: cfg!(debug_assertions),
|
|
release: sentry::release_name!(),
|
|
user_agent: conduit::version::user_agent().into(),
|
|
attach_stacktrace: config.sentry_attach_stacktrace,
|
|
before_send: Some(Arc::new(before_send)),
|
|
before_breadcrumb: Some(Arc::new(before_breadcrumb)),
|
|
..Default::default()
|
|
}
|
|
}
|
|
|
|
fn before_send(event: Event<'static>) -> Option<Event<'static>> {
|
|
if event.exception.iter().any(|e| e.ty == "panic") && !SEND_PANIC.get().unwrap_or(&true) {
|
|
return None;
|
|
}
|
|
|
|
if event.level == Level::Error {
|
|
if !SEND_ERROR.get().unwrap_or(&true) {
|
|
return None;
|
|
}
|
|
|
|
if cfg!(debug_assertions) {
|
|
return None;
|
|
}
|
|
|
|
//NOTE: we can enable this to specify error!(sentry = true, ...)
|
|
if let Some(Context::Other(context)) = event.contexts.get("Rust Tracing Fields") {
|
|
if !context.contains_key("sentry") {
|
|
//return None;
|
|
}
|
|
}
|
|
}
|
|
|
|
if event.level == Level::Fatal {
|
|
trace!("{event:#?}");
|
|
}
|
|
|
|
debug!("Sending sentry event: {event:?}");
|
|
Some(event)
|
|
}
|
|
|
|
fn before_breadcrumb(crumb: Breadcrumb) -> Option<Breadcrumb> {
|
|
if crumb.ty == "log" && crumb.level == Level::Debug {
|
|
return None;
|
|
}
|
|
|
|
trace!("Sentry breadcrumb: {crumb:?}");
|
|
Some(crumb)
|
|
}
|