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
constexpr uint32_t	VERSION_INFO_ID
constexpr uint32_t	PING_PONG_ID
constexpr uint32_t	STATUS_CHECK_ID

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 107 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 &[process, statusFlag] : vCheckFlag)
        {
            oss << "\"" << process << "\":" << (statusFlag->_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

constexpr

Initial value:

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

Definition at line 9 of file ZeroMQServer.cpp.

PING_PONG_ID

uint32_t PING_PONG_ID

constexpr

Initial value:

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

Definition at line 13 of file ZeroMQServer.cpp.

STATUS_CHECK_ID

uint32_t STATUS_CHECK_ID

constexpr

Initial value:

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

Definition at line 15 of file ZeroMQServer.cpp.

VERSION_INFO_ID

uint32_t VERSION_INFO_ID

constexpr

Initial value:

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

Definition at line 11 of file ZeroMQServer.cpp.