.NET 7 Blazor中如何获取RoleManager/IServiceProvider并初始化角色？

核心思路 #

在.NET 6+的顶级语句模式下，我们可以在构建完WebApplication实例后，通过**创建服务范围（Service Scope）**来获取所需的服务（如RoleManager<IdentityRole>），进而执行角色初始化逻辑。

具体实现方式 #

方式一：直接在Program.cs中编写初始化逻辑 #

1. 首先定义角色初始化的异步方法：

public static async Task CreateRolesAsync(IServiceProvider serviceProvider)
{
    var roleManager = serviceProvider.GetRequiredService<RoleManager<IdentityRole>>();

    // 定义需要初始化的角色列表
    string[] roleNames = { "Admin", "Editor", "Viewer" };
    
    foreach (var roleName in roleNames)
    {
        // 检查角色是否已存在
        var roleExists = await roleManager.RoleExistsAsync(roleName);
        if (!roleExists)
        {
            // 创建新角色
            await roleManager.CreateAsync(new IdentityRole(roleName));
        }
    }
}

2. 在Program.cs的顶级语句中调用该方法：

var builder = WebApplication.CreateBuilder(args);

// 必须先配置Identity服务，并添加角色支持
builder.Services.AddDefaultIdentity<IdentityUser>(options => options.SignIn.RequireConfirmedAccount = true)
    .AddRoles<IdentityRole>() // 启用角色功能
    .AddEntityFrameworkStores<ApplicationDbContext>();

// 其他服务配置（如数据库、中间件依赖等）...

var app = builder.Build();

// 创建服务范围，执行角色初始化
using (var scope = app.Services.CreateScope())
{
    var serviceProvider = scope.ServiceProvider;
    await CreateRolesAsync(serviceProvider);
}

// 配置中间件管道
app.UseHttpsRedirection();
app.UseAuthentication();
app.UseAuthorization();

// 映射端点（如Razor Pages、API控制器等）
app.MapRazorPages();

app.Run();

方式二：封装为扩展方法（代码更整洁） #

可以将初始化逻辑封装成WebApplication的扩展方法，提高代码复用性：

public static class WebApplicationExtensions
{
    public static async Task InitializeRolesAsync(this WebApplication app)
    {
        using var scope = app.Services.CreateScope();
        var roleManager = scope.ServiceProvider.GetRequiredService<RoleManager<IdentityRole>>();

        string[] roleNames = { "Admin", "Editor", "Viewer" };
        foreach (var roleName in roleNames)
        {
            if (!await roleManager.RoleExistsAsync(roleName))
            {
                await roleManager.CreateAsync(new IdentityRole(roleName));
            }
        }
    }
}

然后在Program.cs中直接调用：

var app = builder.Build();

// 调用扩展方法完成角色初始化
await app.InitializeRolesAsync();

// 后续中间件配置和端点映射...
app.Run();

关键注意事项 #

• 必须启用角色服务：在配置Identity时，务必添加.AddRoles<IdentityRole>()，否则无法获取RoleManager<IdentityRole>实例。
• 使用服务范围：通过CreateScope()创建服务范围，确保获取的服务符合其生命周期（RoleManager是Scoped生命周期），避免潜在的资源泄漏或生命周期问题。
• 优先使用异步调用：避免使用.Wait()或.Result()同步阻塞调用，防止死锁风险，直接使用await执行异步初始化逻辑。

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