use crate::loom::sync::{Arc, Mutex};
use crate::time::driver::ClockTime;
use std::fmt;
#[derive(Clone)]
pub(crate) struct Handle {
time_source: ClockTime,
inner: Arc<Mutex<super::Inner>>,
}
impl Handle {
pub(super) fn new(inner: Arc<Mutex<super::Inner>>) -> Self {
let time_source = inner.lock().time_source.clone();
Handle { time_source, inner }
}
pub(super) fn time_source(&self) -> &ClockTime {
&self.time_source
}
pub(super) fn lock(&self) -> crate::loom::sync::MutexGuard<'_, super::Inner> {
self.inner.lock()
}
}
cfg_rt! {
impl Handle {
pub(crate) fn current() -> Self {
crate::runtime::context::time_handle()
.expect("A Tokio 1.x context was found, but timers are disabled. Call `enable_time` on the runtime builder to enable timers.")
}
}
}
cfg_not_rt! {
impl Handle {
pub(crate) fn current() -> Self {
panic!(crate::util::error::CONTEXT_MISSING_ERROR)
}
}
}
impl fmt::Debug for Handle {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
write!(f, "Handle")
}
}