1 files changed, 6 insertions, 1 deletions

diff --git a/common/pipeline.h b/common/pipeline.h
index 0461507..c6efaa6 100644
--- a/common/pipeline.h
+++ b/common/pipeline.h
@@ -35,6 +35,9 @@
 #include <resourcecontext.h>
 
 namespace Sink {
+namespace Storage {
+    class EntityStore;
+}
 
 class Preprocessor;
 
@@ -87,7 +90,7 @@ public:
     virtual void deletedEntity(const ApplicationDomain::ApplicationDomainType &oldEntity) {};
     virtual void finalizeBatch();
 
-    void setup(const QByteArray &resourceType, const QByteArray &resourceInstanceIdentifier, Pipeline *);
+    void setup(const QByteArray &resourceType, const QByteArray &resourceInstanceIdentifier, Pipeline *, Storage::EntityStore *entityStore);
 
 protected:
     template <typename DomainType>
@@ -99,6 +102,8 @@ protected:
 
     QByteArray resourceInstanceIdentifier() const;
 
+    Storage::EntityStore &entityStore() const;
+
 private:
     friend class Pipeline;
     class Private;