Question

我有一个存储过程，它生成并执行一段动态T-SQL，一旦构建，就像这样

SELECT 
    tblUsers.strUserName AS [Username]
    ,tblUsers.strEmail AS [Email]
    ,tblUserAuditLog.strIpAddress AS [IP Address]
    ,tblUserAuditLog.dtAuditTimeStamp AS [Timestamp]
    ,tblUserAuditLog.strAuditLogAction AS [Action]
    ,tblUserAuditLog.strLogDetails AS [Details]
FROM         
    tblUserAuditLog 
        LEFT OUTER JOIN tblUsers 
        ON tblUserAuditLog.intUserIdFK = tblUsers.intUserId
WHERE 
    tblUsers.strUserName = 'a12jun'
    AND tblUserAuditLog.dtAuditTimeStamp >= '2012-08-10'

此查询可以在开发环境中返回数千行，并且会在实时返回更多内容。

我想知道在实际返回结果之前动态查询返回了多少行，这样如果数量超过某个限制，我可以返回“缩小查询”错误消息。

我尝试过像这样生成另一段SQL：

DECLARE @sqlrowcount NVARCHAR(MAX);
SET @sqlrowcount = 'SELECT COUNT(*) FROM (' + @sql + ') AS TEMP';
EXEC(@sqlrowcount);

IF @@ROWCOUNT > @limit BEGIN .... END

其中@sql是动态查询。然后我尴尬地意识到EXEC(@sqlrowcount)将始终返回1，因为它返回一条值是记录数的记录。

是否有（相对）优雅的方式来做到这一点，例如不将结果写入临时表？

Answer 1

单向;

--base sql
declare @sql  nvarchar(255) = N'select * from master.dbo.spt_values'

--count wrapper
declare @sqlb nvarchar(255) = N'set @count=(select count(*) from (' + @sql + ') T)'

declare @count int
exec sp_executesql @sqlb, N'@count int output', @count output

select 'rows=',@count

您还可以使用TOP来强制执行限制，两次运行相同的语句效率不高。

Answer 2

您遇到的问题是exec语句保留了以前的@@rowcount值，在您的情况下，该值是set语句中的1（所有set语句使@@ rowcount变为1）。这是必要的，因为execute创建了自己的批次。

获取值的最佳方法是将sp_executesql与输出参数一起使用。这看起来像是：

declare @numRows int
declare @sql nvarchar(max)
set @sql = N'Select @numRows= count(*) from dbo.temp'

exec sp_executesql @sql, N'@numRows int output', @numRows output

--Put your if statement here using @numRows

这使用sp_executesql能够使输出参数从计数中返回值。

我推荐所有严肃的SQL程序的动态查询的更多详细信息的一个很好的来源是The Curse and Blessing of Dynamic SQL，它解释了如何对sp_executesql进行参数化以及为什么你可能想要与其他几个相关的主题一起使用。

Answer 3

Declare @Rowcount int
SELECT @RowCount = count(1) 
FROM         
dbo.tblUserAuditLog 
    LEFT OUTER JOIN dbo.tblUsers 
    ON dbo.tblUserAuditLog.intUserIdFK = dbo.tblUsers.intUserId


SELECT tblUsers.strUserName AS [Username]
    ,tblUsers.strEmail AS [Email]
    ,tblUserAuditLog.strIpAddress AS [IP Address]
    ,tblUserAuditLog.dtAuditTimeStamp AS [Timestamp]
    ,tblUserAuditLog.strAuditLogAction AS [Action]
    ,tblUserAuditLog.strLogDetails AS [Details]
    , @RowCount
FROM         
dbo.tblUserAuditLog 
    LEFT OUTER JOIN dbo.tblUsers 
    ON dbo.tblUserAuditLog.intUserIdFK = dbo.tblUsers.intUserId

提取从动态SQL返回的值

3 个答案: