Telemetry Config Change
Build Library / Build Library (push) Successful in 4m54s

Added implementation instead of separate function to load Telemetry configuration
This commit is contained in:
brotoskyj
2025-11-21 14:00:09 -05:00
parent 303ecd8368
commit 87ab1e3b28
5 changed files with 29 additions and 30 deletions
+1 -1
View File
@@ -26,7 +26,7 @@ dependencies = [
[[package]] [[package]]
name = "airlock_libs" name = "airlock_libs"
version = "3.2.0" version = "4.0.0"
dependencies = [ dependencies = [
"chrono", "chrono",
"indicatif", "indicatif",
+1 -1
View File
@@ -1,6 +1,6 @@
[package] [package]
name = "airlock_libs" name = "airlock_libs"
version = "3.2.0" version = "4.0.0"
edition = "2024" edition = "2024"
[lib] [lib]
+1 -1
View File
@@ -4,7 +4,7 @@ build-backend = "maturin"
[project] [project]
name = "airlock_libs" name = "airlock_libs"
version = "3.2.0" version = "4.0.0"
description = "Airlock Digital API Wrapper" description = "Airlock Digital API Wrapper"
readme = "README.md" readme = "README.md"
license = { text = "AGPL-3.0-only" } license = { text = "AGPL-3.0-only" }
+25 -26
View File
@@ -32,6 +32,27 @@ struct TelemetryConfig {
TELEM_URL: Option<String>, TELEM_URL: Option<String>,
} }
impl TelemetryConfig {
pub fn load() -> Self {
let cfg_path = get_base_directory().join("config\\user_config.json");
if !cfg_path.exists() {
return Self {
TELEMETRY: false,
TELEM_URL: None,
};
}
match fs::read_to_string(&cfg_path) {
Ok(contents) => serde_json::from_str::<Self>(&contents).unwrap_or(Self {
TELEMETRY: false,
TELEM_URL: None,
}),
Err(_) => Self {
TELEMETRY: false,
TELEM_URL: None,
},
}
}
}
#[derive(Debug, Deserialize, Serialize)] #[derive(Debug, Deserialize, Serialize)]
struct ApiResponse { struct ApiResponse {
error: String, error: String,
@@ -92,7 +113,7 @@ pub fn pull_policy_exec_histories(
&& !parent_dir.exists() && !parent_dir.exists()
{ {
match fs::create_dir_all(parent_dir) { match fs::create_dir_all(parent_dir) {
Ok(_) => {}, Ok(_) => {}
Err(e) => { Err(e) => {
println!("Failed to Create Directory {:?}: {}", parent_dir, e); println!("Failed to Create Directory {:?}: {}", parent_dir, e);
std::process::abort(); std::process::abort();
@@ -100,7 +121,7 @@ pub fn pull_policy_exec_histories(
} }
} }
match fs::File::create(&file_path) { match fs::File::create(&file_path) {
Ok(_) => {}, Ok(_) => {}
Err(e) => { Err(e) => {
println!("Failed to Create Directory {:?}: {}", &file_path, e); println!("Failed to Create Directory {:?}: {}", &file_path, e);
std::process::abort(); std::process::abort();
@@ -167,7 +188,7 @@ pub fn pull_policy_exec_histories(
span.set_attribute(Key::new("Days").string(days.to_string().to_string())); span.set_attribute(Key::new("Days").string(days.to_string().to_string()));
loop { loop {
match f.seek(SeekFrom::Start(0)) { match f.seek(SeekFrom::Start(0)) {
Ok(_) => {}, Ok(_) => {}
Err(e) => { Err(e) => {
println!("Failed to seek start of {:?}: {}", f, e); println!("Failed to seek start of {:?}: {}", f, e);
std::process::abort(); std::process::abort();
@@ -360,30 +381,8 @@ fn skipback(days: i64) -> ObjectId {
ObjectId::parse_str(&objectid_hex).expect("Invalid ObjectId hex") ObjectId::parse_str(&objectid_hex).expect("Invalid ObjectId hex")
} }
fn load_telemetry_config() -> TelemetryConfig {
let cfg_path = get_base_directory().join("config\\user_config.json");
if !cfg_path.exists() {
return TelemetryConfig {
TELEMETRY: false,
TELEM_URL: None,
};
}
match fs::read_to_string(&cfg_path) {
Ok(contents) => {
serde_json::from_str::<TelemetryConfig>(&contents).unwrap_or(TelemetryConfig {
TELEMETRY: false,
TELEM_URL: None,
})
}
Err(_) => TelemetryConfig {
TELEMETRY: false,
TELEM_URL: None,
},
}
}
fn init_tracer() -> Result<Option<sdktrace::Tracer>, TraceError> { fn init_tracer() -> Result<Option<sdktrace::Tracer>, TraceError> {
let cfg = load_telemetry_config(); let cfg = TelemetryConfig::load();
if !cfg.TELEMETRY { if !cfg.TELEMETRY {
global::set_tracer_provider(NoopTracerProvider::new()); global::set_tracer_provider(NoopTracerProvider::new());
return Ok(None); return Ok(None);
+1 -1
View File
@@ -11,4 +11,4 @@ urllib3==2.5.0
pyperclip==1.11.0 pyperclip==1.11.0
--extra-index-url https://git.racooncity.org/api/packages/brotoskyj/pypi/simple/ --extra-index-url https://git.racooncity.org/api/packages/brotoskyj/pypi/simple/
airlock_libs==3.2.0 airlock_libs==4.0.0