client_reactor.hpp

//                  ReactivePlusPlus library
//
//          Copyright Aleksey Loginov 2023 - present.
// Distributed under the Boost Software License, Version 1.0.
//    (See accompanying file LICENSE_1_0.txt or copy at
//          https://www.boost.org/LICENSE_1_0.txt)
//
// Project home: https://github.com/victimsnino/ReactivePlusPlus
//
 
#pragma once
 
#include <rpp/observables/fwd.hpp>
 
#include <grpcpp/support/client_callback.h>
#include <rppgrpc/details/base.hpp>
#include <rppgrpc/fwd.hpp>
#include <rppgrpc/utils/exceptions.hpp>
 
namespace rppgrpc
{
    template<rpp::constraint::decayed_type Request, rpp::constraint::decayed_type Response>
    class client_bidi_reactor final : public grpc::ClientBidiReactor<Request, Response>
        , private details::base_writer<Request>
        , private details::base_reader<Response>
    {
        using Base = grpc::ClientBidiReactor<Request, Response>;
 
    public:
        client_bidi_reactor() = default;
 
        void init()
        {
            Base::StartCall();
            details::base_reader<Response>::handle_read_done(true);
        }
 
        using details::base_writer<Request>::get_observer;
        using details::base_reader<Response>::get_observable;
 
    private:
        void start_read(Response& data) override
        {
            Base::StartRead(&data);
        }
 
        void start_write(const Request& v) override
        {
            Base::StartWrite(&v);
        }
 
        void finish_writes(const grpc::Status&) override
        {
            Base::StartWritesDone();
        }
 
        using Base::StartCall;
        using Base::StartRead;
 
        void OnReadDone(bool ok) override
        {
            if (!ok)
                return;
 
            details::base_reader<Response>::handle_read_done();
        }
 
        void OnWriteDone(bool ok) override
        {
            if (!ok)
                return;
 
            details::base_writer<Request>::handle_write_done();
        }
 
        void OnDone(const grpc::Status& s) override
        {
            details::base_writer<Request>::handle_on_done();
            details::base_reader<Response>::handle_on_done(s.ok() ? std::exception_ptr{} : std::make_exception_ptr(rppgrpc::utils::reactor_failed{s.error_message()}));
            delete this;
        }
    };
 
    template<rpp::constraint::decayed_type Request>
    class client_write_reactor final : public grpc::ClientWriteReactor<Request>
        , private details::base_writer<Request>
        , private details::base_reader<rpp::utils::none>
    {
        using Base = grpc::ClientWriteReactor<Request>;
 
    public:
        client_write_reactor() = default;
 
        void init()
        {
            Base::StartCall();
        }
 
        using details::base_writer<Request>::get_observer;
        using details::base_reader<rpp::utils::none>::get_observable;
 
    private:
        void start_read(rpp::utils::none& data) override
        {
        }
 
        void start_write(const Request& v) override
        {
            Base::StartWrite(&v);
        }
 
        void finish_writes(const grpc::Status&) override
        {
            Base::StartWritesDone();
        }
 
        using Base::StartCall;
 
        void OnWriteDone(bool ok) override
        {
            if (!ok)
                return;
 
            details::base_writer<Request>::handle_write_done();
        }
 
        void OnDone(const grpc::Status& s) override
        {
            details::base_writer<Request>::handle_on_done();
            details::base_reader<rpp::utils::none>::handle_on_done(s.ok() ? std::exception_ptr{} : std::make_exception_ptr(rppgrpc::utils::reactor_failed{s.error_message()}));
 
            delete this;
        }
    };
 
    template<rpp::constraint::decayed_type Response>
    class client_read_reactor final : public grpc::ClientReadReactor<Response>
        , private details::base_reader<Response>
    {
        using Base = grpc::ClientReadReactor<Response>;
 
    public:
        client_read_reactor() = default;
 
        void init()
        {
            Base::StartCall();
            details::base_reader<Response>::handle_read_done(true);
        }
 
        using details::base_reader<Response>::get_observable;
 
    private:
        using Base::StartCall;
        using Base::StartRead;
 
        void start_read(Response& data) override
        {
            Base::StartRead(&data);
        }
 
        void OnReadDone(bool ok) override
        {
            if (!ok)
                return;
 
            details::base_reader<Response>::handle_read_done();
        }
 
        void OnDone(const grpc::Status& s) override
        {
            details::base_reader<Response>::handle_on_done(s.ok() ? std::exception_ptr{} : std::make_exception_ptr(rppgrpc::utils::reactor_failed{s.error_message()}));
            delete this;
        }
    };
} // namespace rppgrpc