feat(main.rs): ratelimit (100 requests/minute)

1 files changed, 10 insertions, 1 deletions

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")