refactor: use result instead of waitresult enum

src/limits.rs

@@ -8,11 +8,6 @@ use std::{
 
 use parking_lot::Mutex;
 
-pub enum WaitResult {
-    Ready,
-    Pending(Duration),
-}
-
 #[derive(Debug, Clone)]
 pub struct RateLimiter {
     rate: u64,
@@ -33,11 +28,11 @@ impl RateLimiter {
         }
     }
 
-    pub fn try_wait(&self) -> WaitResult {
+    pub fn try_wait(&self) -> Result<(), Duration> {
         let mut completed = self.completed.lock();
         if let Some(new_completed) = completed.checked_sub(1) {
             *completed = new_completed;
-            return WaitResult::Ready;
+            return Ok(());
         }
 
         *completed = self.rate;
@@ -49,10 +44,10 @@ impl RateLimiter {
 
         if let Some(new_remaining) = remaining.checked_sub(delta) {
             *remaining = new_remaining;
-            WaitResult::Pending(new_remaining)
+            Err(new_remaining)
         } else {
             *remaining = self.reset;
-            WaitResult::Ready
+            Ok(())
         }
     }
 
@@ -97,8 +92,8 @@ impl Future for RateLimitWait {
         *self.waker.lock() = Some(cx.waker().clone());
 
         match self.limiter.try_wait() {
-            WaitResult::Ready => Poll::Ready(()),
-            WaitResult::Pending(_) => Poll::Pending,
+            Ok(()) => Poll::Ready(()),
+            Err(_) => Poll::Pending,
         }
     }
 }