ZMQ subscriber

zmq_sub.cpp

#include <iostream>

#include "zmq.hpp"

static const char *addr = "tcp://localhost:1337";

void sub_loop(void)
{
	zmq::context_t ctx;
	zmq::socket_t so(ctx, ZMQ_SUB);
	so.connect(addr);
	std::cerr << "zmq socket connected to " << addr << std::endl;

	/* subscribe to everything */
	so.setsockopt(ZMQ_SUBSCRIBE, nullptr, 0);
	std::cerr << "socket subscribed to everything" << std::endl;

	while (true) {
		zmq::message_t msg;
		so.recv(&msg);
		std::cout.write(static_cast<const char *>(msg.data()), msg.size());
		std::cout.flush();
		if (!std::cout.good()) break;
	}

	if (!std::cout) {
		exit(EXIT_FAILURE);
	}
}

int main(int argc, char **argv)
{
	if (argc > 2) {
		std::cerr << "usage: " << argv[0] << " [endpoint_url]" << std::endl;
		exit(EXIT_FAILURE);
	} else if (argc == 2) {
		addr = argv[1];
	}

	try {
		sub_loop();
	} catch (zmq::error_t &err) {
		std::cerr << "zmq error: " << err.what() << std::endl;
		exit(EXIT_FAILURE);
	}

	return 0;
}