// Copyright (c) 2020 by Chrono

//

// sudo apt-get install libzmq3-dev

//

// g++ zmq.cpp -std=c++14 -lzmq -lpthread -o a.out;./a.out

#include <iostream>

#include <string>

#include <vector>

#include <future>

#include <zmq.hpp>

using namespace std;

const auto thread_num = 1;

zmq::context_t context(thread_num);

auto make_sock = [&](auto mode)

{

return zmq::socket_t(context, mode);

};

void case1()

{

const auto addr = "ipc:///dev/shm/zmq.sock"s;

//const auto addr = "inproc:///dev/shm/zmq.sock"s;

auto receiver = [=]()

{

auto sock = make_sock(ZMQ_PULL);

sock.bind(addr);

assert(sock.connected());

zmq::message_t msg;

sock.recv(&msg);

cout << msg.size() << endl;

string s = {msg.data<char>(), msg.size()};

cout << s << endl;

};

auto sender = [=]()

{

auto sock = make_sock(ZMQ_PUSH);

sock.connect(addr);

assert(sock.connected());

string s = "hello zmq";

//zmq::message_t msg(begin(s), end(s));

//sock.send(msg);

sock.send(s.data(), s.size());

};

sender();

receiver();

}

void case2()

{

const auto addr = "tcp://127.0.0.1:5555"s;

auto srv = [=]()

{

auto sock = make_sock(ZMQ_PULL);

sock.setsockopt(ZMQ_LINGER, 0);

sock.setsockopt(ZMQ_RCVHWM, 1000);

sock.bind(addr);

assert(sock.connected());

zmq::message_t msg;

for(int i = 0; i < 4; i++) {

sock.recv(&msg);

string s = {msg.data<char>(), msg.size()};

cout << s << endl;

}

};

auto client = [=](int i)

{

auto sock = make_sock(ZMQ_PUSH);

sock.setsockopt(ZMQ_SNDHWM, 100);

sock.connect(addr);

assert(sock.connected());

string s = "zmq msg " + to_string(i);

//zmq::message_t msg(begin(s), end(s));

//sock.send(msg);

//sock.send(begin(s), end(s));

sock.send(s.data(), s.size());

};

auto f = std::async(srv);

for(int i = 0; i < 4; i++) {

client(i);

}

f.wait();

}

int main()

{

int a, b, c;

std::tie(a, b, c) = zmq::version();

cout << "version = "

<< a << b << c << endl;

case1();

case2();

cout << "zmq demo" << endl;

}