#[cfg(any( all(any(feature = "client", feature = "server"), feature = "http2"), all(feature = "server", feature = "http1"), ))] use std::time::Duration; use std::{fmt, sync::Arc}; use std::{pin::Pin, time::Instant}; use crate::rt::Sleep; use crate::rt::Timer; /// A user-provided timer to time background tasks. #[derive(Clone)] pub(crate) enum Time { Timer(Arc), Empty, } #[cfg(all(feature = "server", feature = "http1"))] #[derive(Clone, Copy, Debug)] pub(crate) enum Dur { Default(Option), Configured(Option), } impl fmt::Debug for Time { fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result { f.debug_struct("Time").finish() } } impl Time { #[cfg(all(any(feature = "client", feature = "server"), feature = "http2"))] pub(crate) fn sleep(&self, duration: Duration) -> Pin> { match *self { Time::Empty => { panic!("You must supply a timer.") } Time::Timer(ref t) => t.sleep(duration), } } #[cfg(feature = "http1")] pub(crate) fn sleep_until(&self, deadline: Instant) -> Pin> { match *self { Time::Empty => { panic!("You must supply a timer.") } Time::Timer(ref t) => t.sleep_until(deadline), } } pub(crate) fn reset(&self, sleep: &mut Pin>, new_deadline: Instant) { match *self { Time::Empty => { panic!("You must supply a timer.") } Time::Timer(ref t) => t.reset(sleep, new_deadline), } } #[cfg(all(feature = "server", feature = "http1"))] pub(crate) fn check(&self, dur: Dur, name: &'static str) -> Option { match dur { Dur::Default(Some(dur)) => match self { Time::Empty => { warn!("timeout `{}` has default, but no timer set", name,); None } Time::Timer(..) => Some(dur), }, Dur::Configured(Some(dur)) => match self { Time::Empty => panic!("timeout `{}` set, but no timer set", name,), Time::Timer(..) => Some(dur), }, Dur::Default(None) | Dur::Configured(None) => None, } } }