diff options
Diffstat (limited to 'tokio/src/runtime/blocking/mod.rs')
-rw-r--r-- | tokio/src/runtime/blocking/mod.rs | 63 |
1 files changed, 63 insertions, 0 deletions
diff --git a/tokio/src/runtime/blocking/mod.rs b/tokio/src/runtime/blocking/mod.rs new file mode 100644 index 00000000..63c54b74 --- /dev/null +++ b/tokio/src/runtime/blocking/mod.rs @@ -0,0 +1,63 @@ +//! Abstracts out the APIs necessary to `Runtime` for integrating the blocking +//! pool. When the `blocking` feature flag is **not** enabled, these APIs are +//! shells. This isolates the complexity of dealing with conditional +//! compilation. + +cfg_blocking_impl! { + mod pool; + pub(crate) use pool::{spawn_blocking, BlockingPool, Spawner}; + + mod schedule; + mod shutdown; + mod task; + + use crate::runtime::{self, Builder, io, time}; + + pub(crate) fn create_blocking_pool( + builder: &Builder, + spawner: &runtime::Spawner, + io: &io::Handle, + time: &time::Handle, + clock: &time::Clock, + ) -> BlockingPool { + BlockingPool::new( + builder, + spawner, + io, + time, + clock) + + } +} + +cfg_not_blocking_impl! { + use crate::runtime::{self, io, time, Builder}; + + #[derive(Debug, Clone)] + pub(crate) struct BlockingPool {} + + pub(crate) use BlockingPool as Spawner; + + pub(crate) fn create_blocking_pool( + _builder: &Builder, + _spawner: &runtime::Spawner, + _io: &io::Handle, + _time: &time::Handle, + _clock: &time::Clock, + ) -> BlockingPool { + BlockingPool {} + } + + impl BlockingPool { + pub(crate) fn spawner(&self) -> &BlockingPool { + self + } + + pub(crate) fn enter<F, R>(&self, f: F) -> R + where + F: FnOnce() -> R, + { + f() + } + } +} |