ZMQModuleMessage.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2018 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Nils Braun, Anselm Baur                                  *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
#pragma once
 
#include <framework/pcore/zmq/messages/ZMQMessageHelper.h>
 
#include <zmq.hpp>
#include <memory>
#include <array>
 
namespace Belle2 {
  template <unsigned int AMessageFrameNumber>
  class ZMQModuleMessage {
    friend class ZMQMessageFactory;
 
  public:
    static constexpr unsigned int c_messageParts = AMessageFrameNumber;
 
    using MessageParts = std::array<zmq::message_t, ZMQModuleMessage::c_messageParts>;
 
    static void toSocket(std::unique_ptr<ZMQModuleMessage> message, const std::unique_ptr<zmq::socket_t>& socket)
    {
      for (unsigned int i = 0; i < c_messageParts - 1; i++) {
        socket->send(message->m_messageParts[i], ZMQ_SNDMORE);
      }
      socket->send(message->m_messageParts[c_messageParts - 1]);
    }
 
    ZMQModuleMessage(const ZMQModuleMessage&) = delete;
    void operator=(const ZMQModuleMessage&) = delete;
 
    MessageParts& getMessageParts()
    {
      return m_messageParts;
    };
 
    const MessageParts& getMessageParts() const
    {
      return m_messageParts;
    };
 
    template <unsigned int index>
    const zmq::message_t& getMessagePart() const
    {
      return m_messageParts[index];
    }
 
    template <unsigned int index>
    const char* getMessagePartAsCharArray() const
    {
      const auto& messagePart = getMessagePart<index>();
      return static_cast<const char*>(messagePart.data());
    }
 
    template <unsigned int index>
    std::string getMessagePartAsString() const
    {
      const auto& messagePart = getMessagePart<index>();
      return std::string(static_cast<const char*>(messagePart.data()), messagePart.size());
    }
 
    template <unsigned int index>
    zmq::message_t& getMessagePart()
    {
      return m_messageParts[index];
    }
 
    template <unsigned int index>
    char* getMessagePartAsCharArray()
    {
      auto& messagePart = getMessagePart<index>();
      return static_cast<char*>(messagePart.data());
    }
 
  protected:
    ZMQModuleMessage() = default;
 
    template <class ...T>
    explicit ZMQModuleMessage(T&& ... arguments) :
      m_messageParts( {ZMQMessageHelper::createZMQMessage(std::forward<T>(arguments)) ... })
    {
    }
 
  private:
    MessageParts m_messageParts;
  };
}