FairMQ/fairmq/nanomsg/FairMQPollerNN.cxx

/********************************************************************************
 *    Copyright (C) 2014 GSI Helmholtzzentrum fuer Schwerionenforschung GmbH    *
 *                                                                              *
 *              This software is distributed under the terms of the             * 
 *         GNU Lesser General Public Licence version 3 (LGPL) version 3,        *  
 *                  copied verbatim in the file "LICENSE"                       *
 ********************************************************************************/
/**
 * FairMQPollerNN.cxx
 *
 * @since 2014-01-23
 * @author A. Rybalchenko
 */

#include <cstdlib> // quick_exit()

#include <nanomsg/nn.h>
#include <nanomsg/pipeline.h>
#include <nanomsg/pubsub.h>
#include <nanomsg/reqrep.h>
#include <nanomsg/pair.h>

#include "FairMQPollerNN.h"
#include "FairMQLogger.h"

using namespace std;

FairMQPollerNN::FairMQPollerNN(const vector<FairMQChannel>& channels)
    : items()
    , fNumItems(0)
    , fOffsetMap()
{
    fNumItems = channels.size();
    items = new nn_pollfd[fNumItems];

    for (int i = 0; i < fNumItems; ++i)
    {
        items[i].fd = channels.at(i).fSocket->GetSocket(1);
        items[i].events = NN_POLLIN;
    }
}

FairMQPollerNN::FairMQPollerNN(map<string, vector<FairMQChannel>>& channelsMap, initializer_list<string> channelList)
    : items()
    , fNumItems(0)
    , fOffsetMap()
{
    int offset = 0;

    try
    {
        // calculate offsets and the total size of the poll item set
        for (string channel : channelList)
        {
            fOffsetMap[channel] = offset;
            offset += channelsMap.at(channel).size();
            fNumItems += channelsMap.at(channel).size();
        }

        items = new nn_pollfd[fNumItems];

        int index = 0;
        for (string channel : channelList)
        {
            for (int i = 0; i < channelsMap.at(channel).size(); ++i)
            {
                index = fOffsetMap[channel] + i;
                items[index].fd = channelsMap.at(channel).at(i).fSocket->GetSocket(1);
                items[index].events = NN_POLLIN;
            }
        }
    }
    catch (const std::out_of_range& oor)
    {
        LOG(ERROR) << "At least one of the provided channel keys for poller initialization is invalid";
        LOG(ERROR) << "Out of Range error: " << oor.what() << '\n';
        quick_exit(EXIT_FAILURE);
    }
}

FairMQPollerNN::FairMQPollerNN(FairMQSocket& cmdSocket, FairMQSocket& dataSocket)
    : items()
    , fNumItems(2)
    , fOffsetMap()
{
    items = new nn_pollfd[fNumItems];

    items[0].fd = cmdSocket.GetSocket(1);
    items[0].events = NN_POLLIN;

    int type = 0;
    size_t sz = sizeof(type);
    nn_getsockopt(dataSocket.GetSocket(1), NN_SOL_SOCKET, NN_PROTOCOL, &type, &sz);

    items[1].fd = dataSocket.GetSocket(1);

    if (type == NN_REQ || type == NN_REP || type == NN_PAIR)
    {
        items[1].events = NN_POLLIN|NN_POLLOUT;
    }
    else if (type == NN_PUSH || type == NN_PUB)
    {
        items[1].events = NN_POLLOUT;
    }
    else if (type == NN_PULL || type == NN_SUB)
    {
        items[1].events = NN_POLLIN;
    }
    else
    {
        LOG(ERROR) << "invalid poller configuration, exiting.";
        quick_exit(EXIT_FAILURE);
    }
}

void FairMQPollerNN::Poll(int timeout)
{
    if (nn_poll(items, fNumItems, timeout) < 0)
    {
        if (errno == ETERM)
        {
            LOG(DEBUG) << "polling exited, reason: " << nn_strerror(errno);
        }
        else
        {
            LOG(ERROR) << "polling failed, reason: " << nn_strerror(errno);
        }
    }
}

bool FairMQPollerNN::CheckInput(const int index)
{
    if (items[index].revents & (NN_POLLIN | NN_POLLOUT))
    {
        return true;
    }

    return false;
}

bool FairMQPollerNN::CheckOutput(const int index)
{
    if (items[index].revents & NN_POLLOUT)
    {
        return true;
    }

    return false;
}

bool FairMQPollerNN::CheckInput(const string channelKey, const int index)
{
    try
    {
        if (items[fOffsetMap.at(channelKey) + index].revents & NN_POLLIN)
        {
            return true;
        }

        return false;
    }
    catch (const std::out_of_range& oor)
    {
        LOG(ERROR) << "Invalid channel key: \"" << channelKey << "\"";
        LOG(ERROR) << "Out of Range error: " << oor.what() << '\n';
        quick_exit(EXIT_FAILURE);
    }
}

bool FairMQPollerNN::CheckOutput(const string channelKey, const int index)
{
    try
    {
        if (items[fOffsetMap.at(channelKey) + index].revents & NN_POLLOUT)
        {
            return true;
        }

        return false;
    }
    catch (const std::out_of_range& oor)
    {
        LOG(ERROR) << "Invalid channel key: \"" << channelKey << "\"";
        LOG(ERROR) << "Out of Range error: " << oor.what() << '\n';
        quick_exit(EXIT_FAILURE);
    }
}

FairMQPollerNN::~FairMQPollerNN()
{
    if (items != NULL)
    {
        delete[] items;
    }
}