make API more ergonomic

tokio-rs · Mar 21, 2023 · 87c864f · 87c864f
1 parent 8923ef1
commit 87c864f
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/tokio/src/io/async_fd.rs b/tokio/src/io/async_fd.rs
@@ -559,7 +559,7 @@ impl<T: AsRawFd> AsyncFd<T> {
     ///     let async_fd = AsyncFd::new(socket)?;
     ///
     ///     let written = async_fd
-    ///         .async_io(Interest::WRITABLE, || async_fd.get_ref().send(&[1, 2]))
+    ///         .async_io(Interest::WRITABLE, |inner| inner.send(&[1, 2]))
     ///         .await?;
     ///
     ///     println!("wrote {written} bytes");
@@ -596,9 +596,11 @@ impl<T: AsRawFd> AsyncFd<T> {
     pub async fn async_io<R>(
         &self,
         interest: Interest,
-        f: impl FnMut() -> io::Result<R>,
+        mut f: impl FnMut(&T) -> io::Result<R>,
     ) -> io::Result<R> {
-        self.registration.async_io(interest, f).await
+        self.registration
+            .async_io(interest, || f(self.get_ref()))
+            .await
     }
 }