use crate::disposable::subscription::Subscription;
use crate::safe_lock_option;
use crate::utils::types::{Mutable, NecessarySendSync, Shared};
use crate::{
observable::Observable,
observer::{Observer, Termination},
utils::types::MarkerType,
};
use educe::Educe;
use std::marker::PhantomData;
#[derive(Educe)]
#[educe(Debug, Clone)]
pub struct Catch<E0, OE, F> {
source: OE,
callback: F,
_marker: MarkerType<E0>,
}
impl<E0, OE, F> Catch<E0, OE, F> {
pub fn new<'or, 'sub, T, E, OE1>(source: OE, callback: F) -> Self
where
OE: Observable<'or, 'sub, T, E0>,
OE1: Observable<'or, 'sub, T, E>,
F: FnOnce(E0) -> OE1,
{
Self {
source,
callback,
_marker: PhantomData,
}
}
}
impl<'or, 'sub, T, E0, E, OE, OE1, F> Observable<'or, 'sub, T, E> for Catch<E0, OE, F>
where
E: 'or,
OE: Observable<'or, 'sub, T, E0>,
OE1: Observable<'or, 'sub, T, E>,
F: FnOnce(E0) -> OE1 + NecessarySendSync + 'or,
'sub: 'or,
{
fn subscribe(
self,
observer: impl Observer<T, E> + NecessarySendSync + 'or,
) -> Subscription<'sub> {
let sub = Shared::new(Mutable::new(None));
let onserver = CatchObserver {
observer,
callback: self.callback,
sub: sub.clone(),
_marker: PhantomData,
};
self.source.subscribe(onserver) + sub
}
}
struct CatchObserver<'sub, E, OR, F> {
observer: OR,
callback: F,
sub: Shared<Mutable<Option<Subscription<'sub>>>>,
_marker: MarkerType<E>,
}
impl<'or, 'sub, T, E0, E, OR, OE1, F> Observer<T, E0> for CatchObserver<'sub, E, OR, F>
where
OR: Observer<T, E> + NecessarySendSync + 'or,
OE1: Observable<'or, 'sub, T, E>,
F: FnOnce(E0) -> OE1,
{
fn on_next(&mut self, value: T) {
self.observer.on_next(value);
}
fn on_termination(self, termination: Termination<E0>) {
match termination {
Termination::Completed => self.observer.on_termination(Termination::Completed),
Termination::Error(error) => {
let observable = (self.callback)(error);
let sub = observable.subscribe(self.observer);
safe_lock_option!(replace: self.sub, sub);
}
}
}
}