合并CTE与Update查询并实现多AttributeId循环执行方案咨询

1. 合并CTE与UPDATE为单个查询 #

当然可以把CTE和UPDATE合并成一个查询，几乎所有主流关系型数据库（SQL Server、PostgreSQL、MySQL 8.0+等）都支持这种写法。核心思路是用CTE预先计算出需要更新的数据，然后在UPDATE语句中关联这个CTE来更新目标表。

举个SQL Server的示例（其他数据库语法类似，仅需微调）：

WITH AttributeUpdates AS (
    -- 这里是你的CTE逻辑，比如计算每个AttributeId对应的新值
    SELECT 
        AttributeId,
        -- 假设你需要基于某些计算得到新值
        SUM(SomeColumn) AS NewValue
    FROM SomeSourceTable
    GROUP BY AttributeId
    -- 可以先筛选要处理的AttributeId
    WHERE AttributeId IN (1, 2, 3)
)
UPDATE t
SET t.Value = au.NewValue
FROM YourTargetTable t
JOIN AttributeUpdates au ON t.AttributeId = au.AttributeId;

2. 处理多个AttributeId：替代IN子句的高效写法 #

如果IN子句没得到预期结果，大概率是因为关联逻辑或者数据匹配的问题。与其用显式的FOR EACH循环（数据库层面的循环通常效率很低），不如用集合式的批量更新，这才是数据库的最优处理方式。

如果你需要针对每个AttributeId执行特定逻辑，比如不同的更新规则，也可以在CTE里用条件分支（比如CASE语句）来处理：

WITH AttributeUpdates AS (
    SELECT 
        AttributeId,
        CASE 
            WHEN AttributeId = 1 THEN 'NewValueFor1'
            WHEN AttributeId = 2 THEN 'NewValueFor2'
            WHEN AttributeId = 3 THEN 'NewValueFor3'
            ELSE t.Value -- 保持原有值
        END AS NewValue
    FROM YourTargetTable t
    WHERE AttributeId IN (1, 2, 3)
)
UPDATE t
SET t.Value = au.NewValue
FROM YourTargetTable t
JOIN AttributeUpdates au ON t.AttributeId = au.AttributeId;

如果你的场景是需要对每个AttributeId执行独立的计算逻辑（比如关联不同的数据源），也可以用UNION ALL来构建CTE：

WITH AttributeUpdates AS (
    -- 处理AttributeId=1的逻辑
    SELECT 1 AS AttributeId, CalculatedValue1 AS NewValue FROM SourceTable1
    UNION ALL
    -- 处理AttributeId=2的逻辑
    SELECT 2 AS AttributeId, CalculatedValue2 AS NewValue FROM SourceTable2
    UNION ALL
    -- 处理AttributeId=3的逻辑
    SELECT 3 AS AttributeId, CalculatedValue3 AS NewValue FROM SourceTable3
)
UPDATE t
SET t.Value = au.NewValue
FROM YourTargetTable t
JOIN AttributeUpdates au ON t.AttributeId = au.AttributeId;

为什么不用显式循环？ #

数据库是为集合操作优化的，显式的FOR EACH循环（比如SQL Server的CURSOR、PostgreSQL的FOR LOOP）通常会比批量更新慢很多，尤其是数据量较大的时候。除非你的业务逻辑极端复杂，必须逐行处理，否则优先选择集合式更新。

如果你的IN子句问题是因为数据类型不匹配或者NULL值导致的，可以检查AttributeId的数据类型是否一致，或者用EXISTS替代IN来避免NULL的影响：

UPDATE t
SET t.Value = (SELECT NewValue FROM SourceTable s WHERE s.AttributeId = t.AttributeId)
FROM YourTargetTable t
WHERE EXISTS (
    SELECT 1 FROM SourceTable s WHERE s.AttributeId = t.AttributeId
    AND s.AttributeId IN (1,2,3)
);

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