use crate::prelude::*;
use crate::{complete_proxy_impl, error_proxy_impl};
#[derive(Clone)]
pub struct MapOp<S, M> {
pub(crate) source: S,
pub(crate) func: M,
}
#[doc(hidden)]
macro_rules! observable_impl {
($subscription:ty, $($marker:ident +)* $lf: lifetime) => {
fn actual_subscribe<O>(
self,
subscriber: Subscriber<O, $subscription>,
) -> Self::Unsub
where O: Observer<Item=Self::Item, Err=Self::Err> + $($marker +)* $lf {
let map = self.func;
self.source.actual_subscribe(Subscriber {
observer: MapObserver {
observer: subscriber.observer,
map,
_marker: TypeHint::new(),
},
subscription: subscriber.subscription,
})
}
}
}
impl<Item, S, M> Observable for MapOp<S, M>
where
S: Observable,
M: FnMut(S::Item) -> Item,
{
type Item = Item;
type Err = S::Err;
}
impl<'a, Item, S, M> LocalObservable<'a> for MapOp<S, M>
where
S: LocalObservable<'a>,
M: FnMut(S::Item) -> Item + 'a,
S::Item: 'a,
{
type Unsub = S::Unsub;
observable_impl!(LocalSubscription,'a);
}
impl<Item, S, M> SharedObservable for MapOp<S, M>
where
S: SharedObservable,
M: FnMut(S::Item) -> Item + Send + Sync + 'static,
S::Item: 'static,
{
type Unsub = S::Unsub;
observable_impl!(SharedSubscription, Send + Sync + 'static);
}
#[derive(Clone)]
pub struct MapObserver<O, M, Item> {
observer: O,
map: M,
_marker: TypeHint<*const Item>,
}
impl<Item, Err, O, M, B> Observer for MapObserver<O, M, Item>
where
O: Observer<Item = B, Err = Err>,
M: FnMut(Item) -> B,
{
type Item = Item;
type Err = Err;
fn next(&mut self, value: Item) { self.observer.next((self.map)(value)) }
error_proxy_impl!(Err, observer);
complete_proxy_impl!(observer);
#[inline]
fn is_stopped(&self) -> bool { self.observer.is_stopped() }
}
#[cfg(test)]
mod test {
use crate::prelude::*;
#[test]
fn primitive_type() {
let mut i = 0;
observable::from_iter(100..101)
.map(|v| v * 2)
.subscribe(|v| i += v);
assert_eq!(i, 200);
}
#[test]
fn reference_lifetime_should_work() {
let mut i = 0;
observable::of(100).map(|v| v).subscribe(|v| i += v);
assert_eq!(i, 100);
}
#[test]
fn fork_and_shared() {
let m = observable::from_iter(0..100).map(|v| v);
m.map(|v| v).into_shared().subscribe(|_| {});
let m = observable::from_iter(vec!['a', 'b', 'c']).map(|_v| 1);
m.map(|v| v as f32).into_shared().subscribe(|_| {});
let m = observable::of(&1).map(|v| v);
m.map(|v| v).into_shared().subscribe(|_| {});
}
#[test]
fn map_types_mixed() {
let mut i = 0;
observable::from_iter(vec!['a', 'b', 'c'])
.map(|_v| 1)
.subscribe(|v| i += v);
assert_eq!(i, 3);
}
#[test]
fn benchmark() { do_bench(); }
benchmark_group!(do_bench, bench);
fn bench(b: &mut bencher::Bencher) { b.iter(primitive_type); }
}