AntBlazor tree with checkboxes. How to check some tree items with code, when replacing the tree data - checkbox

Question

I'm searching for a way how to set the check boxes in an AntBlazor Tree component by code for example on button press. Not at startup. In the end I would like to replace the tree data with a filtered set of tree data, and keep the check boxes checked that were checked before. In my attempts the check boxes are cleared when a new DataSource is set, and I found no way to re-set them again. I'm using AntBlazor 0.12.0.1. which is the latest version at this time. I used a simple test implementation : Checkable="true", #bind-CheckedKeys="CheckedKeys" , DefaultCheckedKeys="DefaultCheckedKeys", DataSource="People" (where people is List< Person > and Person has Children of List< Person > ) and no way to get to the checkbox item in the code using : #ref="PeopleTree". I also found no example code on how to do this. Any suggestions on how to approach this would be welcome.
(2 days later)
Finally I found a way to get to the checked property in code : (all objects in this piece of code where found and not null )
public void ReplaceTree()
{
var x = CheckedKeys.ToList();
CheckedKeys = null;
People = CreateTree(); // Resets the tree data to loaded data, all checkboxes are cleared
CheckedKeys = x.ToArray();
resetCheckedItems(CheckedKeys);
StateHasChanged();
}
private void resetCheckedItems(string[] checkedKeys)
{
foreach (string key in checkedKeys)
{
int id = key.ToInt();
var treeNode = PeopleTree.FindFirstOrDefaultNode((node) => node.DataItem.Id == id, true);
treeNode.SetChecked(true);
}
}
But even this does not seem to set the checks visually on the screen. All check boxes stay unchecked.
The code is working and can be used to set the checkboxes, as long as the data source is not replaced. The replaced tree data is the same and all keys which are Id's are also the same, and even the treeNode is found when I use "FindFirstOrDefaultNode", but the checkbox is not set now, when the data source has been replaced.
(8 days later) I have found a work around. It involves keeping the dataSource. I have found duplicate entries in the tree, what makes that the nodes need to be removed further like this :
var treeNode = PeopleTree.FindFirstOrDefaultNode((node) => node.DataItem.Id == id, true);
if (treeNode != null)
{
if (!removedTreeNodes.Contains(treeNode)) removedTreeNodes.Add(treeNode);
treeNode.RemoveNode();
}
When I then add the node again, it is unique and also the set checked will work if it will find the node again in the data.
The problem is that it will not find the node in the same piece of code that will add the data. It first requires the code to complete, and a "StateHasChanged" to have ran in order for the component to update it's node structure based on the data.
In order to set them checked in the same piece of code I created a function that runs asynchronously while the main code continues.
The function waits 1ms so that the main code has finished, and the StateHasChanged has updated the Tree component. And then sets the checkbox to the state that was saved. :
public async Task SetChecked(TreeNode<Person> treenode)
{
await Task.Run(() => { Task.Delay(1).Wait(); });
var treeNode = PeopleTree.FindFirstOrDefaultNode((node) => node.DataItem.Id == treenode.DataItem.Id, true);
if (treeNode != null)
{
treeNode.SetChecked(treenode.Checked);
}
}