zmqx.hpp

Go to the documentation of this file.

#ifndef CORAL_NET_ZMQX_HPP
#define CORAL_NET_ZMQX_HPP

#include <chrono>
#include <cstdint>
#include <cstring>
#include <memory>
#include <string>
#include <vector>

#include "zmq.hpp"

#include "coral/config.h"
#include "coral/net.hpp"


namespace coral
{
namespace net
{

namespace zmqx
{


zmq::context_t& GlobalContext();


std::uint16_t BindToEphemeralPort(
    zmq::socket_t& socket,
    const std::string& networkInterface = "*");


std::string LastEndpoint(zmq::socket_t& socket);


std::uint16_t EndpointPort(const std::string& endpoint);


bool WaitForOutgoing(zmq::socket_t& socket, std::chrono::milliseconds timeout);


bool WaitForIncoming(zmq::socket_t& socket, std::chrono::milliseconds timeout);


enum class SendFlag : int
{
    none = 0,

    more = 1
};
CORAL_DEFINE_BITWISE_ENUM_OPERATORS(SendFlag)



void Send(
    zmq::socket_t& socket,
    std::vector<zmq::message_t>& message,
    SendFlag flags = SendFlag::none);


void Receive(
    zmq::socket_t& socket,
    std::vector<zmq::message_t>& message);


std::string ToString(const zmq::message_t& frame);


zmq::message_t ToFrame(const std::string& s);


class ReqSocket
{
public:
    ReqSocket();

    CORAL_DEFINE_DEFAULT_MOVE(ReqSocket, m_socket)

    ~ReqSocket() CORAL_NOEXCEPT;

    void Connect(const coral::net::Endpoint& server);

    void Bind(const coral::net::Endpoint& localEndpoint);

    void Close();

    void Send(std::vector<zmq::message_t>& msg);

    void Receive(std::vector<zmq::message_t>& msg);

    zmq::socket_t& Socket();
    const zmq::socket_t& Socket() const;

private:
    std::unique_ptr<zmq::socket_t> m_socket;
};


class RepSocket
{
public:
    RepSocket();

    CORAL_DEFINE_DEFAULT_MOVE(RepSocket,
        m_socket, m_boundEndpoint, m_clientEnvelope)

    ~RepSocket() CORAL_NOEXCEPT;

    void Bind(const coral::net::Endpoint& localEndpoint);

    void Connect(const coral::net::Endpoint& clientEndpoint);

    void Close();

    const coral::net::Endpoint& BoundEndpoint() const;

    void Receive(std::vector<zmq::message_t>& msg);

    void Send(std::vector<zmq::message_t>& msg);

    void Ignore();

    zmq::socket_t& Socket();
    const zmq::socket_t& Socket() const;

private:
    std::unique_ptr<zmq::socket_t> m_socket;
    coral::net::Endpoint m_boundEndpoint;
    std::vector<zmq::message_t> m_clientEnvelope;
};


bool Receive(
    RepSocket& socket,
    std::vector<zmq::message_t>& message,
    std::chrono::milliseconds timeout);


}}} // namespace
#endif // header guard