plugins/data_logger/data_logger.cc

/* data_logger.cc
   Jeremy Barnes, March 2011
   Wolfgang Sourdeau, February 2013
   Copyright (c) 2012, 2013 Datacratic.  All rights reserved.

   Launches the router's logger.
*/


#include "data_logger.h"


using namespace std;
using namespace Datacratic;
using namespace RTBKIT;

DataLogger::
DataLogger(std::shared_ptr<ServiceProxies> proxies)
    : ServiceBase("data_logger", proxies),
      Logger(proxies->zmqContext),
      multipleSubscriber(proxies->zmqContext),
      monitorProviderClient(proxies->zmqContext, *this)
{}

DataLogger::
~DataLogger()
{
    monitorProviderClient.shutdown();
    shutdown();
}

void
DataLogger::
init()
{
    Logger::init();
    monitorProviderClient.init(getServices()->config);

    multipleSubscriber.init(getServices()->config);
    multipleSubscriber.messageHandler
        = [&] (vector<zmq::message_t> && msg) {
        // forward to logger class
        vector<string> s;
        s.reserve(msg.size());
        for (auto & m: msg)
            s.push_back(m.toString());
        this->logMessageNoTimestamp(s);
    };

    //messageLoop.addSource("DataLogger::multipleSubscriber",
    //                      multipleSubscriber);
}

void
DataLogger::
start(std::function<void ()> onStop)
{
    Logger::start(onStop);
    multipleSubscriber.start();
    monitorProviderClient.start();
}

void
DataLogger::
shutdown()
{
    monitorProviderClient.shutdown();
    Logger::shutdown();
    multipleSubscriber.shutdown();
}

void
DataLogger::
connectAllServiceProviders(const string & serviceClass, const string & epName)
{
    multipleSubscriber.connectAllServiceProviders(serviceClass, epName);
}

string
DataLogger::
getProviderName()
    const
{
    return serviceName();
}

Json::Value
DataLogger::
getProviderIndicators()
    const
{
    bool status(true);

    for (const auto & pair: multipleSubscriber.subscribers) {
        if (pair.second->getConnectionState()
            == ZmqNamedSocket::ConnectionState::DISCONNECTED) {
            status = false;
            break;
        }
    }

    Json::Value indicators;
    indicators["status"] = status ? "ok" : "failure";

    return indicators;
}