SQL Azure中客户端断开连接时如何终止存储过程执行？

针对你遇到的客户端断开后存储过程仍持续执行导致负载过高的问题，结合你的约束条件（无法修改前端、禁止创建作业、超大数据量查询、全球用户无操作约束），可以通过以下几种方式解决：

一、在存储过程中检测客户端连接状态 #

利用SQL Server的动态管理视图（DMV）sys.dm_exec_sessions，可以实时检查当前会话（@@SPID）是否仍处于活跃状态。将检查逻辑插入到存储过程的关键节点（比如每个大查询执行前、批量处理的循环内），一旦检测到客户端断开，立即终止存储过程。

示例代码：

-- 在存储过程的关键位置插入连接检查
IF NOT EXISTS (
    SELECT 1 
    FROM sys.dm_exec_sessions 
    WHERE session_id = @@SPID 
      AND is_user_process = 1  -- 确保是用户会话
)
BEGIN
    RAISERROR('客户端已断开连接，终止存储过程执行', 16, 1);
    RETURN;  -- 直接退出存储过程
END

对于包含多个大查询的存储过程，建议在每个查询前都添加该检查；如果是处理亿级数据，更推荐将查询拆分为批次处理，每处理一批就执行一次连接检查，避免无效执行。

二、设置会话级/服务器级的查询超时 #

1. 会话级超时控制 #

在存储过程开头添加SET QUERY_TIMEOUT语句，限制当前会话中查询的最长执行时间（单位：秒）。一旦超时，SQL Server会自动终止查询。

示例：

SET NOCOUNT ON;
SET QUERY_TIMEOUT 300;  -- 设置超时时间为5分钟（300秒）

-- 后续存储过程逻辑

注意：该设置仅对当前会话有效，不会影响其他用户的查询。

2. 服务器级成本阈值控制 #

通过设置QUERY_GOVERNOR_COST_LIMIT，限制所有查询的预估执行成本（SQL Server基于统计信息计算的资源消耗值）。当查询的预估成本超过阈值时，SQL Server会自动终止查询。

可以通过以下语句设置服务器级阈值：

sp_configure 'show advanced options', 1;
RECONFIGURE;
sp_configure 'query governor cost limit', 100;  -- 设置成本阈值为100（根据实际场景调整）
RECONFIGURE;

此设置会影响所有用户的查询，需要根据你的业务场景评估合适的阈值，避免误杀正常查询。

三、拆分大查询为批次处理 #

针对数百万/亿级数据的查询，将一次性全量查询拆分为多个小批次执行，既可以降低单批次的服务器负载，也能更频繁地检测客户端连接状态。

示例（以分页方式拆分查询）：

DECLARE @BatchSize INT = 100000;  -- 每次处理10万条数据
DECLARE @CurrentOffset INT = 0;

WHILE 1 = 1
BEGIN
    -- 先检查客户端连接
    IF NOT EXISTS (
        SELECT 1 
        FROM sys.dm_exec_sessions 
        WHERE session_id = @@SPID 
          AND is_user_process = 1
    )
    BEGIN
        RAISERROR('客户端断开，终止批量处理', 16, 1);
        RETURN;
    END

    -- 执行批次查询
    SELECT *
    FROM YourHistoryTable
    ORDER BY PrimaryKeyColumn  -- 必须有排序字段保证分页正确性
    OFFSET @CurrentOffset ROWS FETCH NEXT @BatchSize ROWS ONLY;

    -- 若当前批次返回数据不足BatchSize，说明已处理完所有数据
    IF @@ROWCOUNT < @BatchSize
        BREAK;

    SET @CurrentOffset += @BatchSize;
END

四、补充说明 #

• 上述方案均无需修改前端代码，也不需要创建数据库作业，完全符合你的约束条件；
• sys.dm_exec_sessions的查询权限需要确保存储过程执行者有足够权限，若权限不足，可以通过创建带签名的存储过程或者赋予VIEW SERVER STATE权限解决；
• 服务器级的超时/成本阈值设置需要谨慎评估，建议先在测试环境验证后再应用到生产环境。

内容的提问来源于stack exchange，提问作者Hugo Guadalupe Sánchez Ruiz