rob
/
torrents.csv
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Using mutex.

This commit is contained in:
Dessalines 2019-11-24 12:29:49 -08:00
parent 3df6c9d091
commit 8499b3f9d1
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
server/service/src/main.rs
View File

@ -28,7 +28,7 @@ const RATE_LIMIT_PER_SECOND: i32 = 60;
pub struct State { pub struct State {
rate_limits: Mutex<HashMap<String, RateLimitBucket>>, rate_limits: Mutex<HashMap<String, RateLimitBucket>>,
pool: r2d2::Pool<SqliteConnectionManager>, pool: Mutex<r2d2::Pool<SqliteConnectionManager>>,
} }
#[derive(Debug)] #[derive(Debug)]
@ -45,7 +45,7 @@ fn main() {
let shared_data = web::Data::new(State { let shared_data = web::Data::new(State {
rate_limits: Mutex::new(HashMap::new()), rate_limits: Mutex::new(HashMap::new()),
pool: pool, pool: Mutex::new(pool),
}); });
HttpServer::new(move || { HttpServer::new(move || {
@ -106,7 +106,7 @@ fn search(
.body(format!("{{\"error\": \"{}\"}}", "Empty query".to_string())); .body(format!("{{\"error\": \"{}\"}}", "Empty query".to_string()));
} }
let conn = data.pool.get().unwrap(); let conn = data.pool.lock().unwrap().get().unwrap();
match check_rate_limit_full(data, &ip, RATE_LIMIT, RATE_LIMIT_PER_SECOND) { match check_rate_limit_full(data, &ip, RATE_LIMIT, RATE_LIMIT_PER_SECOND) {
Ok(_) => HttpResponse::Ok() Ok(_) => HttpResponse::Ok()