rate limit (#145)

* rate limit * clippy * clippy * update * cleanup * make rate_limit an extension * change config * fix * update test * reduce default jitter * Update config.yml --------- Co-authored-by: Xiliang Chen <[email protected]>
AcalaNetwork · Nov 27, 2023 · 43cbd17 · 43cbd17
1 parent 158538e
commit 43cbd17
Show file tree

Hide file tree

Showing 8 changed files with 261 additions and 25 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -43,8 +43,8 @@ tracing = "0.1.37"
 tracing-serde = "0.1.3"
 tracing-subscriber = { version = "0.3.16", features = ["env-filter", "json"] }
 
-jsonrpsee = { version = "0.20.0", path = "./vendor/jsonrpsee/jsonrpsee", features = ["full"] }
-governor = { version = "0.6.0", path = "./vendor/governor/governor" }
+jsonrpsee = { path = "./vendor/jsonrpsee/jsonrpsee", features = ["full"] }
+governor = "0.6.0"
 
 [dev-dependencies]
 criterion = { version = "0.5.1", features = ["async_tokio", "html_reports"] }

diff --git a/config.yml b/config.yml
@@ -23,6 +23,9 @@ extensions:
       - path: /liveness
         method: chain_getBlockHash
     cors: all
+  rate_limit: # 20 requests per second per connection
+    burst: 20
+    period_secs: 1
 
 middlewares:
   methods:

diff --git a/src/extensions/mod.rs b/src/extensions/mod.rs
@@ -13,6 +13,7 @@ pub mod cache;
 pub mod client;
 pub mod event_bus;
 pub mod merge_subscription;
+pub mod rate_limit;
 pub mod server;
 pub mod telemetry;
 
@@ -136,4 +137,5 @@ define_all_extensions! {
     eth_api: api::EthApi,
     server: server::SubwayServerBuilder,
     event_bus: event_bus::EventBus,
+    rate_limit: rate_limit::RateLimitBuilder,
 }