use custome log

2020-04-18 02:13:14 +08:00
parent 4ac22d6800
commit ef06866d10
1 changed files with 29 additions and 1 deletions
--- a/hyperrs/src/main.rs
+++ b/hyperrs/src/main.rs
@@ -2,6 +2,32 @@ use std::convert::Infallible;
 use hyper::service::{ make_service_fn, service_fn, };
 use hyper::{ Body, Request, Response, Server, };

+use log::{Record, Level, Metadata};
+use log::{SetLoggerError, LevelFilter};
+
+struct SimpleLogger;
+
+impl log::Log for SimpleLogger {
+    fn enabled(&self, metadata: &Metadata) -> bool {
+        metadata.level() <= Level::Info
+    }
+
+    fn log(&self, record: &Record) {
+        if self.enabled(record.metadata()) {
+            println!("{} - {}", record.level(), record.args());
+        }
+    }
+
+    fn flush(&self) {}
+}
+
+static LOGGER: SimpleLogger = SimpleLogger;
+
+pub fn init() -> Result<(), SetLoggerError> {
+    log::set_logger(&LOGGER).map(|()| log::set_max_level(LevelFilter::Info))
+}
+
+
 async fn hello(_: Request<Body>) -> Result<Response<Body>, Infallible> {
    Ok(Response::new(Body::from("Hello World!\n")))
 }
@@ -9,8 +35,10 @@ async fn hello(_: Request<Body>) -> Result<Response<Body>, Infallible> {
 // https://github.com/hyperium/hyper/blob/master/examples/hello.rs
 #[tokio::main]
 pub async fn main() -> Result<(), Box<dyn std::error::Error + Send + Sync>> {
-    pretty_env_logger::init();
+    // pretty_env_logger::init();
+    init()?;

+    println!("{}", log::STATIC_MAX_LEVEL);
    log::info!("hello hyper.rs");
    log::warn!("hello hyper.rs");
    log::error!("hello hyper.rs");