7

我有一个存储过程，它生成并执行一段动态 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，因为它返回一条记录，其值为记录数。

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

4

3 回答 3

6

单程;

--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来强制执行限制，两次运行相同的语句不是很有效。

于 2012-08-30T16:14:44.680 回答
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 程序推荐的关于动态查询的更多详细信息的一个很好的来源是动态 SQL 的诅咒和祝福，它解释了如何参数化 sp_executesql 以及为什么您可能想要与其他几个相关主题一起。

于 2012-08-30T16:40:46.500 回答
0 
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
于 2012-08-30T18:01:57.537 回答