use crate::effect::trait_def::Effect;
pub struct Map<Inner, F> {
pub(crate) inner: Inner,
pub(crate) f: F,
}
impl<Inner, F> std::fmt::Debug for Map<Inner, F> {
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
f.debug_struct("Map")
.field("inner", &"<effect>")
.field("f", &"<function>")
.finish()
}
}
impl<Inner, F, U> Effect for Map<Inner, F>
where
Inner: Effect,
F: FnOnce(Inner::Output) -> U + Send,
U: Send,
{
type Output = U;
type Error = Inner::Error;
type Env = Inner::Env;
async fn run(self, env: &Self::Env) -> Result<U, Self::Error> {
let value = self.inner.run(env).await?;
Ok((self.f)(value))
}
}
impl<Inner, F, U> crate::effect::writer::WriterEffect for Map<Inner, F>
where
Inner: crate::effect::writer::WriterEffect,
F: FnOnce(Inner::Output) -> U + Send,
U: Send,
{
type Writes = Inner::Writes;
async fn run_writer(
self,
env: &Self::Env,
) -> (Result<Self::Output, Self::Error>, Self::Writes) {
let (result, writes) = self.inner.run_writer(env).await;
(result.map(self.f), writes)
}
}