tornadocontrib/ web3-proxy
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add timeout as a the tower layer

This commit is contained in:
Bryan Stitt 2023-07-19 18:41:26 -07:00
parent c77f1e04b1
commit e6dc64f369
1 changed files with 15 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
web3_proxy/src/frontend/mod.rs
View File

@ -13,8 +13,9 @@ pub mod users;
use crate::app::Web3ProxyApp;
use crate::errors::Web3ProxyResult;
use axum::{
error_handling::HandleErrorLayer,
routing::{get, post},
Extension, Router,
BoxError, Extension, Router,
};
use http::{header::AUTHORIZATION, Request, StatusCode};
use hyper::Body;
@ -25,6 +26,8 @@ use std::{iter::once, time::Duration};
use std::{net::SocketAddr, sync::atomic::Ordering};
use strum::{EnumCount, EnumIter};
use tokio::sync::broadcast;
use tower::timeout::TimeoutLayer;
use tower::ServiceBuilder;
use tower_http::sensitive_headers::SetSensitiveRequestHeadersLayer;
use tower_http::{cors::CorsLayer, trace::TraceLayer};
use tracing::{error_span, info};
@ -261,6 +264,17 @@ pub async fn serve(
.layer(Extension(app.clone()))
// frontend caches
.layer(Extension(Arc::new(response_cache)))
// request timeout
.layer(
ServiceBuilder::new()
// this middleware goes above `TimeoutLayer` because it will receive
// errors returned by `TimeoutLayer`
// TODO: JsonRPC error response
.layer(HandleErrorLayer::new(|_: BoxError| async {
StatusCode::REQUEST_TIMEOUT
}))
.layer(TimeoutLayer::new(Duration::from_secs(5 * 60))),
)
// request id
.layer(
TraceLayer::new_for_http().make_span_with(|request: &Request<Body>| {