Spaces:
Build error
Build error
File size: 3,091 Bytes
d8435ba |
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 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 |
#[cfg(target_os = "linux")]
pub mod pyro {
use pyroscope::pyroscope::PyroscopeAgentRunning;
use pyroscope::{PyroscopeAgent, PyroscopeError};
use pyroscope_pprofrs::{pprof_backend, PprofConfig};
use crate::common::debugger::PyroscopeConfig;
pub struct PyroscopeState {
pub config: PyroscopeConfig,
pub agent: Option<PyroscopeAgent<PyroscopeAgentRunning>>,
}
impl PyroscopeState {
fn build_agent(
config: &PyroscopeConfig,
) -> Result<PyroscopeAgent<PyroscopeAgentRunning>, PyroscopeError> {
let pprof_config = PprofConfig::new().sample_rate(config.sampling_rate.unwrap_or(100));
let backend_impl = pprof_backend(pprof_config);
log::info!(
"Starting pyroscope agent with identifier {}",
&config.identifier
);
// TODO: Add more tags like peerId and peerUrl
let agent = PyroscopeAgent::builder(config.url.to_string(), "qdrant".to_string())
.backend(backend_impl)
.tags(vec![("app", "Qdrant"), ("identifier", &config.identifier)])
.build()?;
let running_agent = agent.start()?;
Ok(running_agent)
}
pub fn from_config(config: Option<PyroscopeConfig>) -> Option<Self> {
match config {
Some(pyro_config) => {
let agent = PyroscopeState::build_agent(&pyro_config);
match agent {
Ok(agent) => Some(PyroscopeState {
config: pyro_config,
agent: Some(agent),
}),
Err(err) => {
log::warn!("Pyroscope agent failed to start {}", err);
None
}
}
}
None => None,
}
}
pub fn stop_agent(&mut self) -> bool {
log::info!("Stopping pyroscope agent");
if let Some(agent) = self.agent.take() {
match agent.stop() {
Ok(stopped_agent) => {
log::info!("Stopped pyroscope agent. Shutting it down");
stopped_agent.shutdown();
log::info!("Pyroscope agent shut down completed.");
return true;
}
Err(err) => {
log::warn!("Pyroscope agent failed to stop {}", err);
return false;
}
}
}
true
}
}
impl Drop for PyroscopeState {
fn drop(&mut self) {
self.stop_agent();
}
}
}
#[cfg(not(target_os = "linux"))]
pub mod pyro {
use crate::common::debugger::PyroscopeConfig;
pub struct PyroscopeState {}
impl PyroscopeState {
pub fn from_config(_config: Option<PyroscopeConfig>) -> Option<Self> {
None
}
}
}
|