refactor: extract tracing setup to logger::setup()

Author: wangeguo

src/lib.rs

@@ -17,6 +17,7 @@ pub mod config;
 pub mod context;
 pub mod errors;
 pub mod handlers;
+pub mod logger;
 pub mod requests;
 pub mod responses;
 pub mod routes;

src/logger.rs

@@ -0,0 +1,22 @@
+// Copyright (c) wangeguo. All rights reserved.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+use tracing::metadata::LevelFilter;
+use tracing_subscriber::EnvFilter;
+
+pub fn setup() {
+    let filter =
+        EnvFilter::builder().with_default_directive(LevelFilter::DEBUG.into()).from_env_lossy();
+    tracing_subscriber::fmt().with_env_filter(filter).init();
+}

src/main.rs

@@ -15,15 +15,11 @@
 use std::sync::Arc;
 
 use clap::Parser;
-use codecraft::{app, config::Config, context::Context};
-use tracing::metadata::LevelFilter;
-use tracing_subscriber::EnvFilter;
+use codecraft::{app, config::Config, context::Context, logger};
 
 #[tokio::main]
 async fn main() -> anyhow::Result<()> {
-    let filter =
-        EnvFilter::builder().with_default_directive(LevelFilter::DEBUG.into()).from_env_lossy();
-    tracing_subscriber::fmt().with_env_filter(filter).init();
+    logger::setup();
 
     // This returns an error if the `.env` file doesn't exist, but that's not what we want
     // since we're not going to use a `.env` file if we deploy this application.