VB.NET实现TreeView数据复制到DataGridView（去重）技术咨询

Hey there! Let's break down how to copy employee data from a selected TreeView node (company or department) to DataGridView without duplicates, plus fix the issues in your existing code.

核心实现机制 #

To meet your requirement, we need to cover 4 key steps:

1. Target the selected node: Only process the node the user clicks on, not all root nodes.
2. Recursively traverse child nodes: Since your TreeView follows a Company → Department → Employee structure, recursion lets us dig through all levels under the selected node to find every employee.
3. Deduplication check: Use the unique employee ID (from your node naming convention empXXX) to avoid adding duplicate entries to DataGridView.
4. Safe row insertion: Dynamically add rows to DataGridView instead of hardcoding indices, which prevents index-out-of-bounds errors.

Issues in Your Existing Code #

Let's go over the problems in your current code:

• You're looping through all root nodes instead of the user-selected node, which doesn't match your "copy after selecting a department/company" requirement.
• A typo in the third loop: Dim Usubnode As TreeNode = subnode.Nodes(j) uses j (from the second loop) instead of z, causing index errors.
• You're assigning values directly to DGV.Rows(z)—if DataGridView has fewer rows than z, this will throw an exception.
• No deduplication logic at all, so repeated clicks will add duplicate employee entries.
• Hardcoded 3-level loops aren't flexible if your TreeView structure ever changes.

Fixed & Complete Implementation #

Here's the revised code that addresses all these issues, tailored to your TreeView structure:

1. Button Click Event (Main Trigger) #

This is the code that runs when your copy button is clicked:

Private Sub btnCopyToDGV_Click(sender As Object, e As EventArgs) Handles btnCopyToDGV.Click
    ' Make sure a node is selected first
    If TreeView2.SelectedNode Is Nothing Then
        MessageBox.Show("Please select a company or department node first!")
        Return
    End If

    ' Get all employee nodes under the selected node (recursively)
    Dim empNodes As List(Of TreeNode) = GetAllEmployeeNodes(TreeView2.SelectedNode)

    ' Add employees to DataGridView, skipping duplicates
    For Each empNode In empNodes
        ' Extract unique employee number from node name (e.g., "emp123" → "123")
        Dim empNum = empNode.Name.Replace("emp", "")
        ' Check if this employee is already in DataGridView
        If Not IsEmployeeInDGV(empNum) Then
            ' Dynamically add a new row and assign values
            Dim newRowIdx As Integer = DGV.Rows.Add()
            DGV.Rows(newRowIdx).Cells(0).Value = empNode.Name
            DGV.Rows(newRowIdx).Cells(1).Value = empNode.Text
        End If
    Next

    MessageBox.Show($"Copied {empNodes.Count} total employee records. {DGV.Rows.Count - empNodes.Count} duplicates skipped.")
End Sub

2. Recursive Method to Fetch All Employee Nodes #

This method digs through all child nodes under the selected node to collect every employee:

Private Function GetAllEmployeeNodes(parentNode As TreeNode) As List(Of TreeNode)
    Dim empNodes As New List(Of TreeNode)()

    For Each childNode In parentNode.Nodes
        ' Identify employee nodes by their "emp" prefix (matches your TreeView setup)
        If childNode.Name.StartsWith("emp") Then
            empNodes.Add(childNode)
        Else
            ' If it's a company/department node, recursively check its children
            empNodes.AddRange(GetAllEmployeeNodes(childNode))
        End If
    Next

    Return empNodes
End Function

3. Deduplication Check Method #

This checks if an employee already exists in DataGridView using their unique ID:

Private Function IsEmployeeInDGV(empNum As String) As Boolean
    For Each row As DataGridViewRow In DGV.Rows
        ' Skip the blank new row at the end of DataGridView
        If row.IsNewRow Then Continue For

        ' Extract existing employee number from the first cell
        Dim existingEmpNum = row.Cells(0).Value.ToString().Replace("emp", "")
        If existingEmpNum = empNum Then
            Return True ' Employee already exists
        End If
    Next

    Return False ' No duplicate found
End Function

Key Notes About the Code #

• Your original TreeView population code is solid! The unique node names (coXXX, depXXX, empXXX) are perfect for identifying node types and checking duplicates.
• Recursion makes the code flexible—even if you add more levels to your TreeView later, it will still work without changes.
• Dynamic row insertion ensures you never hit index-out-of-bounds errors, regardless of how many employees you're adding.

内容的提问来源于stack exchange，提问作者user3540795