HMC/handmade-revolt-backend
6
0
Fork 0
forked from jmug/stoatchat
Code Pull Requests Actions Packages Releases 2 Activity

fix(ratelimiter): mutate entry instead of copying

Browse Source
This commit is contained in:
Paul Makles
2023-03-15 21:54:58 +00:00
parent f309218573
commit 59832e6ba2
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
crates/quark/src/web/ratelimiter.rs
View File

@@ -25,7 +25,7 @@ use dashmap::DashMap;
use once_cell::sync::Lazy; use once_cell::sync::Lazy;
/// Ratelimit Bucket /// Ratelimit Bucket
#[derive(Clone, Copy)] #[derive(Clone, Copy, Debug)]
struct Entry { struct Entry {
used: u8, used: u8,
reset: u128, reset: u128,
@@ -50,7 +50,7 @@ impl Entry {
} }
/// Deduct one unit from the bucket and save /// Deduct one unit from the bucket and save
pub fn deduct(mut self) { pub fn deduct(&mut self) {
let current_time = now().as_millis(); let current_time = now().as_millis();
if current_time > self.reset { if current_time > self.reset {
self.used = 1; self.used = 1;
@@ -192,7 +192,7 @@ impl Ratelimiter {
let key = key.finish(); let key = key.finish();
let limit = resolve_bucket_limit(bucket); let limit = resolve_bucket_limit(bucket);
let entry = Entry::from(key); let mut entry = Entry::from(key);
let remaining = entry.get_remaining(limit); let remaining = entry.get_remaining(limit);
if remaining > 0 { if remaining > 0 {