基于注册日期为SQL表新列交替赋值A/B的实现方案

SQL脚本实现 #

直接利用窗口函数结合UPDATE语句，高效处理50万条数据：
先通过ROW_NUMBER()按注册日期升序给用户编号，最早注册的用户编号为1，再根据编号奇偶性赋值——奇数赋值'B'，偶数赋值'A'，完全匹配需求规则。

基于CTE的更新脚本（适用于SQL Server、PostgreSQL等支持CTE的数据库） #

WITH RankedUsers AS (
    SELECT 
        id, -- 替换为你的用户表主键字段
        new_column, -- 替换为新增的目标列名
        ROW_NUMBER() OVER(ORDER BY register_date ASC) AS rn
    FROM users -- 替换为你的用户表名
)
UPDATE RankedUsers
SET new_column = CASE 
    WHEN rn % 2 = 1 THEN 'B'
    ELSE 'A'
END;

子查询兼容方案（适配更多数据库） #

UPDATE users
SET new_column = CASE
    WHEN (SELECT COUNT(*) FROM users u2 WHERE u2.register_date <= users.register_date) % 2 = 1 THEN 'B'
    ELSE 'A'
END;

注：子查询方式性能略低于CTE方案，数据库支持窗口函数时优先选用前者。

Entity Framework 实现 #

考虑到数据量较大（50万条），避免全量加载到内存导致溢出，优先推荐以下两种方式：

方式1：执行原生SQL（EF Core/EF6通用） #

直接调用数据库执行上述高效SQL脚本：

// EF Core 写法
await _dbContext.Database.ExecuteSqlRawAsync(@"
    WITH RankedUsers AS (
        SELECT 
            id,
            new_column,
            ROW_NUMBER() OVER(ORDER BY register_date ASC) AS rn
        FROM users
    )
    UPDATE RankedUsers
    SET new_column = CASE 
        WHEN rn % 2 = 1 THEN 'B'
        ELSE 'A'
    END;
");

// EF6 写法
_dbContext.Database.ExecuteSqlCommand(@"
    WITH RankedUsers AS (
        SELECT 
            id,
            new_column,
            ROW_NUMBER() OVER(ORDER BY register_date ASC) AS rn
        FROM users
    )
    UPDATE RankedUsers
    SET new_column = CASE 
        WHEN rn % 2 = 1 THEN 'B'
        ELSE 'A'
    END;
");

方式2：EF Core 批量更新（需第三方包支持） #

若偏好EF语法风格，可安装Z.EntityFramework.Extensions.EFCore包实现批量更新，性能接近原生SQL：

await _dbContext.Users
    .OrderBy(u => u.RegisterDate)
    .Select((u, index) => new { User = u, Rank = index + 1 }) // 索引从0开始，+1后与SQL编号逻辑对齐
    .UpdateAsync(x => new User { NewColumn = x.Rank % 2 == 1 ? "B" : "A" });

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