use Waker::clone_from internally

yoshuawuyts · Nov 6, 2023 · da0c28c · da0c28c
1 parent 112b40e
commit da0c28c
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/src/utils/wakers/array/readiness_array.rs b/src/utils/wakers/array/readiness_array.rs
@@ -62,6 +62,9 @@ impl<const N: usize> ReadinessArray<N> {
     /// Set the parent `Waker`. This needs to be called at the start of every
     /// `poll` function.
     pub(crate) fn set_waker(&mut self, parent_waker: &Waker) {
-        self.parent_waker = Some(parent_waker.clone());
+        match &mut self.parent_waker {
+            Some(prev) => prev.clone_from(parent_waker),
+            None => self.parent_waker = Some(parent_waker.clone()),
+        }
     }
 }
diff --git a/src/utils/wakers/vec/readiness_vec.rs b/src/utils/wakers/vec/readiness_vec.rs
@@ -74,7 +74,10 @@ impl ReadinessVec {
     /// Set the parent `Waker`. This needs to be called at the start of every
     /// `poll` function.
     pub(crate) fn set_waker(&mut self, parent_waker: &Waker) {
-        self.parent_waker = Some(parent_waker.clone());
+        match &mut self.parent_waker {
+            Some(prev) => prev.clone_from(parent_waker),
+            None => self.parent_waker = Some(parent_waker.clone()),
+        }
     }
 
     /// Resize `readiness` to the new length.