RocLogger.cpp

Go to the documentation of this file.

#include <RocLogger.h>

#include <cstring>

using namespace std;

RocLogger::RocLogger(zmq::context_t & _ctx,
                     const std::string & _endpoint,
                     const std::string & _node,
                     const std::string & _source):
            node(_node), source(_source) {
    int linger = 100;
    unique_lock<mutex> _(socketMutex);
    socket = new zmq::socket_t(_ctx, ZMQ_PUB);
    socket->setsockopt(ZMQ_LINGER, &linger, sizeof(linger));
    socket->connect(_endpoint.c_str());
}

RocLogger::~RocLogger() {
    unique_lock<mutex> _(socketMutex);
    delete socket;
}

void RocLogger::debug(const std::string & _msg) {
    sendMessage('D', _msg);
}

void RocLogger::info(const std::string & _msg) {
    sendMessage('I', _msg);
}

void RocLogger::warning(const std::string & _msg) {
    sendMessage('W', _msg);
}

void RocLogger::error(const std::string & _msg) {
    sendMessage('E', _msg);
}

void RocLogger::critical(const std::string & _msg) {
    sendMessage('C', _msg);
}


void RocLogger::sendMessage(char code, const std::string & _msg) {
    zmq::message_t header(6 + node.length());
    memcpy(((char*)header.data()) + 0, "ROCL\x01", 5);
    ((char*)header.data())[5] = code;
    memcpy(((char*)header.data()) + 6, node.c_str(), node.length());
    zmq::message_t sourceMsg(source.length());
    memcpy(sourceMsg.data(), source.c_str(), source.length());
    zmq::message_t data(_msg.length());
    memcpy(data.data(), _msg.c_str(), _msg.length());
    unique_lock<mutex> _(socketMutex); /* not ideal, but simplifies multithreaded logging a lot */
    socket->send(header, ZMQ_SNDMORE);
    socket->send(sourceMsg, ZMQ_SNDMORE);
    socket->send(data);
}