RestEndpoint.hpp

Go to the documentation of this file.

#ifndef RESTCMD_INCLUDE_RESTCMD_RESTENDPOINT_HPP_
#define RESTCMD_INCLUDE_RESTCMD_RESTENDPOINT_HPP_
 
#include "restcmd/Issues.hpp"
 
#include "cmdlib/cmd/Nljs.hpp"
 
#include <nlohmann/json.hpp>
 
#include <pistache/client.h>
#include <pistache/description.h>
#include <pistache/endpoint.h>
#include <pistache/http.h>
#include <pistache/http_header.h>
#include <pistache/mime.h>
#include <pistache/router.h>
 
#include <chrono>
#include <future>
#include <memory>
#include <string>
#include <thread>
#include <vector>
 
namespace dunedaq::restcmd {
 
typedef nlohmann::json cmdobj_t;
 
class RestEndpoint
{
public:
  explicit RestEndpoint(const std::string& /*uri*/,
                        int port,
                        std::function<void(const cmdobj_t&, cmdlib::cmd::CommandReply)> callback) noexcept
    : port_{ static_cast<uint16_t>(port) } // NOLINT(build/unsigned)
    , address_{ Pistache::Ipv4::any(), port_ }
    , http_endpoint_{ std::make_shared<Pistache::Http::Endpoint>(address_) }
    , description_{ "DUNE DAQ cmdlib API", "0.1" }
    , accepted_mime_{ MIME(Application, Json) }
    , http_client_{ std::make_shared<Pistache::Http::Client>() }
    , command_callback_{ callback }
  {
  }
 
  void init(size_t threads);
  void start();
  void stop();
  void shutdown();
 
  // Client handler
  void handleResponseCommand(const cmdobj_t& cmd, cmdlib::cmd::CommandReply& meta);
  uint16_t getPort() const { return static_cast<uint16_t>(port_); } // NOLINT(build/unsigned)
  std::shared_ptr<Pistache::Http::Client> getHttpClient() const { return http_client_; }
 
private:
  void createRouting();
  void createDescription();
  // void serveTask();
 
  // Route handler
  void handle_route_command(const Pistache::Rest::Request&, Pistache::Http::ResponseWriter response);
 
  // REST
  Pistache::Port port_;
  Pistache::Address address_;
  std::shared_ptr<Pistache::Http::Endpoint> http_endpoint_;
  Pistache::Rest::Description description_;
  Pistache::Rest::Router router_;
  Pistache::Http::Mime::MediaType accepted_mime_;
 
  // CLIENT
  std::shared_ptr<Pistache::Http::Client> http_client_;
  Pistache::Http::Client::Options http_client_options_;
  std::vector<Pistache::Async::Promise<Pistache::Http::Response>> http_client_responses_;
 
  // Function to call with received POST bodies
  std::function<void(const cmdobj_t&, cmdlib::cmd::CommandReply)> command_callback_;
 
  // Background server thread
  std::thread server_thread_;
};
 
} // namespace dunedaq::restcmd
 
#endif // RESTCMD_INCLUDE_RESTCMD_RESTENDPOINT_HPP_