use crate::de::{DeserializeAs, DeserializeAsWrap};
use serde::de::{MapAccess, SeqAccess};
use std::marker::PhantomData;
#[inline]
pub(crate) fn size_hint_cautious(hint: Option<usize>) -> usize {
std::cmp::min(hint.unwrap_or(0), 4096)
}
pub(crate) const NANOS_PER_SEC: u32 = 1_000_000_000;
pub(crate) struct MapIter<'de, A, K, KAs, V, VAs> {
pub(crate) access: A,
marker: PhantomData<(&'de (), K, KAs, V, VAs)>,
}
impl<'de, A, K, KAs, V, VAs> MapIter<'de, A, K, KAs, V, VAs> {
pub(crate) fn new(access: A) -> Self
where
A: MapAccess<'de>,
{
Self {
access,
marker: PhantomData,
}
}
}
impl<'de, A, K, KAs, V, VAs> Iterator for MapIter<'de, A, K, KAs, V, VAs>
where
A: MapAccess<'de>,
KAs: DeserializeAs<'de, K>,
VAs: DeserializeAs<'de, V>,
{
#[allow(clippy::type_complexity)]
type Item = Result<(DeserializeAsWrap<K, KAs>, DeserializeAsWrap<V, VAs>), A::Error>;
fn next(&mut self) -> Option<Self::Item> {
self.access.next_entry().transpose()
}
fn size_hint(&self) -> (usize, Option<usize>) {
match self.access.size_hint() {
Some(size) => (size, Some(size)),
None => (0, None),
}
}
}
pub(crate) struct SeqIter<'de, A, K, KAs, V, VAs> {
access: A,
marker: PhantomData<(&'de (), K, KAs, V, VAs)>,
}
impl<'de, A, K, KAs, V, VAs> SeqIter<'de, A, K, KAs, V, VAs> {
pub(crate) fn new(access: A) -> Self
where
A: SeqAccess<'de>,
{
Self {
access,
marker: PhantomData,
}
}
}
impl<'de, A, K, KAs, V, VAs> Iterator for SeqIter<'de, A, K, KAs, V, VAs>
where
A: SeqAccess<'de>,
KAs: DeserializeAs<'de, K>,
VAs: DeserializeAs<'de, V>,
{
#[allow(clippy::type_complexity)]
type Item = Result<(DeserializeAsWrap<K, KAs>, DeserializeAsWrap<V, VAs>), A::Error>;
fn next(&mut self) -> Option<Self::Item> {
self.access.next_element().transpose()
}
fn size_hint(&self) -> (usize, Option<usize>) {
match self.access.size_hint() {
Some(size) => (size, Some(size)),
None => (0, None),
}
}
}
pub(crate) fn duration_as_secs_f64(dur: &std::time::Duration) -> f64 {
(dur.as_secs() as f64) + (dur.subsec_nanos() as f64) / (NANOS_PER_SEC as f64)
}
pub(crate) fn duration_from_secs_f64(secs: f64) -> Result<std::time::Duration, String> {
const MAX_NANOS_F64: f64 = ((u64::max_value() as u128 + 1) * (NANOS_PER_SEC as u128)) as f64;
let nanos = secs * (NANOS_PER_SEC as f64);
if !nanos.is_finite() {
return Err("got non-finite value when converting float to duration".into());
}
if nanos >= MAX_NANOS_F64 {
return Err("overflow when converting float to duration".into());
}
if nanos < 0.0 {
return Err("underflow when converting float to duration".into());
}
let nanos = nanos as u128;
Ok(std::time::Duration::new(
(nanos / (NANOS_PER_SEC as u128)) as u64,
(nanos % (NANOS_PER_SEC as u128)) as u32,
))
}