fix: columnlist remove

giard-alexandre · Sep 28, 2024 · 44a7afe · 44a7afe
1 parent f87cbe6
commit 44a7afe
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/src/DynamicTreeDataGrid/Models/Columns/DynamicColumnListBase.cs b/src/DynamicTreeDataGrid/Models/Columns/DynamicColumnListBase.cs
@@ -50,6 +50,17 @@ protected override void SetItem(int index, IDynamicColumn<TModel> item) {
 		base.SetItem(index, item);
 	}
 
+	protected override void ClearItems() {
+		base.ClearItems();
+		_nameSet.Clear();
+	}
+
+	protected override void RemoveItem(int index) {
+		var removedName = this[index].Name;
+		_nameSet.Remove(removedName);
+		base.RemoveItem(index);
+	}
+
 	public Size CellMeasured(int columnIndex, int rowIndex, Size size) {
 		var column = (IUpdateColumnLayout)this[columnIndex];
 		return new Size(column.CellMeasured(size.Width, rowIndex), size.Height);