requestor.hpp

Go to the documentation of this file.

/**
 * @file
 * @ingroup daq_ocm_serverctl
 * @copyright ESO - European Southern Observatory
 *
 * @brief
 */
#ifndef OCF_OCM_SERVERCTL_REQUESTOR_HPP_
#define OCF_OCM_SERVERCTL_REQUESTOR_HPP_
 
#include <iostream>
#include <type_traits>
#include <unordered_map>
#include <vector>
 
#include <CLI/App.hpp>
#include <boost/asio/io_service.hpp>
 
#include <Daqif.hpp>
#include <Stdif.hpp>
#include <rad/logger.hpp>
 
std::ostream& operator<<(std::ostream& os, std::shared_ptr<daqif::DaqStatus> const& s);
std::ostream& operator<<(std::ostream& os, std::shared_ptr<daqif::DaqReply> const& s);
std::ostream& operator<<(std::ostream& os, std::vector<std::shared_ptr<daqif::DaqStatus>> const& s);
std::ostream&
operator<<(std::ostream& os, std::vector<std::shared_ptr<daqif::AwaitDaqReply>> const& s);
 
template <class T>
struct JsonPrint {
    JsonPrint(T const& t_arg) : t(t_arg) {
    }
    T const& t;
};
 
std::ostream& operator<<(std::ostream& os, JsonPrint<std::shared_ptr<daqif::DaqStatus>> const& s);
std::ostream& operator<<(std::ostream& os, JsonPrint<std::shared_ptr<daqif::DaqReply>> const& s);
std::ostream&
operator<<(std::ostream& os, JsonPrint<std::shared_ptr<daqif::AwaitDaqReply>> const& s);
std::ostream& operator<<(std::ostream& os, JsonPrint<std::string> const& s);
std::ostream&
operator<<(std::ostream& os, JsonPrint<std::vector<std::shared_ptr<daqif::DaqStatus>>> const& s);
 
struct CommonArgs {
    bool json = false;
    bool status = false;
    unsigned timeout = 10;
    std::string req_addr;
    std::string pub_addr;
    elt::mal::Mal* mal;
 
    CommonArgs() = default;
    CommonArgs(CommonArgs const&) = delete;
    CommonArgs& operator=(CommonArgs const&) = delete;
};
 
struct Requestor {
    explicit Requestor(CommonArgs& args);
    virtual ~Requestor() {
    }
    /**
     * Add arguments to command.
     */
    virtual void AddOptions(CLI::App* app);
 
    /**
     * Receives a parsed variables_map that include the options added with AddOptions.
     */
    virtual void
    Handle(::stdif::StdCmdsSync& std_cmds, ::daqif::OcmDaqControlSync& daq_cmds, CLI::App* arg) = 0;
 
protected:
    CommonArgs& m_args;
};
 
struct SetLogLevelRequestor : Requestor {
    using Requestor::Requestor;
    virtual void AddOptions(CLI::App* app) override;
    virtual void Handle(::stdif::StdCmdsSync& std_cmds,
                        ::daqif::OcmDaqControlSync& daq_cmds,
                        CLI::App* arg) override;
 
private:
    std::string m_logger;
    std::string m_level;
};
 
/**
 * Simple requestor for commands without argument
 *
 * @tparam Func is either a StdCmdsSync or OcmDaqControlSync member function pointer, taking no
 * arguments and returning a type that has an ostream << operator: E.g.: `using Func =
 * decltype(&daqif::StdCmds::Enable)`;
 */
template <class Func>
struct SimpleRequestor : Requestor {
    SimpleRequestor(Func func, CommonArgs& args) : Requestor(args), m_func(func) {
    }
    void Handle(::stdif::StdCmdsSync& std_cmds,
                ::daqif::OcmDaqControlSync& daq_cmds,
                CLI::App* arg) override {
        if constexpr (std::is_invocable_v<Func, decltype(std_cmds)>) {
            auto reply = (std_cmds.*m_func)();
            if (m_args.json) {
                std::cout << JsonPrint(reply);
            } else {
                std::cout << reply << std::endl;
            }
        } else if constexpr (std::is_invocable_v<Func, decltype(daq_cmds)>) {
            auto reply = (daq_cmds.*m_func)();
            if (m_args.json) {
                std::cout << JsonPrint(reply);
            } else {
                std::cout << reply << std::endl;
            }
        } else {
        }
    }
 
    Func m_func;
};
 
/**
 * Simple Daq commands that accepts a single argument `id` and returns a shared_ptr type
 * that can be formatted with ostream<< operator when dereferenced.
 */
template <class Func>
struct SimpleDaqRequestor : Requestor {
    SimpleDaqRequestor(Func func, CommonArgs& args) : Requestor(args), m_func(func) {
    }
 
    void AddOptions(CLI::App* sub) override {
        sub->add_option("id", m_id, "DAQ ID");
    }
 
    void Handle(::stdif::StdCmdsSync& std_cmds,
                ::daqif::OcmDaqControlSync& daq_cmds,
                CLI::App* arg) override {
        auto reply = (daq_cmds.*m_func)(m_id);
        if (m_args.json) {
            std::cout << JsonPrint(reply);
        } else {
            std::cout << reply << std::endl;
        }
    }
 
private:
    std::string m_id;
    Func m_func;
};
 
template <class Func>
struct NoArgRequestor : Requestor {
    NoArgRequestor(Func func, CommonArgs& args) : Requestor(args), m_func(func) {
    }
 
    void Handle(::stdif::StdCmdsSync& std_cmds,
                ::daqif::OcmDaqControlSync& daq_cmds,
                CLI::App* arg) override {
        auto reply = (daq_cmds.*m_func)();
        if (m_args.json) {
            std::cout << JsonPrint(reply);
        } else {
            std::cout << reply << std::endl;
        }
    }
 
    Func m_func;
};
 
struct StartDaqRequestor : Requestor {
    std::string m_id;
    std::string m_prefix;
    std::string m_primary_sources;
    std::string m_metadata_sources;
    std::string m_properties;
 
    using Requestor::Requestor;
 
    void AddOptions(CLI::App* app) override;
    void Handle(::stdif::StdCmdsSync& std_cmds,
                ::daqif::OcmDaqControlSync& daq_cmds,
                CLI::App* arg) override;
 
private:
};
 
struct StartDaqV2Requestor : Requestor {
    std::string m_spec;
 
    using Requestor::Requestor;
 
    void AddOptions(CLI::App* app) override;
    void Handle(::stdif::StdCmdsSync& std_cmds,
                ::daqif::OcmDaqControlSync& daq_cmds,
                CLI::App* arg) override;
 
private:
};
 
struct UpdateKeywordsRequestor : Requestor {
    using Requestor::Requestor;
 
    void AddOptions(CLI::App* app) override;
    void Handle(::stdif::StdCmdsSync& std_cmds,
                ::daqif::OcmDaqControlSync& daq_cmds,
                CLI::App* arg) override;
 
private:
    std::string m_id;
    std::string m_keywords;
};
 
struct AwaitStateRequestor : Requestor {
    using Requestor::Requestor;
 
    void AddOptions(CLI::App* app) override;
    void Handle(::stdif::StdCmdsSync& std_cmds,
                ::daqif::OcmDaqControlSync& daq_cmds,
                CLI::App* arg) override;
 
private:
    std::string m_id;
    std::string m_state;
    std::string m_substate;
    double m_timeout;
};
 
#endif  // #ifndef OCF_OCM_SERVERCTL_REQUESTOR_HPP_