gburri/recipes
1
0
Fork 0
Code Issues 6 Pull requests Projects Releases Packages Wiki Activity Actions

Limit the number of current request for a certain amount of time for some endpoints.

Browse source
This commit is contained in:
Greg Burri 2025-03-03 10:10:55 +01:00
parent b2572ebfe5
commit b40ee5f765
4 changed files with 34 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

40
backend/src/main.rs
View file

@ -1,7 +1,8 @@
use std::{net::SocketAddr, path::Path};
use axum::{
Router,
BoxError, Router,
error_handling::HandleErrorLayer,
extract::{ConnectInfo, Extension, FromRef, Request, State},
http::StatusCode,
middleware::{self, Next},
@ -13,6 +14,7 @@ use chrono::prelude::*;
use clap::Parser;
use config::Config;
use itertools::Itertools;
use tower::{ServiceBuilder, buffer::BufferLayer, limit::RateLimitLayer};
use tower_http::{
services::{ServeDir, ServeFile},
trace::TraceLayer,
@ -226,23 +228,38 @@ async fn main() {
get(services::fragments::recipes_list_fragments),
);
let html_routes_with_rate_limit = Router::new()
.route("/signin", post(services::user::sign_in_post))
.route("/signup", post(services::user::sign_up_post))
.route(
"/ask_reset_password",
post(services::user::ask_reset_password_post),
)
.layer(
ServiceBuilder::new()
.layer(HandleErrorLayer::new(|err: BoxError| async move {
(
StatusCode::INTERNAL_SERVER_ERROR,
format!("Unhandled error: {}", err),
)
}))
.layer(BufferLayer::new(1024))
.layer(RateLimitLayer::new(
consts::NUMBER_OF_CONCURRENT_HTTP_REQUEST_FOR_RATE_LIMIT,
consts::DURATION_FOR_RATE_LIMIT,
)),
);
let html_routes = Router::new()
.route("/", get(services::home_page))
.route(
"/signup",
get(services::user::sign_up_get).post(services::user::sign_up_post),
)
.route("/signup", get(services::user::sign_up_get))
.route("/validation", get(services::user::sign_up_validation))
.route("/revalidation", get(services::user::email_revalidation))
.route(
"/signin",
get(services::user::sign_in_get).post(services::user::sign_in_post),
)
.route("/signin", get(services::user::sign_in_get))
.route("/signout", get(services::user::sign_out))
.route(
"/ask_reset_password",
get(services::user::ask_reset_password_get)
.post(services::user::ask_reset_password_post),
get(services::user::ask_reset_password_get),
)
.route(
"/reset_password",
@ -257,6 +274,7 @@ async fn main() {
"/user/edit",
get(services::user::edit_user_get).post(services::user::edit_user_post),
)
.merge(html_routes_with_rate_limit)
.nest("/fragments", fragments_routes)
.route_layer(middleware::from_fn(services::ron_error_to_html));