use crate::internals::stream_controller::*;
use crate::prelude::*;
use std::marker::PhantomData;
#[derive(Clone)]
pub struct TakeUntil<'a, Item, TriggerValue>
where
Item: Clone + Send + Sync,
TriggerValue: Clone + Send + Sync,
{
trigger: Observable<'a, TriggerValue>,
_item: PhantomData<Item>,
}
impl<'a, Item, TriggerValue> TakeUntil<'a, Item, TriggerValue>
where
Item: Clone + Send + Sync,
TriggerValue: Clone + Send + Sync,
{
pub fn new(
trigger: Observable<'a, TriggerValue>,
) -> TakeUntil<'a, Item, TriggerValue>
where
TriggerValue: Clone + Send + Sync,
{
TakeUntil { trigger, _item: PhantomData }
}
pub fn execute(&self, source: Observable<'a, Item>) -> Observable<'a, Item> {
let trigger = self.trigger.clone();
Observable::<Item>::create(move |s| {
let sctl = StreamController::new(s);
let obs_trigger = {
let sctl_trigger_next = sctl.clone();
sctl.new_observer(
move |_, _| {
sctl_trigger_next.sink_complete_force();
},
|_, _| {},
|_| {},
)
};
let obs_source = {
let sctl_next = sctl.clone();
let sctl_error = sctl.clone();
let sctl_complete = sctl.clone();
sctl.new_observer(
move |_, x| {
sctl_next.sink_next(x);
},
move |_, e| {
sctl_error.sink_error(e);
},
move |_| sctl_complete.sink_complete_force(), )
};
trigger.inner_subscribe(obs_trigger);
source.inner_subscribe(obs_source);
})
}
}
impl<'a, Item> Observable<'a, Item>
where
Item: Clone + Send + Sync,
{
pub fn take_until<TriggerValue>(
&self,
trigger: Observable<'a, TriggerValue>,
) -> Observable<'a, Item>
where
TriggerValue: Clone + Send + Sync,
{
TakeUntil::new(trigger).execute(self.clone())
}
}
#[cfg(all(test, not(feature = "web")))]
mod test {
use crate::prelude::*;
use std::{thread, time};
#[test]
fn basic() {
let o = Observable::create(|s| {
for n in 0..10 {
thread::sleep(time::Duration::from_millis(100));
s.next(n);
}
s.complete();
});
o.take_until(observables::timer(
time::Duration::from_millis(500),
schedulers::new_thread_scheduler(),
))
.subscribe(
print_next_fmt!("{}"),
print_error!(),
print_complete!(),
);
}
#[test]
fn with_subject() {
let sbj = subjects::Subject::new();
let sbj_ = sbj.clone();
let sbsc = observables::interval(
time::Duration::from_millis(100),
schedulers::new_thread_scheduler(),
)
.take_until(sbj.observable())
.flat_map(move |x| {
if x == 10 {
sbj_.next(());
}
return observables::empty::<u64>();
})
.subscribe(
junk_next!(),
junk_error!(),
junk_complete!(),
);
while sbsc.is_subscribed() {}
}
}