core/agent_configuration/agent_configuration_service.h

/* configuration_service.h                                         -*- C++ -*-
   Jeremy Barnes, 26 November 2012
   Copyright (c) 2012 Datacratic Inc.  All rights reserved.
*/

#pragma once

#include "soa/service/rest_service_endpoint.h"
#include "soa/service/service_base.h"
#include "soa/service/rest_request_router.h"

#include "rtbkit/core/monitor/monitor_provider.h"

namespace RTBKIT {

using namespace Datacratic;

/*****************************************************************************/
/* AGENT CONFIGURATION SERVICE                                               */
/*****************************************************************************/

struct AgentConfigurationService : public RestServiceEndpoint,
                                   public ServiceBase,
                                   public MonitorProvider
{

    AgentConfigurationService(std::shared_ptr<ServiceProxies> services,
                              const std::string & serviceName = "agentConfigurationService");

    ~AgentConfigurationService()
    {
        shutdown();
        agents.shutdown();
        listeners.shutdown();
    }

    void init();

    void start()
    {
        RestServiceEndpoint::start();
        monitorProviderClient.start();
        //listeners.start();
    }

    void shutdown()
    {
        RestServiceEndpoint::shutdown();
        agents.shutdown();
        listeners.shutdown();
        monitorProviderClient.shutdown();
    }

    void handleAgentHeartbeat(const std::string & agent);
    
    void handleAgentConfig(const std::string & agent,
                           const Json::Value & config);

    Json::Value handleGetAgent(const std::string & agent) const;

    std::vector<std::string> handleGetAgentList() const;

    Json::Value handleGetAllAgents() const;

    void bindTcp();

    RestRequestRouter router;

    ZmqNamedClientBus agents;

    ZmqNamedClientBus listeners;

    struct ListenerInfo {
    };
    
    std::unordered_map<std::string, ListenerInfo> listenerInfo;

    struct AgentInfo {
        Json::Value config;
        std::string configStr;
        Date lastHeartbeat;
    };

    std::unordered_map<std::string, AgentInfo> agentInfo;

    /* Reponds to Monitor requests */
    MonitorProviderClient monitorProviderClient;

    /* MonitorProvider interface */
    std::string getProviderName() const;
    Json::Value getProviderIndicators() const;
};

} // namespace RTBKIT