ZeroMQServer.cpp File Reference

#include "zeromq/ZeroMQServer.hpp"
#include "Version.h"
#include "utils/ErrorHelpers.hpp"
#include "utils/Hasher.hpp"
#include <spdlog/spdlog.h>

Include dependency graph for ZeroMQServer.cpp:

Go to the source code of this file.

Functions
bool	ZeroMQServerMessageCallback (const std::vector< zmq::message_t > &recvMsgs, std::vector< zmq::message_t > &replyMsgs)

Variables
constexpr uint32_t	LOG_LEVEL_ID = ('L' | ('O' << 8) | ('G' << 16) | ('L' << 24))
constexpr uint32_t	VERSION_INFO_ID = ('V' | ('E' << 8) | ('R' << 16) | ('I' << 24))
constexpr uint32_t	PING_PONG_ID = ('P' | ('I' << 8) | ('N' << 16) | ('G' << 24))
constexpr uint32_t	STATUS_CHECK_ID = ('S' | ('C' << 8) | ('H' << 16) | ('K' << 24))

Function Documentation

ZeroMQServerMessageCallback()

bool ZeroMQServerMessageCallback	(	const std::vector< zmq::message_t > &	recvMsgs,
		std::vector< zmq::message_t > &	replyMsgs )

ZeroMQ message received callback

Parameters

[in]	recvMsgs	Received messages
[out]	replyMsgs	Reply messages returned by callback

Returns

true If the callback successfully processes the received messages

false otherwise

Definition at line 104 of file ZeroMQServer.cpp.

{
    spdlog::trace("Received {} messages", recvMsgs.size());
    replyMsgs.clear();
 
    std::string replyBody;
    int reply = ZMQ_EVENT_HANDSHAKE_FAILED_NO_DETAIL;
    switch (*(static_cast<const uint64_t *>(recvMsgs[0].data())))
    {
    case LOG_LEVEL_ID: {
        if (recvMsgs.size() != 2)
        {
            spdlog::error("Received unknown number of messages for log level change");
            break;
        }
 
        spdlog::warn("Log level change request received");
        const auto receivedMsg = std::string(static_cast<const char *>(recvMsgs[1].data()), recvMsgs[1].size());
 
        if (receivedMsg == "v")
        {
            spdlog::set_level(spdlog::level::info);
        }
        if (receivedMsg == "vv")
        {
            spdlog::set_level(spdlog::level::debug);
        }
        if (receivedMsg == "vvv")
        {
            spdlog::set_level(spdlog::level::trace);
        }
        reply = ZMQ_EVENT_HANDSHAKE_SUCCEEDED;
        break;
    }
    case VERSION_INFO_ID: {
        if (recvMsgs.size() != 1)
        {
            spdlog::error("Received unknown number of messages for version information");
            break;
        }
 
        reply = ZMQ_EVENT_HANDSHAKE_SUCCEEDED;
        replyBody = PROJECT_FULL_VERSION_STRING;
        break;
    }
    case PING_PONG_ID: {
        if (recvMsgs.size() != 1)
        {
            spdlog::error("Received unknown number of messages for ping");
            break;
        }
 
        reply = ZMQ_EVENT_HANDSHAKE_SUCCEEDED;
        replyBody = "PONG";
        break;
    }
    case STATUS_CHECK_ID: {
        if (recvMsgs.size() != 1)
        {
            spdlog::error("Received unknown number of messages for status check");
            break;
        }
 
        reply = ZMQ_EVENT_HANDSHAKE_SUCCEEDED;
 
        std::ostringstream oss;
        oss << "{";
        for (const auto &entry : vCheckFlag)
        {
            oss << "\"" << entry.first << "\":" << (entry.second->_M_i ? "1," : "0,");
        }
        replyBody = oss.str();
        replyBody.replace(replyBody.size() - 1, 1, "}");
        break;
    }
    /* ################################################################################### */
    /* ############################# MAKE MODIFICATIONS HERE ############################# */
    /* ################################################################################### */
 
    /* ################################################################################### */
    /* ################################ END MODIFICATIONS ################################ */
    /* ################################################################################### */
    default:
        spdlog::error("Unknown command received from control");
        break;
    }
 
    // Prepare reply
    replyMsgs.emplace_back(&reply, sizeof(reply));
    replyMsgs.emplace_back(replyBody.c_str(), replyBody.size());
 
    return reply == ZMQ_EVENT_HANDSHAKE_SUCCEEDED;
}

Here is the caller graph for this function:

Variable Documentation

LOG_LEVEL_ID

uint32_t LOG_LEVEL_ID = ('L' | ('O' << 8) | ('G' << 16) | ('L' << 24))

constexpr

Definition at line 9 of file ZeroMQServer.cpp.

PING_PONG_ID

uint32_t PING_PONG_ID = ('P' | ('I' << 8) | ('N' << 16) | ('G' << 24))

constexpr

Definition at line 11 of file ZeroMQServer.cpp.

STATUS_CHECK_ID

uint32_t STATUS_CHECK_ID = ('S' | ('C' << 8) | ('H' << 16) | ('K' << 24))

constexpr

Definition at line 12 of file ZeroMQServer.cpp.

VERSION_INFO_ID

uint32_t VERSION_INFO_ID = ('V' | ('E' << 8) | ('R' << 16) | ('I' << 24))

constexpr

Definition at line 10 of file ZeroMQServer.cpp.