sigoden/dufs
1
0
Fork 0
mirror of https://github.com/sigoden/dufs.git synced 2026-04-09 00:59:02 +03:00
Code Issues 8 Packages Projects Releases 10 Wiki Activity

feat: add logger

Browse Source
This commit is contained in:
sigoden
2022-05-27 09:01:16 +08:00
parent e6ea7e4f75
commit 63e6906393
7 changed files with 126 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
src/server.rs
View File

@@ -37,7 +37,7 @@ pub async fn serve(args: Args) -> BoxResult<()> {
async {
Ok::<_, Infallible>(service_fn(move |req| {
let inner = inner.clone();
inner.handle(req)
inner.call(req)
}))
}
});
@@ -59,7 +59,18 @@ impl InnerService {
Self { args }
}
pub async fn handle(self: Arc<Self>, req: Request) -> Result<Response, hyper::Error> {
pub async fn call(self: Arc<Self>, req: Request) -> Result<Response, hyper::Error> {
let method = req.method().clone();
let uri = req.uri().clone();
let res = self
.handle(req)
.await
.unwrap_or_else(|_| status_code!(StatusCode::INTERNAL_SERVER_ERROR));
info!(r#""{} {}" - {}"#, method, uri, res.status());
Ok(res)
}
pub async fn handle(self: Arc<Self>, req: Request) -> BoxResult<Response> {
if !self.auth_guard(&req).unwrap_or_default() {
let mut res = status_code!(StatusCode::UNAUTHORIZED);
res.headers_mut()
@@ -67,7 +78,7 @@ impl InnerService {
return Ok(res);
}
let res = if req.method() == Method::GET {
if req.method() == Method::GET {
self.handle_static(req).await
} else if req.method() == Method::PUT {
if self.args.readonly {
@@ -78,8 +89,7 @@ impl InnerService {
self.handle_delete(req).await
} else {
return Ok(status_code!(StatusCode::NOT_FOUND));
};
Ok(res.unwrap_or_else(|_| status_code!(StatusCode::INTERNAL_SERVER_ERROR)))
}
}
async fn handle_static(&self, req: Request) -> BoxResult<Response> {