use azure_data_cosmos_driver::models::CosmosResponseHeaders;
#[derive(Debug, Clone, Default)]
pub struct CosmosDiagnostics {
activity_id: Option<String>,
server_duration_ms: Option<f64>,
}
impl CosmosDiagnostics {
pub(crate) fn from_headers(headers: &CosmosResponseHeaders) -> Self {
Self {
activity_id: headers.activity_id.as_ref().map(|a| a.as_str().to_owned()),
server_duration_ms: headers.server_duration_ms,
}
}
pub fn activity_id(&self) -> Option<&str> {
self.activity_id.as_deref()
}
pub fn server_duration_ms(&self) -> Option<f64> {
self.server_duration_ms
}
}