asp.net core mvc权限控制：分配权限

转载：https://www.cnblogs.com/dxp909/p/6389198.html

 

前面的文章介绍了如何进行权限控制，即访问控制器或者方法的时候，要求当前用户必须具备特定的权限，但是如何在程序中进行权限的分配呢？下面就介绍下如何利用Microsoft.AspNetCore.Identity.EntityFrameworkCore框架进行权限分配。

在介绍分配方法之前，我们必须理解权限关系，这里面涉及到三个对象：用户，角色，权限，权限分配到角色，角色再分配到用户，当某个用户属于某个角色后，这个用户就具有了角色所包含的权限列表，比如现在有一个信息管理员角色，这个角色包含了信息删除权限，当张三这个用户具有信息管理员角色后，张三就具备了信息删除的权限。在某些特殊场景下，权限也可以直接分配到用户，也就是说可以直接把某些特定的权限，绕过角色，直接分配给用户。Microsoft.AspNetCore.Identity.EntityFrameworkCore框架中都提供了这样的支持。

先把框架中主要的业务对象类介绍一下：

IdentityUser：表示一个用户信息

IdentityRole：表示一个角色信息

IdentityRoleClaim:表示角色具有的权限

IdentityUserClaim:表示用户具有的权限

IdentityUserRole：表示用户角色关系

 

基本概念理解后，下面我们就来看一下如何进行权限分配。

1，分配权限到角色：Microsoft.AspNetCore.Identity.EntityFrameworkCore中提供了RoleManager类，类中提供了把权限分配到角色的方法：

　　Task AddClaimAsync(TRole role, Claim claim)

　　第一个参数表示对应的角色对象，第二个参数表示一个权限信息

2，分配权限到用户：Microsoft.AspNetCore.Identity.EntityFrameworkCore中提供了UserManager类，类中提供了把权限分配到用户的方法：

　　Task AddClaimAsync(TUser user, Claim claim)

　　第一个参数表示对应的用户对象，第二个参数表示一个权限信息

3，分配用户到角色：用到的同样是UserManager类，使用的方法：

　　AddToRoleAsync(TUser user, string role)

　　第一个参数表示的是用户对象，第二个是角色的名称

4，获取角色当前具有的权限列表：

　Task> RoleManager.GetClaimsAsync(TRole role)

5，获取用户当前具有的权限列表：

　Task> UserManager.GetClaimsAsync(TUser user)

 

 

通过这样的方式就可以完成权限分配全过程，再结合前面的权限控制方法，系统就实现了完成的权限控制逻辑。

那现在的问题来了，权限列表从什么地方来？一般来说，一个业务系统功能确定后，对应的权限列表也自然就确定了，再实现分配权限到角色，分配权限到用户的功能时，只需要在页面上把所有的权限列出来进行选择即可，效果图如下：

把选择的数据调用对应的方法保存即可。

这个问题解决了，但是新的问题又来了。如果说一个业务功能点特别多，自然会导致权限也会很多，如果完全通过手工的方式写到页面上，那自然工作量会很大很大，再者业务系统可能会不断地变化，这个时候也会去不断地修改权限分配页面，这自然不是一个好的方法，下面我给大家说一个我想的一个方法，不一定是最好的，但是能省很大的事。

首秀我们需要解决的问题是，如何快速生成这个权限配置列表。

思路就是改造AuthorizeAttribute，在这个特性基础上增加权限描述信息，用权限描述信息作为Policy。下面直接上代码：

[AttributeUsage(AttributeTargets.Class | AttributeTargets.Method, AllowMultiple =  true , Inherited = true )]　　 //类名称可以改，因为我把系统操作当作资源来管理
    public  class  ResourceAttribute:AuthorizeAttribute
    {
        private  string  _resouceName;
        private  string  _action;
        public  ResourceAttribute( string  name)
        {
            if  ( string .IsNullOrEmpty(name))
            {
                throw  new  ArgumentNullException(nameof(name));
            }
            _resouceName = name;
　　　　　　　 //把资源名称设置成Policy名称
            Policy = _resouceName;
        }

        public  string  GetResource()
        {
            return  _resouceName;
        }
        public  string  Action
        {
            get
            {
                return  _action;
            }
            set
            {
                _action = value;
                if  (! string .IsNullOrEmpty(value))
                {　　　　　　　　　　　　 //把资源名称跟操作名称组装成Policy
                    Policy = _resouceName +  "-"  + value;
                }
            }
        }
    }

　　

 

类已经定义好了，那我们就看看如何使用，因为是特性定义，所以可以在控制器类或者方法上按照下面结构使用：

[Resource("组织架构", Action = "添加部门")]

到这里基础工作已经做完，下面还有两个问题需要解决：

1，Policy现在只是配置了名称，但是具体验证规则没有定义

2，如何获取所有的权限列表

 

先来看第一个问题，前面的文章介绍了，Policy需要提前在startup里通过AddAuthorization进行配置，但是现在我们并没有做这样的步骤,所以目前权限还不会起作用。框架在权限验证的时候，会依赖一个IAuthorizationPolicyProvider来根据Policy名称获取具体的规则，自然我们会想到自定义一个IAuthorizationPolicyProvider实现，代码如下：

