rust/greptimedb
1
0
Fork 0
mirror of https://github.com/GreptimeTeam/greptimedb.git synced 2026-05-17 13:30:38 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(servers): make http timeout and body limit optional (#4217)

Browse Source 
* feat(servers): make http timeout and body limit optional

Signed-off-by: Ruihang Xia <waynestxia@gmail.com>

* add comment

Signed-off-by: Ruihang Xia <waynestxia@gmail.com>

* chore: make config-docs

---------

Signed-off-by: Ruihang Xia <waynestxia@gmail.com>
Co-authored-by: Ruihang Xia <waynestxia@gmail.com>
This commit is contained in:
Jeremyhi
2024-06-26 14:14:14 +08:00
committed by GitHub
parent a779cb36ec
commit df0fff2f2c
4 changed files with 28 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
src/servers/src/http.rs
View File

@@ -670,20 +670,30 @@ impl HttpServer {
}
pub fn build(&self, router: Router) -> Router {
let timeout_layer = if self.options.timeout != Duration::default() {
Some(ServiceBuilder::new().layer(TimeoutLayer::new(self.options.timeout)))
} else {
info!("HTTP server timeout is disabled");
None
};
let body_limit_layer = if self.options.body_limit != ReadableSize(0) {
Some(
ServiceBuilder::new()
.layer(DefaultBodyLimit::max(self.options.body_limit.0 as usize)),
)
} else {
info!("HTTP server body limit is disabled");
None
};
router
// middlewares
.layer(
ServiceBuilder::new()
.layer(HandleErrorLayer::new(handle_error))
.layer(TraceLayer::new_for_http())
.layer(TimeoutLayer::new(self.options.timeout))
.layer(DefaultBodyLimit::max(
self.options
.body_limit
.0
.try_into()
.unwrap_or_else(|_| DEFAULT_BODY_LIMIT.as_bytes() as usize),
))
.option_layer(timeout_layer)
.option_layer(body_limit_layer)
// auth layer
.layer(middleware::from_fn_with_state(
AuthState::new(self.user_provider.clone()),