event.h

#ifndef PROIO_EVENT_H
#define PROIO_EVENT_H

#include <string>

#include "google/protobuf/descriptor.h"
#include "google/protobuf/dynamic_message.h"

#include "proio/proto/proio.pb.h"

namespace proio {
class Event {
   public:
    Event();
    Event(const std::string &proto);
    virtual ~Event();

    uint64_t AddEntry(google::protobuf::Message *entry, std::string tag = "");
    google::protobuf::Message *GetEntry(uint64_t id);
    void TagEntry(uint64_t id, std::string tag);
    void UntagEntry(uint64_t id, std::string tag);
    void RemoveEntry(uint64_t id);
    std::vector<std::string> Tags();
    std::vector<uint64_t> TaggedEntries(std::string tag);
    std::vector<uint64_t> AllEntries();
    std::vector<std::string> EntryTags(uint64_t id);
    void DeleteTag(std::string tag);
    google::protobuf::Message *Free(const google::protobuf::Descriptor *desc);
    std::map<std::string, std::shared_ptr<const std::string>> &Metadata() { return metadata; }

    std::string String();
    bool SerializeToString(std::string *output) {
        FlushCache();
        return getProto()->SerializeToString(output);
    }
    void FlushCache();
    void Clear();
    void SetDescriptorPool(const google::protobuf::DescriptorPool *pool = NULL);
    void UseGeneratedPool(bool useGenPool = true);

    Event &operator=(const Event &event);

   private:
    uint64_t getTypeID(google::protobuf::Message *entry);
    const google::protobuf::Descriptor *getDescriptor(uint64_t typeID);
    void tagCleanup();
    void clearDescriptors();

    friend class Writer;
    friend class Reader;

    proto::Event *getProto();

    proto::Event *eventProto;
    std::map<std::string, uint64_t> revTypeLookup;
    std::map<uint64_t, google::protobuf::Message *> entryCache;
    std::map<uint64_t, const google::protobuf::Descriptor *> descriptorCache;
    std::map<std::string, std::shared_ptr<const std::string>> metadata;
    const google::protobuf::DescriptorPool *descriptorPool;
    std::unique_ptr<google::protobuf::DynamicMessageFactory> messageFactory;
    bool useGenPool;
    bool dirtyTags;

    std::map<const google::protobuf::Descriptor *, std::vector<google::protobuf::Message *>> store;
};

const class UnknownMessageTypeError : public std::exception {
    virtual const char *what() const throw() { return "Unknown message type"; }
} unknownMessageTypeError;
}  // namespace proio

#endif  // PROIO_EVENT_H