From fb6cd8dacf0a022c8b7a1bf1eb77e3f8b909c4ba Mon Sep 17 00:00:00 2001
From: Fuwn <fuwnzy@gmail.com>
Date: Tue, 4 Jan 2022 17:54:49 -0800
Subject: feat(main.rs): ratelimit (100 requests/minute)

---
 Cargo.toml  |  1 +
 src/main.rs | 11 ++++++++++-
 2 files changed, 11 insertions(+), 1 deletion(-)

diff --git a/Cargo.toml b/Cargo.toml
index e59b8fa..09bbc68 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -24,3 +24,4 @@ dotenv = "0.15.0"
 actix-web = { version = "3.3.2", features = ["rustls"] }
 qstring = "0.7.2"
 actix-cors = "0.5.4"
+actix-ratelimit = "0.3.1"
diff --git a/src/main.rs b/src/main.rs
index 6f07c84..cd689ed 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -15,9 +15,18 @@ pub mod utils;
 async fn main() -> std::io::Result<()> {
   dotenv::dotenv().ok();
 
-  actix_web::HttpServer::new(|| {
+  let store = actix_ratelimit::MemoryStore::new();
+
+  actix_web::HttpServer::new(move || {
     actix_web::App::new()
       .wrap(actix_cors::Cors::default().allow_any_origin())
+      .wrap(
+        actix_ratelimit::RateLimiter::new(
+          actix_ratelimit::MemoryStoreActor::from(store.clone()).start(),
+        )
+        .with_interval(std::time::Duration::from_secs(60))
+        .with_max_requests(100),
+      )
       .service(routes::index)
       .service(
         actix_web::web::scope("/api/v1")
-- 
cgit v1.2.3