use crate::utils::types::NecessarySendSync;
use crate::{
disposable::subscription::Subscription,
observable::Observable,
observer::{Observer, Termination},
utils::types::MarkerType,
};
use educe::Educe;
use std::marker::PhantomData;
#[derive(Educe)]
#[educe(Debug, Clone)]
pub struct Reduce<T, T1, OE, F> {
source: OE,
initial_value: T,
callback: F,
_marker: MarkerType<T1>,
}
impl<T, T1, OE, F> Reduce<T, T1, OE, F> {
pub fn new<'or, 'sub, E>(source: OE, initial_value: T, callback: F) -> Self
where
OE: Observable<'or, 'sub, T1, E>,
F: FnMut(T, T1) -> T,
{
Self {
source,
initial_value,
callback,
_marker: PhantomData,
}
}
}
impl<'or, 'sub, T, T1, E, OE, F> Observable<'or, 'sub, T, E> for Reduce<T, T1, OE, F>
where
T: NecessarySendSync + 'or,
OE: Observable<'or, 'sub, T1, E>,
F: FnMut(T, T1) -> T + NecessarySendSync + 'or,
{
fn subscribe(
self,
observer: impl Observer<T, E> + NecessarySendSync + 'or,
) -> Subscription<'sub> {
let observer = ReduceObserver {
observer,
value: Some(self.initial_value),
callback: self.callback,
};
self.source.subscribe(observer)
}
}
struct ReduceObserver<T, OR, F> {
observer: OR,
value: Option<T>,
callback: F,
}
impl<T, T1, E, OR, F> Observer<T1, E> for ReduceObserver<T, OR, F>
where
OR: Observer<T, E>,
F: FnMut(T, T1) -> T,
{
fn on_next(&mut self, value: T1) {
self.value = Some((self.callback)(self.value.take().unwrap(), value));
}
fn on_termination(mut self, termination: Termination<E>) {
match termination {
Termination::Completed => {
self.observer.on_next(self.value.unwrap());
}
Termination::Error(_) => {}
}
self.observer.on_termination(termination)
}
}