use either::Either;
use futures::future::select;
use crate::{
cat::{functor::*, morphism::*},
traits::{
either::IntoEitherExt,
future::{ToFuture, ToFutureExt},
},
FreeExt, Impl, Trait,
};
pub struct Futures;
impl<Tr: ?Sized + Trait> Unwrap<ToFuture<Tr>> for Futures {
type Tr = Tr;
}
impl Wrap for Futures {
type Wrap<Tr: ?Sized + Trait> = ToFuture<Tr>;
}
impl Pure for Futures {
fn pure<Tr: ?Sized + Trait>(x: impl Impl<Tr>) -> impl Impl<Self::Wrap<Tr>> {
async { x }
}
}
impl Map for Futures {
fn map<F: MapFn<In>, In: ?Sized + Trait>(
x: impl Impl<Self::Wrap<In>>,
f: F,
) -> impl Impl<Self::Wrap<F::Out>> {
async { f.run(x.to_future().await) }
}
}
impl Map2 for Futures {
fn map2<F: MapFn2<In0, In1>, In0: ?Sized + Trait, In1: ?Sized + Trait>(
x0: impl Impl<Self::Wrap<In0>>,
x1: impl Impl<Self::Wrap<In1>>,
f: F,
) -> impl Impl<Self::Wrap<F::Out>> {
async {
let (x0, x1) = futures::join!(x0.to_future(), x1.to_future());
f.run(x0, x1)
}
}
}
impl Select for Futures {
fn select<F: SelectFn<In0, In1>, In0: ?Sized + Trait, In1: ?Sized + Trait>(
x0: impl Impl<Self::Wrap<In0>>,
x1: impl Impl<Self::Wrap<In1>>,
f: F,
) -> impl Impl<Self::Wrap<F::Out>> {
async {
Trait::union(
match select(
std::pin::pin!(x0.to_future()),
std::pin::pin!(x1.to_future()),
)
.await
{
futures::future::Either::Left((x, y)) => {
Either::Left(Trait::union(match f.run0(x) {
Either::Left(x) => Either::Left(x),
Either::Right(x) => Either::Right(F::run01(x, y.await)),
}))
}
futures::future::Either::Right((x, y)) => {
Either::Right(Trait::union(match f.run1(x) {
Either::Left(x) => Either::Left(x),
Either::Right(x) => Either::Right(F::run10(x, y.await)),
}))
}
},
)
}
}
}
impl Flatten for Futures {
fn flatten<Tr: ?Sized + Trait>(
x: impl Impl<Self::Wrap<Self::Wrap<Tr>>>,
) -> impl Impl<Self::Wrap<Tr>> {
async { x.to_future().await.to_future().await }
}
}
impl Iterate for Futures {
fn iterate<F: IterateFn<Self>>(mut f: F) -> impl Impl<Self::Wrap<F::Out>> {
async {
loop {
match f.run().to_future().await.into_either() {
Either::Left(next) => f = next,
Either::Right(x) => break x,
}
}
}
}
}
impl Inspect for Futures {
fn inspect<F: InspectFn<In, Self>, In: ?Sized + Trait>(
x: impl Impl<Self::Wrap<In>>,
f: F,
) -> impl Impl<Self::Wrap<F::Out>> {
async { f.run(&mut x.to_future().await).to_future().await.free() }
}
}
#[cfg(test)]
mod test {
use crate::traits::is::IsExt;
use super::*;
#[test]
fn test() {
let x = Futures::pure(0);
let x = Futures::map(x, |x| x + 1);
let x = Futures::map(x, |x| x + 1);
let x = Futures::map(x, |x| x + 1);
let x = Futures::map(x, |x| x + 1);
let x = Futures::map(x, |x| x + 1);
let x = x.to_future();
let x = futures::executor::block_on(x);
let x = x.into_that();
assert_eq!(x, 5);
}
}