ASP.NET Core 标识 userManager.AddToRoleAsync() - 通过 id 添加

问题描述

我有一个自定义用户角色类，其中 CompanyId 作为外键。所以我可以创建与公司相关的用户角色。这意味着管理员角色或任何其他角色本身可以在多个公司中。我不能与公司和角色建立多对多关系，因为一家公司的管理员声明可能因另一家公司而异。所以我在用户角色实体中添加了公司 ID。

我的问题是，用户管理器将角色名称作为 AddToRoleAsync() 中的参数，我无法使用它，因为角色名称可能是多余的。如果我能够通过 Id 添加用户角色，那就太好了。是否有解决方法来实现这一目标？

我目前的解决方法是将公司名称添加为角色名称的前缀。喜欢：ABC - 管理员； XYZ - 管理员。我不太喜欢这个，用户也不喜欢。

如果我可以为 AddToRoleAsync() 和 IsInRoleAync() 创建一个重载方法，通过它的 ID 而不是名称来查找角色，那就太好了。

解决方法

您可以扩展 UserManager 类并添加另一个方法来添加角色 RoleId

public class ApplicationUserManager : UserManager<IdentityUser>
{
    private readonly UserStore<IdentityUser,IdentityRole,ApplicationDbContext,string,IdentityUserClaim<string>,IdentityUserRole<string>,IdentityUserLogin<string>,IdentityUserToken<string>,IdentityRoleClaim<string>> _store;

    public ApplicationUserManager(
        IUserStore<IdentityUser> store,IOptions<IdentityOptions> optionsAccessor,IPasswordHasher<IdentityUser> passwordHasher,IEnumerable<IUserValidator<IdentityUser>> userValidators,IEnumerable<IPasswordValidator<IdentityUser>> passwordValidators,ILookupNormalizer keyNormalizer,IdentityErrorDescriber errors,IServiceProvider services,ILogger<UserManager<IdentityUser>> logger)
        : base(store,optionsAccessor,passwordHasher,userValidators,passwordValidators,keyNormalizer,errors,services,logger)
    {
        _store = (UserStore<IdentityUser,IdentityRoleClaim<string>>)store;
    }

    public virtual async Task<IdentityResult> AddToRoleByRoleIdAsync(IdentityUser user,string roleId)
    {
        ThrowIfDisposed();

        if (user == null)
            throw new ArgumentNullException(nameof(user));

        if (string.IsNullOrWhiteSpace(roleId))
            throw new ArgumentNullException(nameof(roleId));

        if (await IsInRoleByRoleIdAsync(user,roleId,CancellationToken))
            return IdentityResult.Failed(ErrorDescriber.UserAlreadyInRole(roleId));

        _store.Context.Set<IdentityUserRole<string>>().Add(new IdentityUserRole<string> { RoleId = roleId,UserId = user.Id });

        return await UpdateUserAsync(user);
    }

    public async Task<bool> IsInRoleByRoleIdAsync(IdentityUser user,string roleId,CancellationToken cancellationToken = default(CancellationToken))
    {
        cancellationToken.ThrowIfCancellationRequested();
        ThrowIfDisposed();

        if (user == null)
            throw new ArgumentNullException(nameof(user));

        if (string.IsNullOrWhiteSpace(roleId))
            throw new ArgumentNullException(nameof(roleId));

        var role = await _store.Context.Set<IdentityRole>().FindAsync(roleId);
        if (role == null)
            return false;

        var userRole = await _store.Context.Set<IdentityUserRole<string>>().FindAsync(new object[] { user.Id,roleId },cancellationToken);
        return userRole != null;
    }
}

然后注册您拥有的UserManager：

services.AddIdentity<IdentityUser,IdentityRole>()
    .AddEntityFrameworkStores<ApplicationDbContext>()
    .AddUserManager<ApplicationUserManager>() // Add ApplicationUserManager
    .AddDefaultTokenProviders()
    .AddDefaultUI();

最后：

await _userManager.AddToRoleByRoleIdAsync(user,role.Id);