feat: aware RUST_LOG

This commit is contained in:
sigoden
2022-05-28 20:42:32 +08:00
parent d9a917176a
commit 3673a64ec7
5 changed files with 14 additions and 12 deletions

View File

@@ -25,14 +25,18 @@ async fn main() {
async fn run() -> BoxResult<()> {
let args = Args::parse(matches())?;
let level = if args.log {
LevelFilter::Info
if std::env::var("RUST_LOG").is_ok() {
simple_logger::init()?;
} else {
LevelFilter::Error
};
simple_logger::SimpleLogger::default()
.with_level(level)
.init()?;
let level = if args.log {
LevelFilter::Info
} else {
LevelFilter::Error
};
simple_logger::SimpleLogger::default()
.with_level(level)
.init()?;
}
serve(args).await
}

View File

@@ -187,7 +187,7 @@ impl InnerService {
}
async fn handle_send_dir_zip(&self, path: &Path) -> BoxResult<Response> {
let (mut writer, reader) = tokio::io::duplex(65536);
let (mut writer, reader) = tokio::io::duplex(1024 * 1024 * 20);
dir_zip(&mut writer, path).await?;
let stream = ReaderStream::new(reader);
let body = Body::wrap_stream(stream);