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

2
Cargo.lock generated
View file

@ -258,7 +258,7 @@ dependencies = [
[[package]]
name = "duf"
version = "0.2.0"
version = "0.2.1"
dependencies = [
"async-walkdir",
"async_zip",

View file

@ -1,6 +1,6 @@
[package]
name = "duf"
version = "0.2.0"
version = "0.2.1"
edition = "2021"
authors = ["sigoden <sigoden@gmail.com>"]
description = "Duf is a simple file server."

View file

@ -49,8 +49,6 @@ duf --static
Finally, run this command to see a list of all available option
### Curl
Download a file

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);