Move reap() logic into SharedPool

djc · Jan 29, 2024 · d86015e · d86015e
1 parent 4c2ea49
commit d86015e
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/bb8/src/inner.rs b/bb8/src/inner.rs
@@ -237,10 +237,7 @@ impl<M: ManageConnection> Reaper<M> {
                 None => break,
             };
 
-            let mut locked = pool.inner.internals.lock();
-            let approvals = locked.reap(&pool.inner.statics);
-            drop(locked);
-
+            let approvals = pool.inner.reap();
             pool.spawn_replenishing_approvals(approvals);
         }
     }

diff --git a/bb8/src/internals.rs b/bb8/src/internals.rs
@@ -44,6 +44,11 @@ where
         (conn, approvals)
     }
 
+    pub(crate) fn reap(&self) -> ApprovalIter {
+        let mut locked = self.internals.lock();
+        locked.reap(&self.statics)
+    }
+
     pub(crate) fn forward_error(&self, err: M::Error) {
         self.statics.error_sink.sink(err);
     }