React to removals in reduced live-query.

1 files changed, 4 insertions, 0 deletions

diff --git a/common/datastorequery.cpp b/common/datastorequery.cpp
index 2a2f3f5..8bed015 100644
--- a/common/datastorequery.cpp
+++ b/common/datastorequery.cpp
@@ -224,6 +224,10 @@ public:
         bool foundValue = false;
         while(!foundValue && mSource->next([this, callback, &foundValue](const ResultSet::Result &result) {
                 auto reductionValue = result.entity.getProperty(mReductionProperty);
+                if (result.operation == Sink::Operation_Removal) {
+                    callback(result);
+                    return false;
+                }
                 if (!mReducedValues.contains(getByteArray(reductionValue))) {
                     //Only reduce every value once.
                     mReducedValues.insert(getByteArray(reductionValue));