airlock_libs/Cargo.lock

@@ -26,7 +26,7 @@ dependencies = [
 
 [[package]]
 name = "airlock_libs"
-version = "3.0.0"
+version = "3.1.0"
 dependencies = [
  "chrono",
  "indicatif",

airlock_libs/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "airlock_libs"
-version = "3.0.0"
+version = "3.1.0"
 edition = "2024"
 
 [lib]

airlock_libs/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "maturin"
 
 [project]
 name = "airlock_libs"
-version = "3.0.0"
+version = "3.1.0"
 description = "Airlock Digital API Wrapper"
 readme = "README.md"
 license = { text = "AGPL-3.0-only" }

airlock_libs/src/services.rs

@@ -3,6 +3,7 @@ use indicatif::{MultiProgress, ProgressBar, ProgressDrawTarget, ProgressStyle};
 use mongodb::bson::oid::ObjectId;
 use opentelemetry::global::shutdown_tracer_provider;
 use opentelemetry::sdk::Resource;
+use opentelemetry::trace::noop::NoopTracerProvider;
 use opentelemetry::trace::{Status, TraceContextExt, TraceError};
 use opentelemetry::{Context, KeyValue, sdk::trace as sdktrace, trace::Tracer};
 use opentelemetry::{Key, global};
@@ -23,7 +24,12 @@ use std::{
     path::PathBuf,
     str::FromStr,
 };
-use tracing_subscriber::prelude::*;
+
+#[derive(Deserialize, Debug)]
+struct TelemetryConfig {
+    TELEMETRY: bool,
+    TELEM_URL: Option<String>,
+}
 
 #[derive(Debug, Deserialize, Serialize)]
 struct ApiResponse {
@@ -140,7 +146,6 @@ pub fn pull_policy_exec_histories(
         let span = cx.span();
         span.set_attribute(Key::new("Days").string(days.to_string().to_string()));
         loop {
-            tracing::info!("Starting Airlock Data Retrieval");
             f.seek(SeekFrom::Start(0)).unwrap();
             let execution_histories = tracer.in_span(checkpoint_number.to_string(), |cx| {
                 let results: ApiResponse = history_logging(
@@ -329,19 +334,48 @@ fn skipback(days: i64) -> ObjectId {
     ObjectId::parse_str(&objectid_hex).expect("Invalid ObjectId hex")
 }
 
-fn init_tracer() -> Result<sdktrace::Tracer, TraceError> {
+fn load_telemetry_config() -> TelemetryConfig {
-    opentelemetry_otlp::new_pipeline()
+    let cfg_path = get_base_directory().join("config\\user_config.json");
-        .tracing()
+    if !cfg_path.exists() {
-        .with_exporter(
+        return TelemetryConfig {
-            opentelemetry_otlp::new_exporter()
+            TELEMETRY: false,
-                .tonic()
+            TELEM_URL: None,
-                .with_endpoint("https://signoz.racooncity.org"),
+        };
-        )
+    }
-        .with_trace_config(
+    match fs::read_to_string(&cfg_path) {
-            sdktrace::config().with_resource(Resource::new(vec![KeyValue::new(
+        Ok(contents) => {
-                "service.name",
+            serde_json::from_str::<TelemetryConfig>(&contents).unwrap_or(TelemetryConfig {
-                "LoxideLibs",
+                TELEMETRY: false,
-            )])),
+                TELEM_URL: None,
-        )
+            })
-        .install_simple()
+        }
+        Err(_) => TelemetryConfig {
+            TELEMETRY: false,
+            TELEM_URL: None,
+        },
+    }
+}
+
+fn init_tracer() -> Result<Option<sdktrace::Tracer>, TraceError> {
+    let cfg = load_telemetry_config();
+    if !cfg.TELEMETRY {
+        global::set_tracer_provider(NoopTracerProvider::new());
+        println!("Telemetry Disabled by Config");
+        return Ok(None);
+    }
+    let endpoint = cfg.TELEM_URL.unwrap_or_default();
+    let tracer =
+        opentelemetry_otlp::new_pipeline()
+            .tracing()
+            .with_exporter(
+                opentelemetry_otlp::new_exporter()
+                    .tonic()
+                    .with_endpoint(endpoint),
+            )
+            .with_trace_config(sdktrace::config().with_resource(Resource::new(vec![
+                KeyValue::new("service.name", "LoxideLibs"),
+            ])))
+            .install_simple()
+            .unwrap();
+    Ok(Some(tracer))
 }

requirements.txt

@@ -11,4 +11,4 @@ urllib3==2.5.0
 pyperclip==1.11.0
 
 --extra-index-url https://git.racooncity.org/api/packages/brotoskyj/pypi/simple/
-airlock_libs==3.0.0
+airlock_libs==3.1.0