123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127 |
- #pragma once
- /*
- * Copyright 2018-present Facebook, Inc.
- *
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
- #include <folly/experimental/pushmi/o/extension_operators.h>
- #include <cassert>
- namespace pushmi {
- namespace detail {
- PUSHMI_TEMPLATE(class SideEffects, class Out)
- (requires Receiver<SideEffects>&& Receiver<Out>)
- struct tap_ {
- SideEffects sideEffects;
- Out out;
- // side effect has no effect on the properties.
- using properties = properties_t<Out>;
- PUSHMI_TEMPLATE(class... VN)
- (requires ReceiveValue<SideEffects, const std::remove_reference_t<VN>...>&&
- ReceiveValue<
- Out,
- std::remove_reference_t<VN>...>)
- void value(VN&&... vn) {
- ::pushmi::set_value(sideEffects, as_const(vn)...);
- ::pushmi::set_value(out, (VN &&) vn...);
- }
- PUSHMI_TEMPLATE(class E)
- (requires ReceiveError<SideEffects, const E>&&
- ReceiveError<Out, E>)
- void error(E e) noexcept {
- ::pushmi::set_error(sideEffects, as_const(e));
- ::pushmi::set_error(out, std::move(e));
- }
- void done() {
- ::pushmi::set_done(sideEffects);
- ::pushmi::set_done(out);
- }
- PUSHMI_TEMPLATE(class Up, class UUp = std::remove_reference_t<Up>)
- (requires FlowReceiver<SideEffects>&& FlowReceiver<Out>)
- void starting(
- Up&& up) {
- // up is not made const because sideEffects is allowed to call methods on up
- ::pushmi::set_starting(sideEffects, up);
- ::pushmi::set_starting(out, (Up &&) up);
- }
- };
- PUSHMI_INLINE_VAR constexpr struct make_tap_fn {
- PUSHMI_TEMPLATE(class SideEffects, class Out)
- (requires Receiver<SideEffects>&& Receiver<Out>&&
- Receiver<tap_<SideEffects, Out>>)
- auto operator()(SideEffects se, Out out) const {
- return tap_<SideEffects, Out>{std::move(se), std::move(out)};
- }
- } const make_tap{};
- struct tap_fn {
- private:
- PUSHMI_TEMPLATE(class In, class SideEffects)
- (requires Sender<In>&& Receiver<SideEffects>)
- static auto impl(
- In in,
- SideEffects sideEffects) {
- return ::pushmi::detail::sender_from(
- std::move(in),
- ::pushmi::detail::submit_transform_out<In>(
- out_impl<In, SideEffects>{std::move(sideEffects)}));
- }
- template <class... AN>
- struct in_impl {
- std::tuple<AN...> args_;
- PUSHMI_TEMPLATE(class In)
- (requires Sender<In>)
- auto operator()(In in) {
- return tap_fn::impl(
- std::move(in),
- ::pushmi::detail::receiver_from_fn<In>()(std::move(args_)));
- }
- };
- PUSHMI_TEMPLATE(class In, class SideEffects)
- (requires Sender<In>&& Receiver<SideEffects>)
- struct out_impl {
- SideEffects sideEffects_;
- PUSHMI_TEMPLATE(class Out)
- (requires Receiver<Out>&& SenderTo<In, Out>&& SenderTo<
- In,
- decltype(::pushmi::detail::receiver_from_fn<In>()(detail::make_tap(
- std::declval<SideEffects>(),
- std::declval<Out>())))>)
- auto operator()(Out out) const {
- auto gang{::pushmi::detail::receiver_from_fn<In>()(
- detail::make_tap(sideEffects_, std::move(out)))};
- return gang;
- }
- };
- public:
- template <class... AN>
- auto operator()(AN... an) const {
- return in_impl<AN...>{std::tuple<AN...>{std::move(an)...}};
- }
- };
- } // namespace detail
- namespace operators {
- PUSHMI_INLINE_VAR constexpr detail::tap_fn tap{};
- } // namespace operators
- } // namespace pushmi
|