FunctionGraphConsumer.hpp

#pragma once

#include "GraphConsumer.hpp"

namespace Showcase
{

  class SHOWCASE_API FunctionGraphConsumer : public GraphConsumer
  {
  public:
    typedef std::function<void(const NodeEvent&)> ChildFunc;
    typedef std::function<void(const PropertyEvent&)> PropertyFunc;
    typedef std::function<void(const PropertyEvent&, const EdgeId&)> EdgeFunc;
    typedef std::function<std::shared_ptr<folly::Executor>(void)> ExecutorFunc;

    FunctionGraphConsumer(Radiant::FlagsT<GraphEventSource> monitorFlags);
    virtual ~FunctionGraphConsumer();

    void setChildFunc(ChildFunc f);
    void setPropertyFunc(PropertyFunc f);
    void setEdgePropertyFunc(EdgeFunc f);
    void setExecutorFunc(ExecutorFunc f);

  protected:
    virtual void onChildEvent(const NodeEvent& ev) override;
    virtual void onPropertyEvent(const PropertyEvent& ev) override;
    virtual void onEdgePropertyEvent(const PropertyEvent& ev, const EdgeId& id) override;
    virtual std::shared_ptr<folly::Executor> getExecutor() override;

  private:
    ChildFunc m_childFunc;
    PropertyFunc m_propertyFunc;
    EdgeFunc m_edgeFunc;
    ExecutorFunc m_executorFunc;
  };

}