Fix #179: support nested UIData in Ajax#updateRow()

Should cover a.o <p:columns>
omnifaces · Nov 5, 2015 · b6e10dd · b6e10dd
1 parent 034ede3
commit b6e10dd
Showing 1 changed file with 13 additions and 5 deletions.
diff --git a/src/main/java/org/omnifaces/util/Ajax.java b/src/main/java/org/omnifaces/util/Ajax.java
@@ -159,12 +159,20 @@ private static void updateRowCells(UIData table, int index) {
 		Collection<String> renderIds = getContext().getRenderIds();
 
 		for (UIComponent column : table.getChildren()) {
-			if (!(column instanceof UIColumn)) {
-				continue;
+			if (column instanceof UIColumn) {
+				for (UIComponent cell : column.getChildren()) {
+					renderIds.add(String.format("%s%c%d%c%s", tableId, separator, index, separator, cell.getId()));
+				}
 			}
-
-			for (UIComponent cell : column.getChildren()) {
-				renderIds.add(String.format("%s%c%d%c%s", tableId, separator, index, separator, cell.getId()));
+			else if (column instanceof UIData) { // <p:columns>.
+				String columnId = column.getId();
+				int columnCount = ((UIData) column).getRowCount();
+
+				for (UIComponent cell : column.getChildren()) {
+					for (int columnIndex = 0; columnIndex < columnCount; columnIndex++) {
+						renderIds.add(String.format("%s%c%d%c%s%c%d%c%s", tableId, separator, index, separator, columnId, separator, columnIndex, separator, cell.getId()));
+					}
+				}
 			}
 		}
 	}