我在 SQL Server 中有一个长时间运行的存储过程，我的用户需要能够取消它。我编写了一个小型测试应用程序，如下所示，它演示了 SqlCommand.Cancel() 方法工作得很好:

    private SqlCommand cmd;
    private void TestSqlServerCancelSprocExecution()
    {
        TaskFactory f = new TaskFactory();
        f.StartNew(() =>
            {
              using (SqlConnection conn = new SqlConnection("connStr"))
              {
                conn.InfoMessage += conn_InfoMessage;
                conn.FireInfoMessageEventOnUserErrors = true;
                conn.Open();

                cmd = conn.CreateCommand();
                cmd.CommandType = CommandType.StoredProcedure;
                cmd.CommandText = "dbo.[CancelSprocTest]";
                cmd.ExecuteNonQuery();
              }
           });
    }

    private void cancelButton_Click(object sender, EventArgs e)
    {
        if (cmd != null)
        {
            cmd.Cancel();
        }
    }

调用 cmd.Cancel() 后，我可以验证底层存储过程基本上立即停止执行。鉴于我在我的应用程序中大量使用异步/等待模式，我希望 SqlCommand 上采用 CancellationToken 参数的异步方法同样有效。不幸的是，我发现在 CancellationToken 上调用 Cancel() 导致不再调用 InfoMessage 事件处理程序，但底层存储过程继续执行。我的异步版本测试代码如下:

    private SqlCommand cmd;
    private CancellationTokenSource cts;
    private async void TestSqlServerCancelSprocExecution()
    {
        cts = new CancellationTokenSource();
        using (SqlConnection conn = new SqlConnection("connStr"))
        {
            conn.InfoMessage += conn_InfoMessage;
            conn.FireInfoMessageEventOnUserErrors = true;
            conn.Open();

            cmd = conn.CreateCommand();
            cmd.CommandType = CommandType.StoredProcedure;
            cmd.CommandText = "dbo.[CancelSprocTest]";
            await cmd.ExecuteNonQueryAsync(cts.Token);
        }
    }

    private void cancelButton_Click(object sender, EventArgs e)
    {
        cts.Cancel();
    }

我是否遗漏了 CancellationToken 的工作原理？我使用 .NET 4.5.1 和 SQL Server 2012 以防万一。

编辑:我将测试应用程序重写为控制台应用程序，以防同步上下文是一个因素，并且我看到了相同的行为——CancellationTokenSource.Cancel() 的调用不会停止执行底层存储过程。

编辑:这是我调用的存储过程的主体，以防万一。它以一秒为间隔插入记录并打印结果，以便于查看取消尝试是否及时生效。

WHILE (@loop <= 40)
BEGIN

  DECLARE @msg AS VARCHAR(80) = 'Iteration ' + CONVERT(VARCHAR(15), @loop);
  RAISERROR (@msg,0,1) WITH NOWAIT;
  INSERT INTO foo VALUES (@loop);
  WAITFOR DELAY '00:00:01.01';

  SET @loop = @loop+1;
END;

最佳答案

在查看您的存储过程正在执行的操作后，它似乎以某种方式阻止了取消。

如果你改变

RAISERROR (@msg,0,1) WITH NOWAIT;

删除 WITH NOWAIT 子句，然后取消按预期工作。但是，这会阻止 InfoMessage 事件实时触发。

您可以通过其他方式跟踪长时间运行的存储过程的进度，或者注册 token 取消并调用 cmd.Cancel()，因为您知道这行得通。

还有一点要注意，在 .NET 4.5 中，您可以只使用 Task.Run 而不是实例化 TaskFactory。

所以这是一个可行的解决方案:

private CancellationTokenSource cts;
private async void TestSqlServerCancelSprocExecution()
{
    cts = new CancellationTokenSource();
    try
    {
        await Task.Run(() =>
        {
            using (SqlConnection conn = new SqlConnection("connStr"))
            {
                conn.InfoMessage += conn_InfoMessage;
                conn.FireInfoMessageEventOnUserErrors = true;
                conn.Open();

                var cmd = conn.CreateCommand();
                cts.Token.Register(() => cmd.Cancel());
                cmd.CommandType = CommandType.StoredProcedure;
                cmd.CommandText = "dbo.[CancelSprocTest]";
                cmd.ExecuteNonQuery();
            }
       });
    }
    catch (SqlException)
    {
        // sproc was cancelled
    }
}

private void cancelButton_Click(object sender, EventArgs e)
{
    cts.Cancel();
}

在我对此的测试中，我必须将 ExecuteNonQuery 包装在 Task 中，以便 cmd.Cancel() 工作。如果我使用 ExecuteNonQueryAsync，即使没有向它传递 token ，系统也会在 cmd.Cancel() 上阻塞。我不确定为什么会这样，但是将同步方法包装在 Task 中提供了类似的用法。

关于c# - 使用 CancellationToken 取消 SQL Server 查询，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/24738417/