alamin/websurfx
alamin/websurfx
0
0
Fork 0
mirror of https://github.com/neon-mmd/websurfx.git synced 2024-11-25 15:38:21 -05:00
Code Issues Releases Activity

✨ feat: pass the new config option into the middleware config (#203)

Browse Source
This commit is contained in:
neon_arch 2023-09-02 20:22:24 +03:00
parent 8cba040d80
commit 51937a0d49
1 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
src/lib.rs
View File

@ -14,6 +14,7 @@ use crate::server::routes;
use actix_cors::Cors; use actix_cors::Cors;
use actix_files as fs; use actix_files as fs;
use actix_governor::{Governor, GovernorConfigBuilder};
use actix_web::{dev::Server, http::header, middleware::Logger, web, App, HttpServer}; use actix_web::{dev::Server, http::header, middleware::Logger, web, App, HttpServer};
use config::parser::Config; use config::parser::Config;
use handlebars::Handlebars; use handlebars::Handlebars;
@ -64,10 +65,17 @@ pub fn run(listener: TcpListener, config: Config) -> std::io::Result<Server> {
]); ]);
App::new() App::new()
.wrap(Logger::default()) // added logging middleware for logging.
.app_data(handlebars_ref.clone()) .app_data(handlebars_ref.clone())
.app_data(web::Data::new(config.clone())) .app_data(web::Data::new(config.clone()))
.wrap(cors) .wrap(cors)
.wrap(Logger::default()) // added logging middleware for logging. .wrap(Governor::new(
&GovernorConfigBuilder::default()
.per_second(config.rate_limter.time_limit as u64)
.burst_size(config.rate_limter.number_of_requests as u32)
.finish()
.unwrap(),
))
// Serve images and static files (css and js files). // Serve images and static files (css and js files).
.service( .service(
fs::Files::new("/static", format!("{}/static", public_folder_path)) fs::Files::new("/static", format!("{}/static", public_folder_path))