Message.h

/********************************************************************************
 *    Copyright (C) 2018 GSI Helmholtzzentrum fuer Schwerionenforschung GmbH    *
 *                                                                              *
 *              This software is distributed under the terms of the             *
 *              GNU Lesser General Public Licence (LGPL) version 3,             *
 *                  copied verbatim in the file "LICENSE"                       *
 ********************************************************************************/

#ifndef FAIR_MQ_OFI_MESSAGE_H
#define FAIR_MQ_OFI_MESSAGE_H

#include <FairMQMessage.h>
#include <FairMQUnmanagedRegion.h>

#include <asiofi.hpp>
#include <atomic>
#include <cstddef> // size_t
#include <zmq.h>

namespace fair
{
namespace mq
{
namespace ofi
{

class Message final : public fair::mq::Message
{
  public:
    Message(boost::container::pmr::memory_resource* pmr);
    Message(boost::container::pmr::memory_resource* pmr, const size_t size);
    Message(boost::container::pmr::memory_resource* pmr,
            void* data,
            const size_t size,
            fairmq_free_fn* ffn,
            void* hint = nullptr);
    Message(boost::container::pmr::memory_resource* pmr,
            FairMQUnmanagedRegionPtr& region,
            void* data,
            const size_t size,
            void* hint = 0);

    Message(const Message&) = delete;
    Message operator=(const Message&) = delete;

    auto Rebuild() -> void override;
    auto Rebuild(const size_t size) -> void override;
    auto Rebuild(void* data, const size_t size, fairmq_free_fn* ffn, void* hint = nullptr) -> void override;

    auto GetData() const -> void* override;
    auto GetSize() const -> size_t override;

    auto SetUsedSize(const size_t size) -> bool override;

    auto GetType() const -> fair::mq::Transport override { return fair::mq::Transport::OFI; }

    auto Copy(const fair::mq::Message& msg) -> void override;

    ~Message() override;

  private:
    size_t fInitialSize;
    size_t fSize;
    void* fData;
    fairmq_free_fn* fFreeFunction;
    void* fHint;
    boost::container::pmr::memory_resource* fPmr;
}; /* class Message */  

} /* namespace ofi */
} /* namespace mq */
} /* namespace fair */

#endif /* FAIR_MQ_OFI_MESSAGE_H */