public  class  ResourceAuthorizationPolicyProvider : IAuthorizationPolicyProvider
    {
        private   AuthorizationOptions _options;
        public  ResourceAuthorizationPolicyProvider(IOptions options)
        {
            if  (options ==  null )
            {
                throw  new  ArgumentNullException(nameof(options));
            }

            _options = options.Value;
        }
        public  Task GetDefaultPolicyAsync()
        {
            return  Task.FromResult(_options.DefaultPolicy);
        }

        public  Task GetPolicyAsync( string  policyName)
        {
            AuthorizationPolicy policy = _options.GetPolicy(policyName);　　　　　　　 //因为我们policy的名称其实就是对应的权限名称，所以可以用下列逻辑返回需要的验证规则
            if  (policy ==  null )
            {
                string [] resourceValues = policyName.Split( new  char [] {  '-'  }, StringSplitOptions.None);
                if  (resourceValues.Length == 1)
                {
                    _options.AddPolicy(policyName, builder =>
                    {
                        builder.AddRequirements( new  ClaimsAuthorizationRequirement(resourceValues[0],  null ));
                    });
                }
                else
                {
                    _options.AddPolicy(policyName, builder =>
                    {
                        builder.AddRequirements( new  ClaimsAuthorizationRequirement(resourceValues[0],  new  string [] { resourceValues[1] }));
                    });
                }
            }
            return  Task.FromResult(_options.GetPolicy(policyName));
        }
    }

实现了IAuthorizationPolicyProvider，我们就需要在startup.cs的ConfigureServices(IServiceCollection services)方法中进行注册，操作如下：

services.TryAdd(ServiceDescriptor.Transient());

 

再来看第二个问题，我们已经在控制器或者方法上定义了权限信息，关键是我们如何从这些特性里获取到权限列表，将来用于权限分配的时候使用。在asp.net core mvc中提供了一个类解析机制，IApplicationModelProvider，这个依赖信息比较多，这里就不过多介绍，后续我会单独开一个系列，介绍asp.net core mvc的内部机制。

直接上代码

public  class  ResourceApplicationModelProvider : IApplicationModelProvider
    {
        private  readonly  IAuthorizationPolicyProvider _policyProvider;

        public  ResourceApplicationModelProvider(IAuthorizationPolicyProvider policyProvider)
        {
            _policyProvider = policyProvider;
        }


        public  void  OnProvidersExecuted(ApplicationModelProviderContext context)
        {

        }

        public  void  OnProvidersExecuting(ApplicationModelProviderContext context)
        {
            if  (context ==  null )
            {
                throw  new  ArgumentNullException(nameof(context));
            }

            List attributeData =  new  List();　　　　　　　　 //循环获取所有的控制器
            foreach  ( var  controllerModel  in  context.Result.Controllers)
            {　　　　　　　　 //得到ResourceAttribute
                var  resourceData = controllerModel.Attributes.OfType().ToArray();
                if  (resourceData.Length > 0)
                {
                    attributeData.AddRange(resourceData);
                }
　　　　　　　　　　 //循环控制器方法
                foreach  ( var  actionModel  in  controllerModel.Actions)
                {                    //得到方法的ResourceAttribute
                    var  actionResourceData = actionModel.Attributes.OfType().ToArray();
                    if  (actionResourceData.Length > 0)
                    {
                        attributeData.AddRange(actionResourceData);
                    }
                }
            }
　　　　　　　 //把所有的resourceattribute的信息写到一个全局的resourcedata中，resourcedata就可以在其他地方进行使用，resourcedata定义后面补充
            foreach  ( var  item  in  attributeData)
            {
                ResourceData.AddResource(item.GetResource(), item.Action);
            }
        }

        public  int  Order {  get  {  return  -1000 + 11; } }
    }

resourcedata定义如下

public  class  ResourceData
    {
        static  ResourceData()
        {
            Resources =  new  Dictionary< string , List< string >>();
        }

        public  static  void  AddResource( string  name)
        {
            AddResource(name,  "" );
        }

        public  static  void  AddResource( string  name, string  action)
        {
            if  ( string .IsNullOrEmpty(name))
            {
                return ;
            }
            if  (!Resources.ContainsKey(name))
            {
                Resources.Add(name,  new  List< string >());
            }

            if  (! string .IsNullOrEmpty(action) && !Resources[name].Contains(action))
            {
                Resources[name].Add(action);
            }
        }

        public  static  Dictionary< string , List< string >> Resources {  get ;  set ; }
    }

　然后在startup中注册我们刚刚定义的IApplicationModelProvider：

services.TryAddEnumerable(ServiceDescriptor.Transient());

　然后在权限分配页面通过ResourceData.Resources就获取到了所有的权限信息，然后通过循环的方式直接显示到页面上即可。　

 

终于写完了，哈哈~~

　

　　附上实例代码：http://files.cnblogs.com/files/dxp909/AuthorizeSample.rar