requestor.hpp

Go to the documentation of this file.

#ifndef OCF_OCM_SERVERCTL_REQUESTOR_HPP_
#define OCF_OCM_SERVERCTL_REQUESTOR_HPP_

#include <iostream>
#include <unordered_map>
#include <vector>

#include <boost/asio/io_service.hpp>
#include <boost/program_options.hpp>

#include <rad/MsgRequestor.hpp>

#include <ocmServerIf/requests.pb.h>

struct CommonArgs {
    unsigned timeout;
    std::string request_ep;
    std::string publish_ep;

    CommonArgs() = default;
    CommonArgs(CommonArgs const&) = delete;
    CommonArgs& operator=(CommonArgs const&) = delete;
};

struct Requestor {
    Requestor(boost::asio::io_service& ctx, CommonArgs& args);
    virtual ~Requestor() {}
    virtual std::string Synopsis();
    virtual void AddOptions(boost::program_options::options_description& descr);
    virtual int Handle(boost::program_options::variables_map& vm, std::vector<std::string>& opts) = 0;
protected:
    boost::asio::io_service& m_io_ctx;
    CommonArgs& m_args;
};

template <class REQ, class REP>
struct SimpleRequestor : Requestor {
    using Requestor::Requestor;
    int Handle(boost::program_options::variables_map& vm, std::vector<std::string>& opts) override {
        m_requestor = std::make_shared<rad::MsgRequestor2<REQ, REP>>(m_args.request_ep,
                                                                  "ocmserver-ctl", m_io_ctx);
        m_requestor->AsyncSendReceive(REQ(),
                              [](std::error_code const& ec, REP rep) {
                                  if (ec) {
                                      std::cout << "Got error: " << ec.message() << std::endl;
                                  } else {
                                      std::cout << "Got reply: " << rep.reply() << std::endl;
                                  }
                              },
                              std::chrono::seconds(m_args.timeout));
        // The rest will take care of itself. On timeouts the loop will be stopped.
        return 0;
    }
private:
    std::shared_ptr<rad::MsgRequestor2<REQ, REP>> m_requestor;

};

using PingRequestor = SimpleRequestor<ocmServerIf::ReqPing, ocmServerIf::RepPing>;
using StateRequestor = SimpleRequestor<ocmServerIf::ReqState, ocmServerIf::RepState>;
using StatusRequestor = SimpleRequestor<ocmServerIf::ReqStatus, ocmServerIf::RepStatus>;

using InitRequestor = SimpleRequestor<ocmServerIf::ReqInit, ocmServerIf::RepInit>;

/*
struct StatusRequestor : Requestor {
    using Requestor::Requestor;
    virtual std::string Synopsis() override;
    void AddOptions(boost::program_options::options_description& descr) override;
    int Handle(boost::program_options::variables_map& vm, std::vector<std::string>& opts) override;

    std::string m_posarg1;
};*/
#endif // #ifndef OCF_OCM_SERVERCTL_REQUESTOR_HPP_