obfuscation is now turnoffable

README.md

@@ -29,13 +29,15 @@ API_KEY = "App1Secret"
 BEAM_APP_ID_LONG = "app1.broker.example.com"
 ```
 
-optional variable
+### Optional variables
+
 ```bash
 RETRY_COUNT = "32"
 ```
 
 The default maximum number of retries for beam and blaze healthchecks is 32. 
 
+To disable the count obfuscation, set the env. variable `DO_NOT_OBFUSCATE`.
 
 Optionally, you can provide the `TLS_CA_CERTIFICATES_DIR` environment variable to add additional trusted certificates, e.g., if you have a TLS-terminating proxy server in place. The application respects the `HTTP_PROXY`, `HTTPS_PROXY`, `ALL_PROXY`, `NO_PROXY`, and their respective lowercase equivalents.
 

src/config.rs

@@ -42,6 +42,10 @@ struct CliArgs {
     #[clap(long, env, value_parser)]
     blaze_url: Uri,
 
+    /// Should the results not be obfuscated
+    #[clap(long, env, value_parser)]
+    do_not_obfuscate: bool,
+
     /// Outgoing HTTP proxy: Directory with CA certificates to trust for TLS connections (e.g. /etc/samply/cacerts/)
     #[clap(long, env, value_parser)]
     tls_ca_certificates_dir: Option<PathBuf>,
@@ -53,6 +57,7 @@ pub(crate) struct Config {
     pub api_key: String,
     pub retry_count: usize,
     pub blaze_url: Uri,
+    pub do_not_obfuscate: bool,
     tls_ca_certificates: Vec<Certificate>,
     pub client: Client,
 }
@@ -71,6 +76,7 @@ impl Config {
             api_key: cli_args.api_key,
             retry_count: cli_args.retry_count,
             blaze_url: cli_args.blaze_url,
+            do_not_obfuscate: cli_args.do_not_obfuscate,
             tls_ca_certificates,
             client
         };

src/main.rs

@@ -167,9 +167,12 @@ async fn run_cql_query(task: &BeamTask, query: &Query, obf_cache: &mut ObfCache)
         }
     };
 
-    let cql_result_new = obfuscate_counts_mr(&cql_result, obf_cache);
+    let cql_result_new = match CONFIG.do_not_obfuscate {
+        false => obfuscate_counts_mr(&cql_result, obf_cache)?,
+        true => cql_result
+    };
 
-    let result = beam_result(task.to_owned(), cql_result_new?
+    let result = beam_result(task.to_owned(), cql_result_new
     .to_string()).unwrap_or_else(|e| {
         err.body = e.to_string();
         return err;