Distributed storage to use the new sigs

Kuadrant · May 21, 2024 · dbd7f43 · dbd7f43
1 parent e83fbad
commit dbd7f43
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 16 deletions.
diff --git a/limitador/src/storage/distributed/cr_counter_value.rs b/limitador/src/storage/distributed/cr_counter_value.rs
@@ -20,7 +20,7 @@ impl<A: Ord> CrCounterValue<A> {
             ourselves: actor,
             value: Default::default(),
             others: RwLock::default(),
-            expiry: AtomicExpiryTime::from_now(time_window),
+            expiry: AtomicExpiryTime::new(SystemTime::now() + time_window),
         }
     }
 
@@ -43,7 +43,7 @@ impl<A: Ord> CrCounterValue<A> {
     }
 
     pub fn inc_at(&self, increment: u64, time_window: Duration, when: SystemTime) {
-        if self.expiry.update_if_expired(time_window.as_secs(), when) {
+        if self.expiry.update_if_expired(time_window, when) {
             self.value.store(increment, Ordering::SeqCst);
         } else {
             self.value.fetch_add(increment, Ordering::SeqCst);
@@ -59,10 +59,7 @@ impl<A: Ord> CrCounterValue<A> {
             self.inc_at(increment, time_window, when);
         } else {
             let mut guard = self.others.write().unwrap();
-            if self
-                .expiry
-                .update_if_expired(time_window.as_micros() as u64, when)
-            {
+            if self.expiry.update_if_expired(time_window, when) {
                 guard.insert(actor, increment);
             } else {
                 *guard.entry(actor).or_insert(0) += increment;
@@ -109,7 +106,7 @@ impl<A: Ord> CrCounterValue<A> {
     }
 
     pub fn ttl(&self) -> Duration {
-        self.expiry.duration()
+        self.expiry.ttl()
     }
 
     pub fn expiry(&self) -> SystemTime {
@@ -282,6 +279,6 @@ mod tests {
         a.inc(3, later);
         b.inc(2, later);
         a.merge(b);
-        assert!(a.expiry.duration() < sooner);
+        assert!(a.expiry.ttl() < sooner);
     }
 }
diff --git a/limitador/src/storage/distributed/mod.rs b/limitador/src/storage/distributed/mod.rs
@@ -72,7 +72,7 @@ impl CounterStorage for CrInMemoryStorage {
                 None => self.qualified_counters.get_with(counter.clone(), || {
                     Arc::new(CrCounterValue::new(
                         self.identifier.clone(),
-                        Duration::from_secs(counter.seconds()),
+                        counter.window(),
                     ))
                 }),
                 Some(counter) => counter,
@@ -82,16 +82,16 @@ impl CounterStorage for CrInMemoryStorage {
             match limits_by_namespace.entry(counter.limit().namespace().clone()) {
                 Entry::Vacant(v) => {
                     let mut limits = HashMap::new();
-                    let duration = Duration::from_secs(counter.seconds());
-                    let counter_val = CrCounterValue::new(self.identifier.clone(), duration);
+                    let counter_val =
+                        CrCounterValue::new(self.identifier.clone(), counter.window());
                     self.increment_counter(counter.clone(), &counter_val, delta, now);
                     limits.insert(counter.limit().clone(), counter_val);
                     v.insert(limits);
                 }
                 Entry::Occupied(mut o) => match o.get_mut().entry(counter.limit().clone()) {
                     Entry::Vacant(v) => {
-                        let duration = Duration::from_secs(counter.seconds());
-                        let counter_value = CrCounterValue::new(self.identifier.clone(), duration);
+                        let counter_value =
+                            CrCounterValue::new(self.identifier.clone(), counter.window());
                         self.increment_counter(counter.clone(), &counter_value, delta, now);
                         v.insert(counter_value);
                     }
@@ -158,7 +158,7 @@ impl CounterStorage for CrInMemoryStorage {
                 None => self.qualified_counters.get_with(counter.clone(), || {
                     Arc::new(CrCounterValue::new(
                         self.identifier.clone(),
-                        Duration::from_secs(counter.seconds()),
+                        counter.window(),
                     ))
                 }),
                 Some(counter) => counter,
@@ -394,7 +394,7 @@ impl CrInMemoryStorage {
         delta: u64,
         when: SystemTime,
     ) {
-        counter.inc_at(delta, Duration::from_secs(key.seconds()), when);
+        counter.inc_at(delta, key.window(), when);
         let sender = self.sender.clone();
         let counter = counter.clone();
         tokio::spawn(async move {
@@ -429,7 +429,7 @@ impl From<Counter> for CounterKey {
     fn from(value: Counter) -> Self {
         Self {
             namespace: value.namespace().clone(),
-            seconds: value.seconds(),
+            seconds: value.window().as_secs(),
             variables: value.limit().variables(),
             conditions: value.limit().conditions(),
             vars: value.set_variables().clone(),