VB.NET中基于条件为DataGridView动态添加按钮的实现问题

步骤1：预获取用户收藏的影片ID集合 #

先从数据库查询当前用户的收藏影片ID，存入HashSet以提升后续判断效率：

' 替换为实际的当前用户ID变量
Dim userId As Integer = 123
Dim watchlistMovieIds As New HashSet(Of Integer)()

Using conn As New OleDbConnection("Provider=Microsoft.ACE.OLEDB.12.0;Data Source=D:\moviesdb.accdb")
    conn.Open()
    Dim cmd As New OleDbCommand("SELECT movie_id FROM user_watchlist WHERE user_id = @userId", conn)
    cmd.Parameters.Add("@userId", OleDbType.Integer).Value = userId
    
    Dim reader As OleDbDataReader = cmd.ExecuteReader()
    While reader.Read()
        watchlistMovieIds.Add(CInt(reader("movie_id")))
    End While
End Using

步骤2：给查询结果添加状态标记列 #

填充DataTable后，新增布尔列标记影片是否已在收藏列表：

conn = New OleDbConnection("Provider=Microsoft.ACE.OLEDB.12.0;Data Source=D:\moviesdb.accdb")
conn.Open()

adp = New OleDbDataAdapter("Select * from movie_master where audience_age <= @audienceAge", conn)
adp.SelectCommand.Parameters.Add("@audienceAge", userAge)

Using ds As New DataSet
    adp.Fill(ds, 0)
    Dim dt As DataTable = ds.Tables(0)
    
    ' 添加标记列，用于判断收藏状态
    dt.Columns.Add("IsInWatchlist", GetType(Boolean))
    For Each row As DataRow In dt.Rows
        Dim movieId As Integer = CInt(row("movie_id")) ' 假设表中存在movie_id主键字段
        row("IsInWatchlist") = watchlistMovieIds.Contains(movieId)
    Next
    
    DataGridView1.DataSource = dt
End Using

步骤3：创建单个按钮列并绑定动态逻辑 #

仅添加一个按钮列，通过事件动态控制按钮文本和点击行为：

' 创建收藏操作按钮列
Dim watchlistBtnCol As New DataGridViewButtonColumn()
watchlistBtnCol.Name = "WatchlistOperation"
watchlistBtnCol.HeaderText = "收藏操作"
DataGridView1.Columns.Add(watchlistBtnCol)

' 绑定CellFormatting事件，根据状态设置按钮文本
AddHandler DataGridView1.CellFormatting, Sub(sender As Object, e As DataGridViewCellFormattingEventArgs)
    If e.ColumnIndex = DataGridView1.Columns("WatchlistOperation").Index AndAlso e.RowIndex >= 0 Then
        Dim isInWatchlist As Boolean = CBool(DataGridView1.Rows(e.RowIndex).Cells("IsInWatchlist").Value)
        e.Value = If(isInWatchlist, "移除", "添加")
    End If
End Sub

' 绑定CellClick事件，处理按钮点击的添加/移除逻辑
AddHandler DataGridView1.CellClick, Sub(sender As Object, e As DataGridViewCellEventArgs)
    If e.ColumnIndex = DataGridView1.Columns("WatchlistOperation").Index AndAlso e.RowIndex >= 0 Then
        Dim targetRow As DataGridViewRow = DataGridView1.Rows(e.RowIndex)
        Dim movieId As Integer = CInt(targetRow.Cells("movie_id").Value)
        Dim isInWatchlist As Boolean = CBool(targetRow.Cells("IsInWatchlist").Value)

        Using conn As New OleDbConnection("Provider=Microsoft.ACE.OLEDB.12.0;Data Source=D:\moviesdb.accdb")
            conn.Open()
            If isInWatchlist Then
                ' 执行移除收藏操作
                Dim delCmd As New OleDbCommand("DELETE FROM user_watchlist WHERE user_id = @userId AND movie_id = @movieId", conn)
                delCmd.Parameters.Add("@userId", OleDbType.Integer).Value = userId
                delCmd.Parameters.Add("@movieId", OleDbType.Integer).Value = movieId
                delCmd.ExecuteNonQuery()
                
                ' 更新行状态和按钮文本
                targetRow.Cells("IsInWatchlist").Value = False
                targetRow.Cells("WatchlistOperation").Value = "添加"
            Else
                ' 执行添加收藏操作
                Dim addCmd As New OleDbCommand("INSERT INTO user_watchlist (user_id, movie_id) VALUES (@userId, @movieId)", conn)
                addCmd.Parameters.Add("@userId", OleDbType.Integer).Value = userId
                addCmd.Parameters.Add("@movieId", OleDbType.Integer).Value = movieId
                addCmd.ExecuteNonQuery()
                
                ' 更新行状态和按钮文本
                targetRow.Cells("IsInWatchlist").Value = True
                targetRow.Cells("WatchlistOperation").Value = "移除"
            End If
        End Using
    End If
End Sub

核心说明 #

• 用HashSet存储收藏ID，利用其O(1)的查询效率，避免逐行查询数据库的性能损耗。
• 通过CellFormatting事件动态渲染按钮文本，确保每行按钮状态与实际收藏状态一致。
• 点击操作后直接更新当前行的标记列和按钮文本，无需重新绑定整个数据源，提升交互流畅度。

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