如何在Asp.Net站点启动时自动执行update-database并完成角色同步？

针对你的需求，我们可以放弃依赖Database.SetInitializer（它的设计目标是数据库初始化，而非持续的角色/用户同步），转而在应用启动时执行自定义初始化逻辑，结合EF自动迁移实现部署后的自动更新。下面是具体实现步骤：

1. 确保启动时自动执行数据库迁移 #

首先让应用在启动时自动应用所有未执行的迁移，替代手动运行Update-Database的操作：

.NET Core/.NET 5+（Program.cs） #

using Microsoft.EntityFrameworkCore;

var app = builder.Build();

// 自动应用所有未完成的迁移
using (var scope = app.Services.CreateScope())
{
    var dbContext = scope.ServiceProvider.GetRequiredService<ApplicationDbContext>();
    dbContext.Database.Migrate(); // 自动执行所有未应用的迁移脚本
}

// 后续中间件配置...

.NET Framework（Global.asax.cs） #

using System.Data.Entity;

protected void Application_Start()
{
    // 其他初始化逻辑...
    Database.SetInitializer(new MigrateDatabaseToLatestVersion<ApplicationDbContext, YourMigrationConfiguration>());
    using (var context = new ApplicationDbContext())
    {
        context.Database.Initialize(false);
    }
}

2. 创建自定义身份初始化类 #

我们创建一个专门的类处理角色同步和用户角色批量更新，依赖ApplicationDbContext、RoleManager和UserManager完成操作：

using Microsoft.AspNetCore.Identity;
using System.Reflection;

public class IdentityInitializer
{
    private readonly ApplicationDbContext _context;
    private readonly RoleManager<IdentityRole> _roleManager;
    private readonly UserManager<ApplicationUser> _userManager;

    // 构造函数注入依赖
    public IdentityInitializer(ApplicationDbContext context, RoleManager<IdentityRole> roleManager, UserManager<ApplicationUser> userManager)
    {
        _context = context;
        _roleManager = roleManager;
        _userManager = userManager;
    }

    public async Task InitializeAsync()
    {
        // 第一步：同步RolesName中的所有角色到数据库
        await SyncRolesAsync();

        // 第二步：为指定原有角色的用户添加新角色
        await SyncUserRolesAsync();
    }

    private async Task SyncRolesAsync()
    {
        Type roleType = typeof(RolesName);
        foreach (var field in roleType.GetFields(BindingFlags.Public | BindingFlags.Static))
        {
            var roleName = field.GetValue(null)?.ToString();
            if (string.IsNullOrEmpty(roleName)) continue;

            if (!await _roleManager.RoleExistsAsync(roleName))
            {
                await _roleManager.CreateAsync(new IdentityRole(roleName));
                // 可选：添加日志记录，例如"已创建角色：{roleName}"
            }
        }
    }

    private async Task SyncUserRolesAsync()
    {
        // 定义规则：原有角色 -> 需要添加的新角色，可根据需求灵活调整
        var roleMappings = new Dictionary<string, string>
        {
            { RolesName.Admin, RolesName.SuperAdmin },
            { RolesName.Editor, RolesName.ContentManager }
            // 新增角色时，直接添加新的映射关系即可
        };

        foreach (var mapping in roleMappings)
        {
            var originalRoleName = mapping.Key;
            var newRoleName = mapping.Value;

            // 验证角色是否存在，避免无效操作
            if (!await _roleManager.RoleExistsAsync(originalRoleName) || !await _roleManager.RoleExistsAsync(newRoleName))
                continue;

            // 查询所有拥有原有角色，但未拥有新角色的用户
            var usersToUpdate = _context.Users
                .Where(u => u.Roles.Any(r => r.Role.Name == originalRoleName)
                         && !u.Roles.Any(r => r.Role.Name == newRoleName))
                .ToList();

            // 批量为用户添加新角色
            foreach (var user in usersToUpdate)
            {
                await _userManager.AddToRoleAsync(user, newRoleName);
                // 可选：添加日志记录，例如"已为用户{user.UserName}添加角色{newRoleName}"
            }
        }
    }
}

3. 在应用启动时调用初始化逻辑 #

最后在应用启动时触发初始化逻辑，确保Octopus部署后站点启动时自动执行：

.NET Core/.NET 5+（Program.cs） #

// 在迁移执行完毕后调用初始化逻辑
using (var scope = app.Services.CreateScope())
{
    var initializer = scope.ServiceProvider.GetRequiredService<IdentityInitializer>();
    await initializer.InitializeAsync();
}

// 启动应用
app.Run();

.NET Framework（Global.asax.cs） #

protected async void Application_Start()
{
    // 其他初始化逻辑...
    var roleManager = new RoleManager<IdentityRole>(new RoleStore<IdentityRole>(new ApplicationDbContext()));
    var userManager = new UserManager<ApplicationUser>(new UserStore<ApplicationUser>(new ApplicationDbContext()));
    var initializer = new IdentityInitializer(new ApplicationDbContext(), roleManager, userManager);
    await initializer.InitializeAsync();
}

关键说明 #

• 幂等性：所有操作都是幂等的——角色仅在不存在时创建，用户角色仅在未拥有时添加，重复执行不会导致重复数据或错误。
• 部署适配：Octopus部署后，站点启动时会自动执行迁移和初始化逻辑，无需手动干预。
• 灵活性：SyncUserRolesAsync中的角色映射可随时调整，新增角色时只需添加新的映射规则即可。

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