common/augmentation.h

#ifndef __rtb__augmentation_h__
#define __rtb__augmentation_h__

#include "rtbkit/common/account_key.h"
#include "soa/jsoncpp/value.h"

#include <set>
#include <string>

namespace RTBKIT {

/******************************************************************************/
/* AUGMENTATION                                                               */
/******************************************************************************/

struct Augmentation
{
    Augmentation() {}

    Augmentation(
            const std::set<std::string>& tags,
            const Json::Value& data = Json::Value()) :
        tags(tags), data(data)
    {}

    Augmentation(const Json::Value& data) : tags(), data(data) {}

    std::set<std::string> tags;
    Json::Value data;

    void mergeWith(const Augmentation& other);

    Json::Value toJson() const;
    static Augmentation fromJson(const Json::Value& json);
};

typedef std::map<std::string, std::string> AgentAugmentations;


/******************************************************************************/
/* AUGMENTATION LIST                                                          */
/******************************************************************************/

// \todo may want to change this to unordered_map.
typedef std::map<AccountKey, Augmentation> AugmentationListBase;

struct AugmentationList : public AugmentationListBase
{

    void insertGlobal(const Augmentation& aug)
    {
        insert(std::make_pair(AccountKey(), aug));
    }

    void mergeWith(const AugmentationList& other);


    Augmentation filterForAccount(AccountKey account) const;


    std::vector<std::string> tagsForAccount(AccountKey account) const;


    Json::Value toJson() const;
    static AugmentationList fromJson(const Json::Value& json);
};


} // namespace RTBKIT

#endif // __rtb__augmentation_h__