使用@@ROWCOUNT的旧SQL函数在C#调用时偶发空值问题排查

问题背景 #

我们有一个SQL函数[dbo].[GetCurrentSessionId]，用于从RestoreSession表获取最新符合SessionState≠3且≠7条件的SessionId，当无匹配记录时返回空字符串。但偶发出现异常：数据库存在符合条件的会话时，C#应用（使用System.Data.SqlClient）调用函数始终返回空字符串；但在SSMS中执行该函数却能正常返回结果。移除函数中的@@ROWCOUNT判断后，C#调用恢复正常，重新添加判断则问题复现。

原因分析 #

@@ROWCOUNT是SQL Server的会话级变量，用于记录上一条SQL语句影响的行数。它的失效主要受以下因素影响：

• 执行计划差异：SSMS与C#应用的默认SET选项不同（比如SSMS默认SET ARITHABORT ON，而SqlClient默认OFF），这会导致查询优化器生成不同的执行计划。在某些计划下，SELECT TOP 1 @v = ...语句的@@ROWCOUNT可能无法被正确统计。
• 会话上下文干扰：@@ROWCOUNT是会话级的，如果函数执行过程中存在隐式系统操作，或是连接池复用导致的上下文残留，都可能意外覆盖@@ROWCOUNT的值。
• 标量函数内联（SQL Server 2019+）：如果数据库兼容性级别为150及以上，标量函数可能被自动内联，此时函数内部的@@ROWCOUNT会受外部查询上下文影响，导致判断逻辑失效。

解决方案 #

方案1：替换@@ROWCOUNT判断（推荐） #

直接检查变量@v是否为NULL（无匹配记录时@v会保持初始的NULL状态），替代依赖@@ROWCOUNT的逻辑，这是最可靠的方式：

ALTER FUNCTION [dbo].[GetCurrentSessionId]()
RETURNS nvarchar(255)
AS
BEGIN
    declare @v nvarchar(255)

    select top 1 @v = SessionId
    from RestoreSession
    where 
        SessionState <> 3 and -- Complete
        SessionState <> 7 -- CompletedWithWarnings
    order by SessionId desc

    return isnull(@v, '')
END

方案2：立即捕获@@ROWCOUNT到变量 #

如果必须保留原逻辑，可在SELECT语句后立即将@@ROWCOUNT存入局部变量，避免后续操作覆盖其值：

ALTER FUNCTION [dbo].[GetCurrentSessionId]()
RETURNS nvarchar(255)
AS
BEGIN
    declare @v nvarchar(255)
    declare @rowCount int

    select top 1 @v = SessionId
    from RestoreSession
    where 
        SessionState <> 3 and -- Complete
        SessionState <> 7 -- CompletedWithWarnings
    order by SessionId desc

    set @rowCount = @@ROWCOUNT -- 立即捕获行数

    if @rowCount = 0
        set @v = ''

    return @v
END

方案3：统一连接SET选项 #

在C#应用中打开连接后，设置与SSMS一致的SET选项（比如SET ARITHABORT ON），确保执行计划一致：

public string GetCurrentSessionID()
{
    string res = string.Empty;
    using (var connection = _persistence.CreateSqlConnection())
    {
        connection.Open();
        // 设置与SSMS一致的SET选项
        using (var setCmd = connection.CreateCommand())
        {
            setCmd.CommandText = "SET ARITHABORT ON";
            setCmd.ExecuteNonQuery();
        }

        var sessionsQuery = "select [dbo].GetCurrentSessionId() as SID";
        using (var sqlCommand = connection.CreateCommand(sessionsQuery))
        {
            using (var sessionReader = sqlCommand.ExecuteReader())
            {
                if (sessionReader.Read())
                {
                    if (!sessionReader.IsDBNull("SID"))
                    {
                        res = sessionReader.GetString("SID");
                    }
                }
            }
        }
    }
    return res;
}

验证方法 #

修改函数后，在C#应用中多次调用测试，确认是否还会出现偶发返回空的情况；同时在SSMS中执行函数，确保原有逻辑不受影响。

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