From 367dbbb3b256227c8e1c235c292d8dbd0263f78e Mon Sep 17 00:00:00 2001 From: Brandon Thomas Date: Tue, 21 Jun 2022 02:46:17 -0400 Subject: [PATCH] Disable newrelic flag --- .../storyteller_web/src/tts_inference_job.rs | 16 ++++++++++++---- 1 file changed, 12 insertions(+), 4 deletions(-) diff --git a/crates/service/storyteller_web/src/tts_inference_job.rs b/crates/service/storyteller_web/src/tts_inference_job.rs index e1d57e30e6..76aa83b44b 100644 --- a/crates/service/storyteller_web/src/tts_inference_job.rs +++ b/crates/service/storyteller_web/src/tts_inference_job.rs @@ -114,6 +114,8 @@ struct Inferencer { pub newrelic_client: NewRelicClient, + pub newrelic_disabled: bool, + pub worker_details: InferencerWorkerDetails, // Keep tabs of which models to hold in the sidecar memory with this virtual LRU cache @@ -310,6 +312,9 @@ async fn main() -> AnyhowResult<()> { }; let license_key = easyenv::get_env_string_required("NEWRELIC_API_KEY")?; + + let newrelic_disabled = easyenv::get_env_bool_or_default("IS_NEWRELIC_DISABLED", false); + let newrelic_client = ClientBuilder::new(&license_key).build().unwrap(); let maybe_minimum_priority = easyenv::get_env_string_optional("MAYBE_MINIMUM_PRIORITY") @@ -329,6 +334,7 @@ async fn main() -> AnyhowResult<()> { tts_inference_command, tts_inference_sidecar_client, newrelic_client, + newrelic_disabled, worker_details: InferencerWorkerDetails { is_on_prem, worker_hostname: server_hostname.clone(), @@ -431,11 +437,13 @@ async fn main_loop(inferencer: Inferencer) { } }; - span_batch.append(&mut spans); + if !inferencer.newrelic_disabled { + span_batch.append(&mut spans); - if span_batch.len() > 50 { - let spans_to_send = span_batch.split_off(0).into(); - inferencer.newrelic_client.send_spans(spans_to_send).await; + if span_batch.len() > 50 { + let spans_to_send = span_batch.split_off(0).into(); + inferencer.newrelic_client.send_spans(spans_to_send).await; + } } error_timeout_millis = START_TIMEOUT_MILLIS; // reset