ci: Run loom test in tokio-util

.github/workflows/loom.yml

@@ -95,3 +95,21 @@ jobs:
         working-directory: tokio
         env:
           SCOPE: ${{ matrix.scope }}
+
+  loom-cancellation-token:
+    name: loom-cancellation-token
+    if: github.repository_owner == 'tokio-rs' && (contains(github.event.pull_request.labels.*.name, 'R-loom-sync-util') || (github.base_ref == null))
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v5
+      - name: Install Rust ${{ env.rust_stable }}
+        uses: dtolnay/rust-toolchain@master
+        with:
+          toolchain: ${{ env.rust_stable }}
+      - uses: Swatinem/rust-cache@v2
+      - name: Run CancellationToken Loom tests
+        run: |
+          LOOM_MAX_PREEMPTIONS=1 LOOM_MAX_BRANCHES=10000 \
+          RUSTFLAGS="-C debug_assertions --cfg tokio_unstable --cfg loom" \
+          cargo test --features full --test sync_cancellation_token -- --test-threads=1 --nocapture
+        working-directory: tokio

tokio-stream/src/wrappers.rs

@@ -22,9 +22,9 @@ cfg_sync! {
 }
 
 cfg_signal! {
-    #[cfg(unix)]
+    #[cfg(all(unix, not(loom)))]
     mod signal_unix;
-    #[cfg(unix)]
+    #[cfg(all(unix, not(loom)))]
     pub use signal_unix::SignalStream;
 
     #[cfg(any(windows, docsrs))]
@@ -39,12 +39,14 @@ cfg_time! {
 }
 
 cfg_net! {
+    #[cfg(not(loom))]
     mod tcp_listener;
+    #[cfg(not(loom))]
     pub use tcp_listener::TcpListenerStream;
 
-    #[cfg(unix)]
+    #[cfg(all(unix, not(loom)))]
     mod unix_listener;
-    #[cfg(unix)]
+    #[cfg(all(unix, not(loom)))]
     pub use unix_listener::UnixListenerStream;
 }
 
@@ -57,6 +59,8 @@ cfg_io_util! {
 }
 
 cfg_fs! {
+    #[cfg(not(loom))]
     mod read_dir;
+    #[cfg(not(loom))]
     pub use read_dir::ReadDirStream;
 }

tokio-stream/src/wrappers/read_dir.rs

@@ -1,3 +1,4 @@
+#![cfg(not(loom))]
 use crate::Stream;
 use std::io;
 use std::pin::Pin;

tokio-stream/src/wrappers/signal_unix.rs

@@ -1,3 +1,4 @@
+#![cfg(not(loom))]
 use crate::Stream;
 use std::pin::Pin;
 use std::task::{Context, Poll};

tokio-stream/src/wrappers/tcp_listener.rs

@@ -1,3 +1,4 @@
+#![cfg(not(loom))]
 use crate::Stream;
 use std::io;
 use std::pin::Pin;

tokio-stream/src/wrappers/unix_listener.rs

@@ -1,3 +1,4 @@
+#![cfg(not(loom))]
 use crate::Stream;
 use std::io;
 use std::pin::Pin;

tokio-util/src/net/mod.rs

@@ -1,5 +1,5 @@
 //! TCP/UDP/Unix helpers for tokio.
-
+#![cfg(not(loom))]
 use crate::either::Either;
 use std::future::Future;
 use std::io::Result;

tokio-util/src/net/unix/mod.rs

@@ -1,5 +1,5 @@
 //! Unix domain socket helpers.
-
+#![cfg(not(loom))]
 use super::Listener;
 use std::io::Result;
 use std::task::{Context, Poll};

tokio-util/src/udp/mod.rs

@@ -1,4 +1,4 @@
 //! UDP framing
-
+#![cfg(not(loom))]
 mod frame;
 pub use frame::UdpFramed;