Fix table column consistency

Refs: #6890
public-ui · Nov 8, 2024 · 95947bc · 95947bc
1 parent c4d1706
commit 95947bc
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/packages/components/src/components/table-stateless/component.tsx b/packages/components/src/components/table-stateless/component.tsx
@@ -398,6 +398,23 @@ export class KolTableStateless implements TableStatelessAPI {
 				dataField[0].push(emptyCell);
 			}
 		}
+		dataField.forEach((row, rowIndex) => {
+			if (row.length > maxCols) {
+				row = row.slice(0, maxCols);
+			}
+
+			if (row.length < maxCols) {
+				const missingColumns = maxCols - row.length;
+				for (let i = 0; i < missingColumns; i++) {
+					row.push({
+						label: '',
+						colSpan: undefined,
+						rowSpan: undefined,
+					});
+				}
+			}
+			dataField[rowIndex] = row;
+		});
 		return dataField;
 	}