[ISSUE #2616]🚀Add Level for log setting🔥 (#2617)

Author: mxsm

rocketmq-common/src/log.rs

@@ -14,9 +14,10 @@
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */
+use std::fmt;
 use std::str::FromStr;
 
-use tracing::level_filters::LevelFilter;
+use tracing::metadata::LevelFilter;
 
 /// Initializes the logger with the specified configuration.
 ///
@@ -31,6 +32,87 @@ pub fn init_logger() {
         .with_level(true)
         .with_line_number(true)
         .with_thread_ids(true)
-        .with_max_level(LevelFilter::from_str(info_level.as_str()).unwrap())
+        .with_max_level(tracing::Level::from_str(info_level.as_str()).expect("Invalid log level"))
         .init();
 }
+
+pub fn init_logger_with_level(level: Level) {
+    tracing_subscriber::fmt()
+        .with_thread_names(true)
+        .with_level(true)
+        .with_line_number(true)
+        .with_thread_ids(true)
+        .with_max_level(tracing::Level::from_str(level.as_str()).expect("Invalid log level"))
+        .init();
+}
+
+#[derive(Clone, Debug, PartialEq, Eq, Hash)]
+pub struct Level(&'static str);
+
+impl Level {
+    /// Constant representing the ERROR log level.
+    pub const ERROR: Level = Level("ERROR");
+
+    /// Constant representing the WARN log level.
+    pub const WARN: Level = Level("WARN");
+
+    /// Constant representing the INFO log level.
+    pub const INFO: Level = Level("INFO");
+
+    /// Constant representing the DEBUG log level.
+    pub const DEBUG: Level = Level("DEBUG");
+
+    /// Constant representing the TRACE log level.
+    pub const TRACE: Level = Level("TRACE");
+
+    pub fn as_str(&self) -> &'static str {
+        self.0
+    }
+}
+
+impl From<&'static str> for Level {
+    fn from(level: &'static str) -> Self {
+        match level {
+            "ERROR" | "WARN" | "INFO" | "DEBUG" | "TRACE" => Level(level),
+            _ => panic!("Invalid log level: {}", level),
+        }
+    }
+}
+
+impl fmt::Display for Level {
+    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
+        f.pad(self.0)
+    }
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn level_as_str_returns_correct_value() {
+        assert_eq!(Level::ERROR.as_str(), "ERROR");
+        assert_eq!(Level::WARN.as_str(), "WARN");
+        assert_eq!(Level::INFO.as_str(), "INFO");
+        assert_eq!(Level::DEBUG.as_str(), "DEBUG");
+        assert_eq!(Level::TRACE.as_str(), "TRACE");
+    }
+
+    #[test]
+    fn level_from_str_creates_correct_level() {
+        assert_eq!(Level::from("ERROR"), Level::ERROR);
+        assert_eq!(Level::from("WARN"), Level::WARN);
+        assert_eq!(Level::from("INFO"), Level::INFO);
+        assert_eq!(Level::from("DEBUG"), Level::DEBUG);
+        assert_eq!(Level::from("TRACE"), Level::TRACE);
+    }
+
+    #[test]
+    fn level_display_formats_correctly() {
+        assert_eq!(format!("{}", Level::ERROR), "ERROR");
+        assert_eq!(format!("{}", Level::WARN), "WARN");
+        assert_eq!(format!("{}", Level::INFO), "INFO");
+        assert_eq!(format!("{}", Level::DEBUG), "DEBUG");
+        assert_eq!(format!("{}", Level::TRACE), "TRACE");
+    }
+}