GraphConsumer.hpp

#pragma once

#include <Radiant/Flags.hpp>
#include <unordered_map>
#include <folly/Executor.h>

#include "Graph.hpp"
#include "Hashes.hpp"

namespace Showcase
{



  class SHOWCASE_API GraphConsumer : public std::enable_shared_from_this<GraphConsumer>
  {
  public:
    GraphConsumer(Radiant::FlagsT<GraphEventSource> monitorFlags);
    virtual ~GraphConsumer();

    void setMonitorFlags(Radiant::FlagsT<GraphEventSource> monitorFlags);
    Radiant::FlagsT<GraphEventSource> monitorFlags() const;

    void initStreams(GraphPtr graph, const GraphNode& node, int consume=0);

    GraphPtr sourceGraph() const;

    PropertyEventStreamPtr propertyStream();


  protected:
    virtual std::shared_ptr<folly::Executor> getExecutor() { return nullptr; }

    void initPropertyListener();
    void initChildListener();
    void initEdgePropertyListener(const EdgeId& id);

    virtual void onChildEvent(const NodeEvent& ev) = 0;
    virtual void onPropertyEvent(const PropertyEvent& ev) = 0;
    virtual void onEdgePropertyEvent(const PropertyEvent& ev, const EdgeId& id) = 0;

  private:
    void onChildEventInternal(const NodeEvent& ev, bool consume=false);

    NodeEventStreamPtr m_childrenStream;
    PropertyEventStreamPtr m_propertyStream;
    std::unordered_map<EdgeId, PropertyEventStreamPtr> m_edgePropertyStreams;

    std::mutex m_edgeStreamMutex;

    std::weak_ptr<Graph> m_graph;

    Radiant::FlagsT<GraphEventSource> m_eventsToMonitor;
  };

}