Add task::yield_to() helper (#375)

ntex-util/CHANGES.md

@@ -1,5 +1,9 @@
 # Changes
 
+## [2.1.0] - 2024-06-26
+
+* Add task::yield_to() helper
+
 ## [2.0.1] - 2024-05-28
 
 * Re-enable BufferService

ntex-util/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "ntex-util"
-version = "2.0.1"
+version = "2.1.0"
 authors = ["ntex contributors <team@ntex.rs>"]
 description = "Utilities for ntex framework"
 keywords = ["network", "framework", "async", "futures"]
@@ -16,7 +16,7 @@ name = "ntex_util"
 path = "src/lib.rs"
 
 [dependencies]
-ntex-service = "3.0"
+ntex-service = "3"
 ntex-rt = "0.4"
 bitflags = "2"
 fxhash = "0.2"

ntex-util/src/task.rs

@@ -69,3 +69,40 @@ impl fmt::Debug for LocalWaker {
         write!(f, "LocalWaker")
     }
 }
+
+#[doc(hidden)]
+/// Yields execution back to the current runtime.
+pub async fn yield_to() {
+    use std::{future::Future, pin::Pin, task::Context, task::Poll};
+
+    struct Yield {
+        completed: bool,
+    }
+
+    impl Future for Yield {
+        type Output = ();
+
+        fn poll(mut self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<()> {
+            if self.completed {
+                return Poll::Ready(());
+            }
+
+            self.completed = true;
+            cx.waker().wake_by_ref();
+
+            Poll::Pending
+        }
+    }
+
+    Yield { completed: false }.await;
+}
+
+#[cfg(test)]
+mod test {
+    use super::*;
+
+    #[ntex_macros::rt_test2]
+    async fn yield_test() {
+        yield_to().await;
+    }
+}

ntex/Cargo.toml

@@ -62,7 +62,7 @@ ntex-http = "0.1.12"
 ntex-router = "0.5.3"
 ntex-service = "3.0"
 ntex-macros = "0.1.3"
-ntex-util = "2.0"
+ntex-util = "2"
 ntex-bytes = "0.1.27"
 ntex-server = "2.0"
 ntex-h2 = "1.0"