Skip to content

Commit a76dbbc

Browse files
committed
chore(cubestore): Respect telemetry env variable
1 parent 41088a5 commit a76dbbc

File tree

1 file changed

+14
-2
lines changed

1 file changed

+14
-2
lines changed

rust/cubestore/cubestore/src/bin/cubestored.rs

Lines changed: 14 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,10 @@
1-
use cubestore::app_metrics;
21
use cubestore::config::{validate_config, Config, CubeServices};
32
use cubestore::http::status::serve_status_probes;
43
use cubestore::telemetry::{init_agent_sender, track_event};
54
use cubestore::util::logger::init_cube_logger;
65
use cubestore::util::metrics::init_metrics;
76
use cubestore::util::{metrics, spawn_malloc_trim_loop};
7+
use cubestore::{app_metrics, CubeError};
88
use datafusion::cube_ext;
99
use log::debug;
1010
use serde_json::Value;
@@ -32,7 +32,19 @@ fn main() {
3232
Err(_) => metrics::Compatibility::StatsD,
3333
};
3434
init_metrics("127.0.0.1:0", "127.0.0.1:8125", metrics_mode);
35-
init_cube_logger(true);
35+
let telemetry_env = std::env::var("CUBESTORE_TELEMETRY")
36+
.or(std::env::var("CUBEJS_TELEMETRY"))
37+
.unwrap_or("true".to_string());
38+
let enable_telemetry = telemetry_env
39+
.parse::<bool>()
40+
.map_err(|e| {
41+
CubeError::user(format!(
42+
"Can't parse telemetry env variable '{}': {}",
43+
telemetry_env, e
44+
))
45+
})
46+
.unwrap();
47+
init_cube_logger(enable_telemetry);
3648

3749
log::info!("Cube Store version {}", version);
3850

0 commit comments

Comments
 (0)