如何在ClaimsPrincipal扩展中高效初始化DbContext并优化性能？

核心问题拆解 #

你需要实现IsLimitedUser扩展方法判断用户关联公司的IsLimited标记，但扩展方法无法直接注入DbContext，且该方法会在每次页面请求时触发，需兼顾性能与资源管理。

最优实现方案 #

方案一：登录时将IsLimited存入Claims（推荐） #

仅在用户登录时查询一次数据库，后续请求直接读取Claims，性能最优：

1. 登录逻辑中添加Claims

// 登录时获取用户及关联公司信息
var user = await dbContext.Users.Include(u => u.Facility.Company)
                               .FirstOrDefaultAsync(u => u.Id == userId);
if (user != null)
{
    var claims = new List<Claim>
    {
        new Claim(ClaimTypes.NameIdentifier, user.Id),
        // 将IsLimited标记存入Claims
        new Claim("IsLimitedUser", user.Facility.Company.IsLimited.ToString())
    };

    var identity = new ClaimsIdentity(claims, CookieAuthenticationDefaults.AuthenticationScheme);
    var principal = new ClaimsPrincipal(identity);

    await HttpContext.SignInAsync(principal);
}

2. 扩展方法直接读取Claims

public static class ClaimsPrincipalExtensions
{
    public static bool IsLimitedUser(this ClaimsPrincipal principal)
    {
        ArgumentNullException.ThrowIfNull(principal);
        var isLimitedClaim = principal.FindFirst("IsLimitedUser");
        return isLimitedClaim != null && bool.TryParse(isLimitedClaim.Value, out var result) && result;
    }
}

方案二：首次查询后缓存到当前请求Claims #

若无法在登录时添加Claims，可在首次调用时查询数据库，将结果缓存到当前请求的ClaimsPrincipal中（仅本次请求有效）：

1. 创建服务类处理查询与缓存

public class UserLimitService
{
    private readonly AppDbContext _dbContext;
    private readonly IHttpContextAccessor _httpContextAccessor;

    public UserLimitService(AppDbContext dbContext, IHttpContextAccessor httpContextAccessor)
    {
        _dbContext = dbContext;
        _httpContextAccessor = httpContextAccessor;
    }

    public async Task<bool> IsLimitedUserAsync()
    {
        var principal = _httpContextAccessor.HttpContext?.User;
        if (principal == null)
            return false;

        // 优先读取当前请求缓存的Claims
        var cachedClaim = principal.FindFirst("IsLimitedUser");
        if (cachedClaim != null)
            return bool.TryParse(cachedClaim.Value, out var result) && result;

        var userIdClaim = principal.FindFirst(ClaimTypes.NameIdentifier);
        if (userIdClaim == null)
            return false;

        // 查询数据库
        var isLimited = await _dbContext.Users
            .Where(u => u.Id == userIdClaim.Value)
            .Select(u => u.Facility.Company.IsLimited)
            .FirstOrDefaultAsync();

        // 将结果存入当前请求的Claims
        var identity = principal.Identity as ClaimsIdentity;
        identity?.AddClaim(new Claim("IsLimitedUser", isLimited.ToString()));

        return isLimited;
    }
}

2. 注册服务

// Program.cs中注册依赖
builder.Services.AddScoped<UserLimitService>();
builder.Services.AddHttpContextAccessor();

3. 在_Layout.cshtml中注入使用

@inject UserLimitService UserLimitService

@{
    var isLimited = await UserLimitService.IsLimitedUserAsync();
}

<!-- 根据isLimited值渲染页面内容 -->

方案三：扩展方法传入DbContext（不推荐） #

若坚持使用扩展方法，避免直接初始化DbContext，通过参数传入，但无缓存性能较差：

public static class ClaimsPrincipalExtensions
{
    public static async Task<bool> IsLimitedUserAsync(this ClaimsPrincipal principal, AppDbContext dbContext)
    {
        ArgumentNullException.ThrowIfNull(principal);
        ArgumentNullException.ThrowIfNull(dbContext);

        var claim = principal.FindFirst(ClaimTypes.NameIdentifier);
        if (claim != null)
        {
            return await dbContext.Users
                .Where(u => u.Id == claim.Value)
                .Select(u => u.Facility.Company.IsLimited)
                .FirstOrDefaultAsync();
        }
        return false;
    }
}

注意事项 #

• 若公司IsLimited标记可能在用户登录后变更，需添加缓存失效机制：比如使用分布式缓存并设置过期时间，或在标记变更时强制用户重新登录/刷新Claims。
• 使用IHttpContextAccessor时，必须在Program.cs中注册该服务。

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