SQL Server编写更新函数实现GROUP字段TEMP值分配使A/B类各达250条

需求逻辑确认 #

• 当前'A'类共150条，需补充100条达到250条
• 当前'B'类共50条，需补充200条达到250条
• 待分配的'TEMP'类共300条，刚好匹配补充缺口，C类保持不变

实现代码 #

1. 随机分配版本（无指定分配规则时使用） #

使用SQL Server的NEWID()实现随机排序分配：

WITH temp_group_rows AS (
    SELECT
        [GROUP],
        -- 为所有TEMP类数据生成排序序号
        ROW_NUMBER() OVER (ORDER BY NEWID()) AS row_num
    FROM 替换为实际表名
    WHERE [GROUP] = 'TEMP'
)
UPDATE temp_group_rows
SET [GROUP] = CASE
    WHEN row_num <= 100 THEN 'A'
    ELSE 'B'
END;

2. 按指定字段排序分配版本 #

如果需要按ID、DATE等字段的顺序分配（比如优先给更早的DATE分配'A'），修改ROW_NUMBER()的排序规则即可，示例按DATE升序分配：

WITH temp_group_rows AS (
    SELECT
        [GROUP],
        ROW_NUMBER() OVER (ORDER BY DATE ASC) AS row_num
    FROM 替换为实际表名
    WHERE [GROUP] = 'TEMP'
)
UPDATE temp_group_rows
SET [GROUP] = CASE
    WHEN row_num <= 100 THEN 'A'
    ELSE 'B'
END;

结果验证 #

执行更新后运行以下语句校验数据分布是否符合要求：

SELECT [GROUP], COUNT(*) AS record_count
FROM 替换为实际表名
GROUP BY [GROUP];

注意事项 #

• GROUP为SQL Server保留关键字，代码中使用[]包裹避免语法报错
• 执行更新操作前建议先备份TEMP类数据，避免误操作无法回滚
• 若使用SAS的SQL语法实现，逻辑一致，仅需调整对应函数的写法即可

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