examples/augmentor_ex.h

#pragma once

#include "rtbkit/core/agent_configuration/agent_configuration_listener.h"
#include "rtbkit/plugins/augmentor/augmentor_base.h"
#include "soa/service/zmq_named_pub_sub.h"
#include "soa/service/service_base.h"

#include <string>
#include <memory>

namespace RTBKIT {

struct FrequencyCapStorage;

struct AgentConfigEntry;

/******************************************************************************/
/* FREQUENCY CAP AUGMENTOR                                                    */
/******************************************************************************/

struct FrequencyCapAugmentor :
    public RTBKIT::SyncAugmentor
{

    FrequencyCapAugmentor(
            std::shared_ptr<Datacratic::ServiceProxies> services,
            const std::string& serviceName,
            const std::string& augmentorName = "frequency-cap-ex");

    void init();

private:

    virtual RTBKIT::AugmentationList
    onRequest(const RTBKIT::AugmentationRequest& request);

    size_t getCap(
            const std::string& augmentor,
            const std::string& agent,
            const RTBKIT::AgentConfigEntry& config) const;


    std::shared_ptr<FrequencyCapStorage> storage;

    RTBKIT::AgentConfigurationListener agentConfig;
    Datacratic::ZmqNamedMultipleSubscriber palEvents;
};


} // namespace RTBKIT