diff options
| -rw-r--r-- | Cargo.toml | 1 | ||||
| -rw-r--r-- | src/main.rs | 11 |
2 files changed, 11 insertions, 1 deletions
@@ -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") |