From ce0feb3ef62c9438b0aedd601461cbb340faa021 Mon Sep 17 00:00:00 2001
From: Christian Mollekopf <chrigi_1@fastmail.fm>
Date: Fri, 10 Jun 2016 15:28:06 +0200
Subject: Allow preprocessors to inject entities.

Currently only working when creating an entity, the new entity is
always preprended in the store.
---
 common/pipeline.cpp | 45 ++++++++++++++++++++++++++++++++++-----------
 1 file changed, 34 insertions(+), 11 deletions(-)

(limited to 'common/pipeline.cpp')

diff --git a/common/pipeline.cpp b/common/pipeline.cpp
index 049d228..2c08aa0 100644
--- a/common/pipeline.cpp
+++ b/common/pipeline.cpp
@@ -181,16 +181,6 @@ KAsync::Job<qint64> Pipeline::newEntity(void const *command, size_t size)
         key = Sink::Storage::generateUid();
     }
     Q_ASSERT(!key.isEmpty());
-    const qint64 newRevision = Storage::maxRevision(d->transaction) + 1;
-
-    // Add metadata buffer
-    flatbuffers::FlatBufferBuilder metadataFbb;
-    auto metadataBuilder = MetadataBuilder(metadataFbb);
-    metadataBuilder.add_revision(newRevision);
-    metadataBuilder.add_operation(Operation_Creation);
-    metadataBuilder.add_replayToSource(replayToSource);
-    auto metadataBuffer = metadataBuilder.Finish();
-    FinishMetadataBuffer(metadataFbb, metadataBuffer);
 
     auto adaptorFactory = Sink::AdaptorFactoryRegistry::instance().getFactory(d->resourceType, bufferType);
     if (!adaptorFactory) {
@@ -201,8 +191,22 @@ KAsync::Job<qint64> Pipeline::newEntity(void const *command, size_t size)
     auto adaptor = adaptorFactory->createAdaptor(*entity);
     auto memoryAdaptor = QSharedPointer<Sink::ApplicationDomain::MemoryBufferAdaptor>::create(*(adaptor), adaptor->availableProperties());
     for (auto processor : d->processors[bufferType]) {
-        processor->newEntity(key, newRevision, *memoryAdaptor, d->transaction);
+        processor->resourceType = d->resourceType;
+        processor->pipeline = this;
+        processor->newEntity(key, Storage::maxRevision(d->transaction) + 1, *memoryAdaptor, d->transaction);
     }
+    //The maxRevision may have changed meanwhile if the entity created sub-entities
+    const qint64 newRevision = Storage::maxRevision(d->transaction) + 1;
+
+    // Add metadata buffer
+    flatbuffers::FlatBufferBuilder metadataFbb;
+    auto metadataBuilder = MetadataBuilder(metadataFbb);
+    metadataBuilder.add_revision(newRevision);
+    metadataBuilder.add_operation(Operation_Creation);
+    metadataBuilder.add_replayToSource(replayToSource);
+    auto metadataBuffer = metadataBuilder.Finish();
+    FinishMetadataBuffer(metadataFbb, metadataBuffer);
+
     flatbuffers::FlatBufferBuilder fbb;
     adaptorFactory->createBuffer(memoryAdaptor, fbb, metadataFbb.GetBufferPointer(), metadataFbb.GetSize());
 
@@ -461,6 +465,25 @@ void Preprocessor::finalize()
 {
 }
 
+void Preprocessor::createEntity(const Sink::ApplicationDomain::ApplicationDomainType &entity, const QByteArray &typeName)
+{
+    flatbuffers::FlatBufferBuilder entityFbb;
+    auto adaptorFactory = Sink::AdaptorFactoryRegistry::instance().getFactory(resourceType, typeName);
+    adaptorFactory->createBuffer(entity, entityFbb);
+    const auto entityBuffer = BufferUtils::extractBuffer(entityFbb);
+
+    flatbuffers::FlatBufferBuilder fbb;
+    auto entityId = fbb.CreateString(entity.identifier());
+    // This is the resource buffer type and not the domain type
+    auto type = fbb.CreateString(typeName);
+    auto delta = Sink::EntityBuffer::appendAsVector(fbb, entityBuffer.constData(), entityBuffer.size());
+    auto location = Sink::Commands::CreateCreateEntity(fbb, entityId, type, delta);
+    Sink::Commands::FinishCreateEntityBuffer(fbb, location);
+
+    const auto data = BufferUtils::extractBuffer(fbb);
+    pipeline->newEntity(data, data.size()).exec();
+}
+
 } // namespace Sink
 
 #pragma clang diagnostic push
-- 
cgit v1.2.3