如何从运行从 C# 代码调用的存储过程中检索 @@SPID

Question

我正在尝试在可能长时间运行 select 查询或进程的页面上实现“取消”功能。 已经进行了索引和其他优化以最大限度地减少这种可能性，但在较高的服务器负载时间期间，仍然可能出现一些等待复杂进程的情况。

进程启动后，将显示一个带有取消按钮的模态忙碌微调器。 如果用户选择停止进程，我想使用以下步骤：

我有一个测试存储过程：

BEGIN
    DECLARE @SpID varchar(10) = CONVERT(varchar(10), @@SPID);
    RAISERROR(99999, 10, 1, @SpID) WITH NOWAIT;
    WAITFOR DELAY '00:00:15'
END

这个想法是为了完成以下......

1. 在被调用的存储过程中，从@@SPID获取指定的SPID
2. 使用RAISERROR语句创建包含SPID的非错误条件消息
3. 当WAITFOR延迟运行时，调用存储过程的应用程序 function 检索带有SPID的消息
4. 将SPID发送到另一个存储过程以终止该进程（ KILL 将负责关闭连接并回滚数据操作）

到目前为止，我的理解是来自此设置的消息可以通过调用 ASP.NET C# 代码获取，但我找不到一个示例来检索消息，同时仍然允许调用代码继续（使用ExecuteNonQuery 、 ExecuteReader或ExecuteScalar ).

/* CommentTest Procedure */
CREATE PROCEDURE CommentTest
AS
BEGIN
    DECLARE @SpID varchar(10) = CONVERT(varchar(10), @@SPID);
    RAISERROR(99999, 10, 1, @SpID) WITH NOWAIT;
    WAITFOR DELAY '00:00:30'
END

private string _spID;

private string RunCommentTestWaitForSP()
{
   _spID = "";

   try
   {
      using (SqlConnection conn = new SqlConnection(connStr))
      {
         using (SqlCommand cmd = new SqlCommand("CommentTest", conn))
         {
            cmd.CommandType = CommandType.StoredProcedure;

            if (conn.State == ConnectionState.Closed)
            {
               conn.Open();
            }

            // this is where I'm stuck...
            cmd.ExecuteNonQuery();
            _spID = "The message returned from RAISERROR...";
         }
      }
   }
   catch (Exception ex)
   {
      log.Error("RunCommentTestWaitForSP exception: " + ex.ToString() + " / " + ex.Message);
      log.Error("RunCommentTestWaitForSP stack trace: " + ex.StackTrace);
   }

   return spID;
}

private void LinkButtonCancelProcess_Click(object sender, EventArgs e)
{
    RunCommentTestWaitForSP();
    if (_spID.Length > 0 && "IsNumeric") {
        "call function to run sp_KillSpID"
    }
}

CREATE PROCEDURE [dbo].[sp_KillSpID]
    @SpID varchar(5)
AS
BEGIN
    SET NOCOUNT ON;

    DECLARE @KillStatement varchar(10) = 'KILL ' + @SpID;
    EXEC(@KillStatement);
END

Answer 1

您需要处理 SqlConnection object 上的InfoMessage事件：

private string RunCommentTestWaitForSP()
{
    try 
    {
        using (SqlConnection conn = new SqlConnection(connStr))
        using (SqlCommand cmd = new SqlCommand("CommentTest", conn))
        {
            cmd.CommandType = CommandType.StoredProcedure; 
            conn.InfoMessage += ReceiveSQLInfo;
            conn.Open();
            cmd.ExecuteNonQuery();
        }
    }
   catch (Exception ex)
   {
      log.Error($"RunCommentTestWaitForSP exception: {ex} / {ex.Message}");
      log.Error($"RunCommentTestWaitForSP stack trace: {ex.StackTrace}");
   }
}

private void ReceiveSQLInfo(object sender, SqlInfoMessageEventArgs e)
{
    //sender will be the "conn" connection above
    var spID = e.Message;
  
    // Here you could open a new connection 
    //   (we expect the existing connection is still busy)
    //    and use it to kill the spID
}

请记住ExecuteNonQuery() function 直到存储过程完成后才会完成，直到WAIT FOR DELAY代码也完成之后。 因此LinkButtonCancelProcess_Click()方法也不会执行您期望的操作，因为if()块直到存储过程完成后才到达。

此外，根据 C# RunCommentTestWaitForSP()的调用方式，计算机可能无法处理任何事件消息，直到它完成。