在C#中如何取消执行中的PostgreSQL查询并关闭数据库连接？

核心思路 #

要安全中断查询并关闭连接，关键是在查询执行过程中支持异步取消，而非直接强制关闭连接。Npgsql原生支持CancellationToken，结合异步API可实现优雅中断。

具体实现步骤 #

• 引入CancellationTokenSource：生成取消令牌，点击停止按钮时触发取消信号。
• 改用异步查询API：将同步的ExecuteReader替换为异步的ExecuteReaderAsync，并传入取消令牌。
• 循环中检查取消状态：每次执行查询前检查令牌是否已取消，及时终止循环。
• 安全处理连接关闭：取消后等待当前查询中断完成，再关闭连接。

代码示例 #

页面后端逻辑（简化版） #

// 类级别变量，用于管理取消信号和数据库连接
private CancellationTokenSource _cts;
private NpgsqlConnection _connection;

// 开始按钮点击事件
public async Task StartQueries()
{
    _cts = new CancellationTokenSource();
    var queries = GetQueries(); // 你的查询集合

    try
    {
        _connection = new NpgsqlConnection("你的PostgreSQL连接字符串");
        await _connection.OpenAsync(_cts.Token);

        using var cmd = new NpgsqlCommand();
        cmd.Connection = _connection;

        for (int i = 0; i < 100; i++)
        {
            // 检查取消信号，提前终止循环
            if (_cts.Token.IsCancellationRequested)
                break;

            cmd.CommandText = queries[i];
            // 异步执行查询并绑定取消令牌
            using var reader = await cmd.ExecuteReaderAsync(_cts.Token);
            // 按需处理查询结果
            await reader.ReadAsync(_cts.Token);
        }
    }
    catch (OperationCanceledException)
    {
        // 捕获取消异常，属于预期中断，无需额外处理
    }
    finally
    {
        // 确保连接被安全关闭并释放资源
        if (_connection?.State == ConnectionState.Open)
            await _connection.CloseAsync();
        
        _cts?.Dispose();
        _connection?.Dispose();
    }
}

// 停止按钮点击事件
public void StopQueries()
{
    _cts?.Cancel();
}

关键细节说明 #

• CancellationToken的作用：调用Cancel()时，Npgsql会向PostgreSQL发送查询取消请求，同时终止当前异步操作，还能让循环提前退出，避免执行后续查询。
• 异步API的必要性：同步ExecuteReader无法中途中断，必须等待查询完成；异步API结合取消令牌可实现即时中断。
• 资源清理：finally块确保连接和取消令牌源被正确释放，避免资源泄漏。

关于PID取消失败的原因 #

你之前尝试用PID取消查询失败，大概率是因为：

• 查询执行速度快，当获取PID并发送取消请求时，目标查询已执行完成。
• 未正确捕获当前正在执行的查询PID（循环中PID会随查询切换，可能捕获到已完成的查询PID）。
  相比之下，CancellationToken绑定到当前命令，能精准中断正在执行的查询，响应更及时。

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