mirror of https://github.com/nostar/urfd.git
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
58 lines
1.4 KiB
58 lines
1.4 KiB
#include "NNGPublisher.h"
|
|
#include <iostream>
|
|
|
|
CNNGPublisher::CNNGPublisher()
|
|
: m_started(false)
|
|
{
|
|
m_sock.id = 0;
|
|
}
|
|
|
|
CNNGPublisher::~CNNGPublisher()
|
|
{
|
|
Stop();
|
|
}
|
|
|
|
bool CNNGPublisher::Start(const std::string &addr)
|
|
{
|
|
std::lock_guard<std::mutex> lock(m_mutex);
|
|
if (m_started) return true;
|
|
|
|
int rv;
|
|
if ((rv = nng_pub0_open(&m_sock)) != 0) {
|
|
std::cerr << "NNG: Failed to open pub socket: " << nng_strerror(rv) << std::endl;
|
|
return false;
|
|
}
|
|
|
|
if ((rv = nng_listen(m_sock, addr.c_str(), nullptr, 0)) != 0) {
|
|
std::cerr << "NNG: Failed to listen on " << addr << ": " << nng_strerror(rv) << std::endl;
|
|
nng_close(m_sock);
|
|
return false;
|
|
}
|
|
|
|
m_started = true;
|
|
std::cout << "NNG: Publisher started at " << addr << std::endl;
|
|
return true;
|
|
}
|
|
|
|
void CNNGPublisher::Stop()
|
|
{
|
|
std::lock_guard<std::mutex> lock(m_mutex);
|
|
if (!m_started) return;
|
|
|
|
nng_close(m_sock);
|
|
m_started = false;
|
|
std::cout << "NNG: Publisher stopped" << std::endl;
|
|
}
|
|
|
|
void CNNGPublisher::Publish(const nlohmann::json &event)
|
|
{
|
|
std::lock_guard<std::mutex> lock(m_mutex);
|
|
if (!m_started) return;
|
|
|
|
std::string msg = event.dump();
|
|
int rv = nng_send(m_sock, (void *)msg.c_str(), msg.size(), NNG_FLAG_NONBLOCK);
|
|
if (rv != 0 && rv != NNG_EAGAIN) {
|
|
std::cerr << "NNG: Send error: " << nng_strerror(rv) << std::endl;
|
|
}
|
|
}
|