SQL SERVER - 在 WHERE 子句中设置变量

Question

我正在执行以下更新：

UPDATE MyTable
SET    Field1 = 0       
FROM   MyTable INNER JOIN AnotherTable
       ON MyTable.ID = AnotherTable.ID
WHERE  0=MyFunc(MyTable.ID)

MyFunc 是一个返回 int 值的函数。

但现在我正在尝试使用 MyFunc 返回的值来更新表 MyTable 上的另一个字段。

UPDATE MyTable
SET    Field1 = 0,
       Field2 = CASE WHEN MyFunc(MyTable.ID) = 0 THEN 'OK' ELSE 'NOK' END
FROM   MyTable INNER JOIN AnotherTable
       ON MyTable.ID = AnotherTable.ID
WHERE  0=MyFunc(MyTable.ID)

这里的问题是我调用了两次函数 MyFunc，一个在 SET 中，一个在 WHERE 子句中，然后我担心性能。

我想知道是否还有其他方法可以做到这一点。我考虑过使用变量，我的意思是，在 WHERE 子句中设置这个变量，然后在更新的集合中使用它，但我想这是不可能的，对吧？

那么我该如何做才能获得更好的数据库性能呢？

Answer 1

您可以使用CROSS APPLY 来模拟变量。以下SELECT 将能够使用变量名 FuncResult 处理函数的结果。

SELECT MyTable.Field1 
      ,MyTable.Field2
      ,x.FuncResult
FROM  MyTable 
CROSS APPLY(SELECT dbo.MyFunc(MyTable.ID) AS FuncResult) AS x
INNER JOIN AnotherTable
       ON MyTable.ID = AnotherTable.ID
WHERE FuncResult=0

此外，您还可以查看可更新的 CTE，如下所示：

WITH UpdateableCTE AS
(
    SELECT MyTable.Field1 
          ,MyTable.Field2
          ,x.FuncResult
    FROM  MyTable 
    CROSS APPLY(SELECT dbo.MyFunc(MyTable.ID) AS FuncResult) AS x
    INNER JOIN AnotherTable
           ON MyTable.ID = AnotherTable.ID
    WHERE FuncResult=0
)
UPDATE UpdateableCTE
SET Field1=0
   ,Field2=CASE WHEN FuncResult = 0 /*senseless, as your where clause filters to only this value!*/ THEN 'OK' ELSE 'NOK' END

但是 - 老实说 - 整个问题看起来像开始在二楼盖房子...我很确定，引擎足够聪明，可以计算函数的结果并使用相同的输入重复使用它（至少每行）。

如果你真的很在意性能避免使用标量函数。如果您需要一个函数，内联表值函数，它可能只返回一行中的列，速度要快得多。

更新

这里是 INLINED tabe value function 的示例（单语句，ad-hoc，没有 BEGIN...END）：

CREATE FUNCTION dbo.TestInlineTVF(@input INT)
RETURNS TABLE
AS
RETURN SELECT @input*2 AS FuncResult;
GO

DECLARE @tbl TABLE(ID INT IDENTITY,SomeValue VARCHAR(100));
INSERT INTO @tbl VALUES('Row 1'),('Row 2');

SELECT *
FROM @tbl AS t
CROSS APPLY dbo.TestInlineTVF(t.ID) AS x
GO
DROP FUNCTION dbo.TestInlineTVF;