From 2b6331ab35b55ceec609ebd13b2c6b1f9a7c07d2 Mon Sep 17 00:00:00 2001
From: Enola Knezevic <enola.knezevic@dkfz-heidelberg.com>
Date: Tue, 23 Apr 2024 14:36:37 +0200
Subject: [PATCH] obf "patient" stratifier, some logs info

---
 src/beam.rs            | 4 ++--
 src/blaze.rs           | 4 ++--
 src/task_processing.rs | 4 ++--
 src/util.rs            | 2 +-
 4 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/src/beam.rs b/src/beam.rs
index d67ef9f..9c2f3c6 100644
--- a/src/beam.rs
+++ b/src/beam.rs
@@ -4,7 +4,7 @@ use beam_lib::{TaskResult, BeamClient, BlockingOptions, MsgId, TaskRequest, RawS
 use http::StatusCode;
 use once_cell::sync::Lazy;
 use serde::Serialize;
-use tracing::{debug, warn};
+use tracing::{debug, warn, info};
 
 use crate::{config::CONFIG, errors::FocusError};
 
@@ -87,7 +87,7 @@ pub async fn retrieve_tasks() -> Result<Vec<TaskRequest<String>>, FocusError> {
 }
 
 pub async fn answer_task<T: Serialize + 'static>(result: &TaskResult<T>) -> Result<(), FocusError> {
-    debug!("Answer task with id: {}", result.task);
+    info!("Answer task with id: {}", result.task);
     BEAM_CLIENT.put_result(result, &result.task)
     .await
     .map(|_| ())
diff --git a/src/blaze.rs b/src/blaze.rs
index 6e5f310..3d1c885 100644
--- a/src/blaze.rs
+++ b/src/blaze.rs
@@ -2,7 +2,7 @@ use http::StatusCode;
 use serde::Deserialize;
 use serde::Serialize;
 use serde_json::Value;
-use tracing::{debug, warn};
+use tracing::{debug, warn, info};
 
 use crate::BeamTask;
 use crate::errors::FocusError;
@@ -93,7 +93,7 @@ pub async fn evaluate_measure(url: String) -> Result<String, FocusError> {
         .map_err(FocusError::MeasureEvaluationErrorReqwest)?;
 
     if resp.status() == StatusCode::OK {
-        debug!(
+        info!(
             "Successfully evaluated the Measure with canonical URL: {}",
             url
         );
diff --git a/src/task_processing.rs b/src/task_processing.rs
index dddaa1f..fc5f703 100644
--- a/src/task_processing.rs
+++ b/src/task_processing.rs
@@ -3,7 +3,7 @@ use std::{sync::Arc, collections::HashMap, time::Duration};
 use base64::{engine::general_purpose, Engine as _};
 use laplace_rs::ObfCache;
 use tokio::sync::{mpsc, Semaphore, Mutex};
-use tracing::{error, warn, debug, Instrument, info_span};
+use tracing::{error, warn, debug, info, Instrument, info_span};
 
 use crate::{ReportCache, errors::FocusError, beam, BeamTask, BeamResult, run_exporter_query, config::{EndpointType, CONFIG}, run_cql_query, intermediate_rep, ast, run_intermediate_rep_query, Metadata, blaze::parse_blaze_query, util};
 
@@ -91,7 +91,7 @@ async fn process_task(
     obf_cache: Arc<Mutex<ObfCache>>,
     report_cache: Arc<Mutex<ReportCache>>,
 ) -> Result<BeamResult, FocusError> {
-    debug!("Processing task {}", task.id);
+    info!("Processing task {}", task.id);
 
     let metadata: Metadata = serde_json::from_value(task.metadata.clone()).unwrap_or(Metadata {
         project: "default_obfuscation".to_string(),
diff --git a/src/util.rs b/src/util.rs
index 26960c5..4f763bb 100644
--- a/src/util.rs
+++ b/src/util.rs
@@ -130,7 +130,7 @@ pub fn obfuscate_counts_mr(
         .map_err(|e| FocusError::DeserializationError(format!(r#"{}. Is obfuscation turned on when it shouldn't be? Is the metadata in the task formatted correctly, like this {{"project": "name"}}? Are there any other projects stated in the projects_no_obfuscation parameter in the bridgehead?"#, e)))?;
     for g in &mut measure_report.group {
         match &g.code.text[..] {
-            "patients" => {
+            "patient" | "patients" => { // Prism used "patient" for catalogue, Lens uses "patients"
                 obfuscate_counts_recursive(
                     &mut g.population,
                     delta_patient,