use crate::collector::{Collector, CollectorBase};
use std::ops::ControlFlow;
#[derive(Debug, Clone)]
pub struct Copying<C>(C);
impl<C> Copying<C> {
pub(in crate::collector) fn new(collector: C) -> Self {
Self(collector)
}
}
impl<C> CollectorBase for Copying<C>
where
C: CollectorBase,
{
type Output = C::Output;
#[inline]
fn finish(self) -> Self::Output {
self.0.finish()
}
#[inline]
fn break_hint(&self) -> ControlFlow<()> {
self.0.break_hint()
}
}
impl<'a, C, T> Collector<&'a T> for Copying<C>
where
C: Collector<T>,
T: Copy,
{
#[inline]
fn collect(&mut self, &item: &'a T) -> ControlFlow<()> {
self.0.collect(item)
}
#[inline]
fn collect_many(&mut self, items: impl IntoIterator<Item = &'a T>) -> ControlFlow<()> {
self.0.collect_many(items.into_iter().cloned())
}
fn collect_then_finish(self, items: impl IntoIterator<Item = &'a T>) -> Self::Output {
self.0.collect_then_finish(items.into_iter().cloned())
}
}
impl<'a, C, T> Collector<&'a mut T> for Copying<C>
where
C: Collector<T>,
T: Copy,
{
#[inline]
fn collect(&mut self, &mut item: &'a mut T) -> ControlFlow<()> {
self.0.collect(item)
}
#[inline]
fn collect_many(&mut self, items: impl IntoIterator<Item = &'a mut T>) -> ControlFlow<()> {
self.0.collect_many(items.into_iter().map(|&mut item| item))
}
fn collect_then_finish(self, items: impl IntoIterator<Item = &'a mut T>) -> Self::Output {
self.0
.collect_then_finish(items.into_iter().map(|&mut item| item))
}
}