1 files changed, 11 insertions, 3 deletions

diff --git a/subprojects/store-query-viatra/src/main/java/tools/refinery/store/query/viatra/internal/update/RelationViewUpdateListener.java b/subprojects/store-query-viatra/src/main/java/tools/refinery/store/query/viatra/internal/update/RelationViewUpdateListener.java
index e0d44e34..bf6b4197 100644
--- a/subprojects/store-query-viatra/src/main/java/tools/refinery/store/query/viatra/internal/update/RelationViewUpdateListener.java
+++ b/subprojects/store-query-viatra/src/main/java/tools/refinery/store/query/viatra/internal/update/RelationViewUpdateListener.java
@@ -5,6 +5,7 @@ import org.eclipse.viatra.query.runtime.matchers.context.IQueryRuntimeContextLis
 import org.eclipse.viatra.query.runtime.matchers.tuple.ITuple;
 import org.eclipse.viatra.query.runtime.matchers.tuple.Tuple;
 import tools.refinery.store.model.InterpretationListener;
+import tools.refinery.store.query.viatra.internal.ViatraModelQueryAdapterImpl;
 import tools.refinery.store.query.view.RelationView;
 import tools.refinery.store.query.view.TuplePreservingRelationView;
 
@@ -12,8 +13,13 @@ import java.util.ArrayList;
 import java.util.List;
 
 public abstract class RelationViewUpdateListener<T> implements InterpretationListener<T> {
+        private final ViatraModelQueryAdapterImpl adapter;
         private final List<RelationViewFilter> filters = new ArrayList<>();
 
+        protected RelationViewUpdateListener(ViatraModelQueryAdapterImpl adapter) {
+                this.adapter = adapter;
+        }
+
         public void addFilter(IInputKey inputKey, ITuple seed, IQueryRuntimeContextListener listener) {
                 filters.add(new RelationViewFilter(inputKey, seed, listener));
         }
@@ -23,6 +29,7 @@ public abstract class RelationViewUpdateListener<T> implements InterpretationLis
         }
 
         protected void processUpdate(Tuple tuple, boolean isInsertion) {
+                adapter.markAsPending();
                 int size = filters.size();
                 // Use a for loop instead of a for-each loop to avoid <code>Iterator</code> allocation overhead.
                 //noinspection ForLoopReplaceableByForEach
@@ -31,10 +38,11 @@ public abstract class RelationViewUpdateListener<T> implements InterpretationLis
                 }
         }
 
-        public static <T> RelationViewUpdateListener<T> of(RelationView<T> relationView) {
+        public static <T> RelationViewUpdateListener<T> of(ViatraModelQueryAdapterImpl adapter,
+                                                                                                           RelationView<T> relationView) {
                 if (relationView instanceof TuplePreservingRelationView<T> tuplePreservingRelationView) {
-                        return new TuplePreservingRelationViewUpdateListener<>(tuplePreservingRelationView);
+                        return new TuplePreservingRelationViewUpdateListener<>(adapter, tuplePreservingRelationView);
                 }
-                return new TupleChangingRelationViewUpdateListener<>(relationView);
+                return new TupleChangingRelationViewUpdateListener<>(adapter, relationView);
         }
 }