Bommels05/Modrinth
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

See if RateLimit is cors issue (#303)

Browse Source
This commit is contained in:
Geometrically 2022-02-19 17:51:24 -07:00 committed by GitHub
parent 7c80b61666
commit c0c80c0fdf
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 27 additions and 27 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

54
src/main.rs
View File

@ -246,33 +246,33 @@ async fn main() -> std::io::Result<()> {
.max_age(3600)
.send_wildcard(),
)
.wrap(
RateLimiter::new(MemoryStoreActor::from(store.clone()).start())
.with_identifier(|req| {
let connection_info = req.connection_info();
let ip =
String::from(if parse_var("CLOUDFLARE_INTEGRATION").unwrap_or(false) {
if let Some(header) = req.headers().get("CF-Connecting-IP") {
header.to_str().map_err(|_| ARError::IdentificationError)?
} else {
connection_info
.peer_addr()
.ok_or(ARError::IdentificationError)?
}
} else {
connection_info
.peer_addr()
.ok_or(ARError::IdentificationError)?
});
Ok(ip)
})
.with_interval(std::time::Duration::from_secs(60))
.with_max_requests(300)
.with_ignore_ips(
parse_strings_from_var("RATE_LIMIT_IGNORE_IPS").unwrap_or_default(),
),
)
// .wrap(
// RateLimiter::new(MemoryStoreActor::from(store.clone()).start())
// .with_identifier(|req| {
// let connection_info = req.connection_info();
// let ip =
// String::from(if parse_var("CLOUDFLARE_INTEGRATION").unwrap_or(false) {
// if let Some(header) = req.headers().get("CF-Connecting-IP") {
// header.to_str().map_err(|_| ARError::IdentificationError)?
// } else {
// connection_info
// .peer_addr()
// .ok_or(ARError::IdentificationError)?
// }
// } else {
// connection_info
// .peer_addr()
// .ok_or(ARError::IdentificationError)?
// });
//
// Ok(ip)
// })
// .with_interval(std::time::Duration::from_secs(60))
// .with_max_requests(300)
// .with_ignore_ips(
// parse_strings_from_var("RATE_LIMIT_IGNORE_IPS").unwrap_or_default(),
// ),
// )
.app_data(web::Data::new(pool.clone()))
.app_data(web::Data::new(file_host.clone()))
.app_data(web::Data::new(indexing_queue.clone()))