ViewChangesCollector.collectRowChanged

#577
metasfresh · Sep 7, 2017 · 83ae06d · 83ae06d
1 parent fb7e2a7
commit 83ae06d
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 6 deletions.
diff --git a/src/main/java/de/metas/ui/web/view/event/ViewChanges.java b/src/main/java/de/metas/ui/web/view/event/ViewChanges.java
@@ -8,6 +8,7 @@
 import de.metas.ui.web.view.ViewId;
 import de.metas.ui.web.window.datatypes.DocumentId;
 import de.metas.ui.web.window.datatypes.DocumentIdsSelection;
+import lombok.NonNull;
 
 /*
  * #%L
@@ -86,30 +87,30 @@ public boolean isFullyChanged()
 	{
 		return fullyChanged;
 	}
-	
+
 	public boolean hasChanges()
 	{
-		if(fullyChanged)
+		if (fullyChanged)
 		{
 			return true;
 		}
-		
+
 		return changedRowIds != null && !changedRowIds.isEmpty();
 	}
 
 	public void addChangedRowIds(final DocumentIdsSelection rowIds)
 	{
 		// Don't collect rowIds if this was already flagged as fully changed.
-		if(fullyChanged)
+		if (fullyChanged)
 		{
 			return;
 		}
-		
+
 		if (rowIds == null || rowIds.isEmpty())
 		{
 			return;
 		}
-		
+
 		else if (rowIds.isAll())
 		{
 			fullyChanged = true;
@@ -125,6 +126,15 @@ else if (rowIds.isAll())
 		}
 	}
 
+	public void addChangedRowId(@NonNull final DocumentId rowId)
+	{
+		if (changedRowIds == null)
+		{
+			changedRowIds = new HashSet<>();
+		}
+		changedRowIds.add(rowId);
+	}
+
 	public DocumentIdsSelection getChangedRowIds()
 	{
 		final boolean fullyChanged = this.fullyChanged;

diff --git a/src/main/java/de/metas/ui/web/view/event/ViewChangesCollector.java b/src/main/java/de/metas/ui/web/view/event/ViewChangesCollector.java
@@ -21,6 +21,7 @@
 import de.metas.ui.web.view.ViewId;
 import de.metas.ui.web.websocket.WebSocketConfig;
 import de.metas.ui.web.websocket.WebsocketSender;
+import de.metas.ui.web.window.datatypes.DocumentId;
 import de.metas.ui.web.window.datatypes.DocumentIdsSelection;
 
 /*
@@ -178,6 +179,13 @@ public void collectRowsChanged(final IView view, final DocumentIdsSelection rowI
 
 		autoflushIfEnabled();
 	}
+
+	public void collectRowChanged(final IView view, final DocumentId rowId)
+	{
+		viewChanges(view).addChangedRowId(rowId);
+
+		autoflushIfEnabled();
+	}
 
 	private void collectFromChanges(final ViewChanges changes)
 	{