feat: add compression support to metrics and RPC servers (#4906)

ChainSafe · Oct 17, 2024 · efca75f · efca75f
1 parent 01589d2
commit efca75f
Show file tree

Hide file tree

Showing 4 changed files with 109 additions and 43 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -184,7 +184,8 @@ tokio = { version = "1", features = ['full'] }
 tokio-stream = { version = "0.1", features = ["fs", "io-util"] }
 tokio-util = { version = "0.7", features = ["compat", "io-util"] }
 toml = "0.8"
-tower = { version = "0.5", features = ["full"] }
+tower = { version = "0.4", features = ["util"] }
+tower-http = { version = "0.6", features = ["compression-full", "sensitive-headers"] }
 tracing = "0.1"
 tracing-appender = "0.2"
 tracing-subscriber = { version = "0.3", features = ["env-filter", "json"] }

diff --git a/src/metrics/mod.rs b/src/metrics/mod.rs
@@ -18,6 +18,7 @@ use prometheus_client::{
 use std::sync::Arc;
 use std::{path::PathBuf, time::Instant};
 use tokio::net::TcpListener;
+use tower_http::compression::CompressionLayer;
 use tracing::warn;
 
 static DEFAULT_REGISTRY: Lazy<RwLock<prometheus_client::registry::Registry>> =
@@ -91,6 +92,7 @@ where
     let app = Router::new()
         .route("/metrics", get(collect_prometheus_metrics))
         .route("/stats/db", get(collect_db_metrics::<DB>))
+        .layer(CompressionLayer::new())
         .with_state(db);
 
     // Wait for server to exit