feat(rpc): add max connections config

alchemyplatform · Jul 26, 2023 · 702b9fd · 702b9fd
1 parent a29ff9f
commit 702b9fd
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/src/cli/rpc.rs b/src/cli/rpc.rs
@@ -51,6 +51,15 @@ pub struct RpcArgs {
         default_value = "20"
     )]
     timeout_seconds: String,
+
+    /// Maximum number of concurrent connections
+    #[arg(
+        long = "rpc.max_connections",
+        name = "rpc.max_connections",
+        env = "RPC_MAX_CONNECTIONS",
+        default_value = "100"
+    )]
+    max_connections: u32,
 }
 
 impl RpcArgs {
@@ -99,6 +108,7 @@ impl RpcArgs {
             sim_settings,
             estimation_settings,
             rpc_timeout: Duration::from_secs(self.timeout_seconds.parse()?),
+            max_connections: self.max_connections,
             mempool_configs,
         })
     }

diff --git a/src/rpc/task.rs b/src/rpc/task.rs
@@ -51,6 +51,7 @@ pub struct Args {
     pub sim_settings: simulation::Settings,
     pub estimation_settings: estimation::Settings,
     pub rpc_timeout: Duration,
+    pub max_connections: u32,
     pub mempool_configs: HashMap<H256, MempoolConfig>,
 }
 
@@ -146,6 +147,7 @@ impl Task for RpcTask {
         let server = ServerBuilder::default()
             .set_logger(RpcMetricsLogger)
             .set_middleware(service_builder)
+            .max_connections(self.args.max_connections)
             .http_only()
             .build(addr)
             .await?;