use crate::utils::types::NecessarySendSync;
use crate::{
disposable::subscription::Subscription,
observable::{Observable, observable_ext::ObservableExt},
observer::{Observer, Termination},
};
use educe::Educe;
#[derive(Educe)]
#[educe(Debug, Clone)]
pub struct DoBeforeTermination<OE, F> {
source: OE,
callback: F,
}
impl<OE, F> DoBeforeTermination<OE, F> {
pub fn new<'or, 'sub, T, E>(source: OE, callback: F) -> Self
where
OE: Observable<'or, 'sub, T, E>,
F: FnOnce(&Termination<E>),
{
Self { source, callback }
}
}
impl<'or, 'sub, T, E, OE, F> Observable<'or, 'sub, T, E> for DoBeforeTermination<OE, F>
where
T: 'or,
E: 'or,
OE: Observable<'or, 'sub, T, E>,
F: FnOnce(&Termination<E>) + NecessarySendSync + 'or,
{
fn subscribe(
self,
observer: impl Observer<T, E> + NecessarySendSync + 'or,
) -> Subscription<'sub> {
self.source
.hook_on_termination(move |observer, termination| {
(self.callback)(&termination);
observer.on_termination(termination)
})
.subscribe(observer)
}
}