then_each.h

Go to the documentation of this file.

#pragma once
 
#include "di/execution/algorithm/then.h"
#include "di/execution/sequence/transform_each.h"
 
namespace di::execution {
namespace then_each_ns {
    struct Function {
        template<concepts::Sender Seq, concepts::MovableValue Fun>
        auto operator()(Seq&& sequence, Fun&& transformer) const -> concepts::SequenceSender auto {
            if constexpr (concepts::TagInvocable<Function, Seq, Fun>) {
                return function::tag_invoke(*this, util::forward<Seq>(sequence), util::forward<Fun>(transformer));
            } else {
                return transform_each(util::forward<Seq>(sequence), then(util::forward<Fun>(transformer)));
            }
        }
    };
 
    struct ErrorFunction {
        template<concepts::Sender Seq, concepts::MovableValue Fun>
        auto operator()(Seq&& sequence, Fun&& transformer) const -> concepts::SequenceSender auto {
            if constexpr (concepts::TagInvocable<Function, Seq, Fun>) {
                return function::tag_invoke(*this, util::forward<Seq>(sequence), util::forward<Fun>(transformer));
            } else {
                return transform_each(util::forward<Seq>(sequence), upon_error(util::forward<Fun>(transformer)));
            }
        }
    };
 
    struct StoppedFunction {
        template<concepts::Sender Seq, concepts::MovableValue Fun>
        auto operator()(Seq&& sequence, Fun&& transformer) const -> concepts::SequenceSender auto {
            if constexpr (concepts::TagInvocable<Function, Seq, Fun>) {
                return function::tag_invoke(*this, util::forward<Seq>(sequence), util::forward<Fun>(transformer));
            } else {
                return transform_each(util::forward<Seq>(sequence), upon_stopped(util::forward<Fun>(transformer)));
            }
        }
    };
}

constexpr inline auto then_each = function::curry_back(then_each_ns::Function {}, meta::c_<2ZU>);

constexpr inline auto upon_error_each = function::curry_back(then_each_ns::ErrorFunction {}, meta::c_<2ZU>);

constexpr inline auto upon_stopped_each = function::curry_back(then_each_ns::StoppedFunction {}, meta::c_<2ZU>);
}