chore(query): keep queries detail and profile log format (#16822)

databendlabs · Nov 12, 2024 · aca2fe4 · aca2fe4
1 parent 5e2095c
commit aca2fe4
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/src/common/tracing/src/init.rs b/src/common/tracing/src/init.rs
@@ -270,7 +270,7 @@ pub fn init_logging(
                     EnvFilterBuilder::new()
                         .filter(Some("databend::log::query"), LevelFilter::Trace),
                 ))
-                .append(query_log_file);
+                .append(query_log_file.with_layout(get_layout("identical")));
             logger = logger.dispatch(dispatch);
         }
         if let Some(endpoint) = &cfg.query.otlp {
@@ -312,7 +312,7 @@ pub fn init_logging(
                     EnvFilterBuilder::new()
                         .filter(Some("databend::log::profile"), LevelFilter::Trace),
                 ))
-                .append(profile_log_file);
+                .append(profile_log_file.with_layout(get_layout("identical")));
             logger = logger.dispatch(dispatch);
         }
         if let Some(endpoint) = &cfg.profile.otlp {

diff --git a/src/common/tracing/src/loggers.rs b/src/common/tracing/src/loggers.rs
@@ -49,10 +49,15 @@ pub fn get_layout(format: &str) -> Layout {
     match format {
         "text" => text_layout(),
         "json" => json_layout(),
+        "identical" => identical_layout(),
         _ => unimplemented!("file logging format {format} is not supported"),
     }
 }
 
+fn identical_layout() -> Layout {
+    CustomLayout::new(|record: &Record| Ok(format!("{}\n", record.args()).into_bytes())).into()
+}
+
 fn text_layout() -> Layout {
     CustomLayout::new(|record: &Record| {
         let s = match ThreadTracker::query_id() {