Utilize BoxFuture & remove unnecessary unsafe impl Send

mindeng · Aug 22, 2024 · 511b25b · 511b25b
1 parent 7ee2955
commit 511b25b
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/src/token_bucket.rs b/src/token_bucket.rs
@@ -1,5 +1,4 @@
 use std::{
-    pin::Pin,
     sync::{
         atomic::{
             AtomicUsize,
@@ -11,7 +10,7 @@ use std::{
     time::{Duration, Instant},
 };
 
-use futures::Future;
+use futures::{future::BoxFuture, Future};
 
 use crate::rt::delay;
 
@@ -219,13 +218,11 @@ impl TokenBucketRateLimiter {
 }
 
 struct Token<'a> {
-    fut: Pin<Box<dyn Future<Output = ()>>>,
+    fut: BoxFuture<'a, ()>,
     token_bucket: &'a TokenBucketRateLimiter,
     consumed: bool,
 }
 
-unsafe impl Send for Token<'_> {}
-
 impl<'a> Token<'a> {
     fn new(duration: Duration, token_bucket: &'a TokenBucketRateLimiter) -> Token<'a> {
         let fut = delay(duration);