如何將自定義角色管理器插入AuthorizeAttribute？

Question

我有一個較舊的數據庫模式，我無法更改。它有一個帶有整數字段的單個用戶表來指定用戶級別，其中1是標準用戶，5是管理員。我正在編寫一個MVC前端，我想使用ASP.NET身份。我從研究和樣板代碼中找出了一切。我似乎無法弄清楚如何創建一個自定義角色系統。我意識到這與實施角色管理器和角色存儲庫有關。這很好，但我如何將它與MVC連接起來，讓AuthorizeAttribute確認我的經理？

我道歉，如果這是顯而易見的，但我已經做了我的研究，我有麻煩釘下來。

Answer 1

從你的問題，我假設你已經想通了如何創建自己的角色經理和你唯一缺少的配置來使用它。如果我的假設是錯誤的，請告訴我，我將添加關於如何創建CustomRoleManager的解釋。

的Web.config

<configuration> 
    <system.web> 
    <roleManager enabled="true" defaultProvider="CustomRoleProvider"> 
     <providers> 
     <clear/> 
     <add name="CustomRoleProvider" 
      type="MyNamespace.CustomRoleProvider, 
        MyNamespace, Version=1.0.0.0, Culture=neutral" 
      connectionStringName="MyConnectionString" 
      enablePasswordRetrieval="false" 
      enablePasswordReset="false" 
      requiresQuestionAndAnswer="false" 
      writeExceptionsToEventLog="false" /> 
     </providers> 
    </roleManager> 
    </system.web> 
</configuration> 

Answer 2

這裏是我使用的RoleProvider，如果任何人有同樣的瑣碎的要求。如果您知道此實施不安全的原因，請告訴我。我在我的Web.Config中使用@ Pluc的答案將此提供程序連接到我的應用程序。它奇妙地工作。

public class AppRole : IRole<int> 
{ 
    public AppRole(int a_id, string a_name) 
    { 
     Id = a_id; 
     Name = a_name; 
    } 

    public int Id { get; private set; } 
    public string Name { get; set; } 
} 

public class AppRoleProvider : RoleProvider 
{ 
    private readonly IServiceLocator _container = UnityConfig.GetServiceLocator(); 
    private ITrainingRepository _repository; // Thin wrapper around my DbContext 

    private AppRole[] _roles = new[] 
     { 
      new AppRole(0, "User"), 
      new AppRole(5, "Admin"), 
     }; 

    public AppRoleProvider() 
    { 
     ApplicationName = "TrainingCenter"; 

     _repository = _container.GetInstance<ITrainingRepository>(); 
    } 

    public override string ApplicationName { get; set; } 

    public override bool IsUserInRole(string username, string roleName) 
    { 
     var user = _repository.GetUserByUserName(username); 
     if (user == null) 
      return false; 

     var role = _roles.FirstOrDefault(i => i.Name.Equals(roleName, StringComparison.OrdinalIgnoreCase)); 

     if (role == null) 
      return false; 

     if (user.UserLevel >= role.Id) 
      return true; 

     return false; 
    } 

    public override string[] GetRolesForUser(string username) 
    { 
     var user = _repository.GetUserByUserName(username); 
     if (user == null) 
      return new string[] {}; 

     return _roles.Where(i => i.Id <= user.UserLevel).Select(i => i.Name).ToArray(); 
    } 

    public override void CreateRole(string roleName) 
    { 
     // Does not create. 
    } 

    public override bool DeleteRole(string roleName, bool throwOnPopulatedRole) 
    { 
     // Does not delete. 
     return false; 
    } 

    public override bool RoleExists(string roleName) 
    { 
     return _roles.Any(i => i.Name.Equals(roleName, StringComparison.OrdinalIgnoreCase)); 
    } 

    public override void AddUsersToRoles(string[] usernames, string[] roleNames) 
    { 
     // Does not add user to role. 
    } 

    public override void RemoveUsersFromRoles(string[] usernames, string[] roleNames) 
    { 
     // Does not remove users from roles. 
    } 

    public override string[] GetUsersInRole(string roleName) 
    { 
     // Does not get users in role. 
     return new string[] {}; 
    } 

    public override string[] GetAllRoles() 
    { 
     return _roles.Select(i => i.Name).ToArray(); 
    } 

    public override string[] FindUsersInRole(string roleName, string usernameToMatch) 
    { 
     // Does not find users in role. 
     return new string[] { }; 
    } 

}