map target

Author: forsaken628

src/common/tracing/src/loggers.rs

@@ -123,24 +123,54 @@ impl<const FIXED_TIME: bool> Layout for TextLayout<FIXED_TIME> {
             PRINTER.print_timestamp_with_offset(&timestamp, offset, &mut buf)?;
         }
 
-        write!(
-            buf,
-            " {:>5} {}: {}:{} {}",
-            record.level(),
-            record.module_path().unwrap_or(""),
-            Path::new(record.file().unwrap_or_default())
-                .file_name()
-                .and_then(|name| name.to_str())
-                .unwrap_or_default(),
-            record.line().unwrap_or(0),
-            record.args(),
-        )?;
+        let level = record.level();
+        let module = map_module(record.module_path().unwrap_or(""));
+        let log_file = Path::new(record.file().unwrap_or_default())
+            .file_name()
+            .and_then(|name| name.to_str())
+            .unwrap_or_default();
+        let line = record.line().unwrap_or(0);
+        let msg = record.args();
+        write!(buf, " {level:>5} {module}: {log_file}:{line} {msg}",)?;
         record.key_values().visit(&mut KvWriter(&mut buf))?;
 
         Ok(buf)
     }
 }
 
+fn map_module(full_name: &str) -> &str {
+    if full_name.starts_with("databend_query") {
+        if full_name.starts_with("databend_query::sessions::query_ctx") {
+            return "[QUERY-CTX]";
+        }
+        if full_name.starts_with("databend_query::sessions::queue_mgr") {
+            return "[QUERY-QUEUE]";
+        }
+        if full_name.starts_with("databend_query::interpreters::interpreter") {
+            return "[INTERPRETER]";
+        }
+        if full_name.starts_with("databend_query::servers::http::v1::query") {
+            return "[HTTP-QUERY]";
+        }
+        if full_name.starts_with("databend_query::pipelines::builders") {
+            return "[BUILD-PIPELINES]";
+        }
+        if full_name.starts_with("databend_query::pipelines::executor::query_pipeline_executor") {
+            return "[PIPELINE-EXECUTOR]";
+        }
+        if full_name.starts_with("databend_query::pipelines::processors::transforms") {
+            return "[TRANSFORMS]";
+        }
+    }
+    if full_name.starts_with("databend_common_pipeline_transforms") {
+        return "[TRANSFORMS]";
+    }
+    if full_name.starts_with("databend_common_sql::planner::planner") {
+        return "[SQL-PLANNER]";
+    }
+    full_name
+}
+
 #[derive(Debug)]
 pub struct JsonLayout<const FIXED_TIME: bool = false>;