Fix imports

Author: AlphaKeks

tokio-util/src/task/join_map.rs

@@ -2,10 +2,10 @@ use hashbrown::hash_map::RawEntryMut;
 use hashbrown::HashMap;
 use std::borrow::Borrow;
 use std::collections::hash_map::RandomState;
+use std::fmt;
 use std::future::Future;
 use std::hash::{BuildHasher, Hash, Hasher};
 use std::marker::PhantomData;
-use std::{fmt, io};
 use tokio::runtime::Handle;
 use tokio::task::{AbortHandle, Id, JoinError, JoinSet, LocalSet};
 
@@ -932,7 +932,7 @@ where
     ///
     /// [`join_next`]: JoinMap::join_next
     #[track_caller]
-    pub fn spawn<F>(self, key: K, task: F) -> io::Result<()>
+    pub fn spawn<F>(self, key: K, task: F) -> std::io::Result<()>
     where
         F: Future<Output = V>,
         F: Send + 'static,
@@ -959,7 +959,7 @@ where
     ///
     /// [`join_next`]: JoinMap::join_next
     #[track_caller]
-    pub fn spawn_on<F>(&mut self, key: K, task: F, handle: &Handle) -> io::Result<()>
+    pub fn spawn_on<F>(&mut self, key: K, task: F, handle: &Handle) -> std::io::Result<()>
     where
         F: Future<Output = V>,
         F: Send + 'static,
@@ -995,7 +995,7 @@ where
     ///
     /// [`join_next`]: JoinMap::join_next
     #[track_caller]
-    pub fn spawn_blocking<F>(&mut self, key: K, f: F) -> io::Result<()>
+    pub fn spawn_blocking<F>(&mut self, key: K, f: F) -> std::io::Result<()>
     where
         F: FnOnce() -> V,
         F: Send + 'static,
@@ -1027,7 +1027,7 @@ where
     ///
     /// [`join_next`]: JoinMap::join_next
     #[track_caller]
-    pub fn spawn_blocking_on<F>(&mut self, key: K, f: F, handle: &Handle) -> io::Result<()>
+    pub fn spawn_blocking_on<F>(&mut self, key: K, f: F, handle: &Handle) -> std::io::Result<()>
     where
         F: FnOnce() -> V,
         F: Send + 'static,
@@ -1059,7 +1059,7 @@ where
     /// [`LocalSet`]: tokio::task::LocalSet
     /// [`join_next`]: JoinMap::join_next
     #[track_caller]
-    pub fn spawn_local<F>(&mut self, key: K, task: F) -> io::Result<()>
+    pub fn spawn_local<F>(&mut self, key: K, task: F) -> std::io::Result<()>
     where
         F: Future<Output = V>,
         F: 'static,
@@ -1086,7 +1086,7 @@ where
     /// [`LocalSet`]: tokio::task::LocalSet
     /// [`join_next`]: JoinMap::join_next
     #[track_caller]
-    pub fn spawn_local_on<F>(&mut self, key: K, task: F, local_set: &LocalSet) -> io::Result<()>
+    pub fn spawn_local_on<F>(&mut self, key: K, task: F, local_set: &LocalSet) -> std::io::Result<()>
     where
         F: Future<Output = V>,
         F: 'static,

tokio-util/src/task/mod.rs

@@ -7,7 +7,11 @@ pub use spawn_pinned::LocalPoolHandle;
 
 #[cfg(tokio_unstable)]
 #[cfg_attr(docsrs, doc(cfg(all(tokio_unstable, feature = "rt"))))]
-pub use join_map::{Builder as JoinMapBuilder, JoinMap, JoinMapKeys};
+pub use join_map::{JoinMap, JoinMapKeys};
+
+#[cfg(tokio_unstable)]
+#[cfg_attr(docsrs, doc(cfg(all(tokio_unstable, feature = "rt"))))]
+pub use join_map::Builder as JoinMapBuilder;
 
 pub mod task_tracker;
 pub use task_tracker::TaskTracker;