拦截器完成权限控制

Bertha 。 2022-09-15 12:59 160阅读 0赞

## 如何通过拦截器完成权限控制 ##

### 拦截器 ###

#### 概念 ####

java里的拦截器是动态拦截Action调用的对象，它提供了一种机制可以使开发者在一个Action执行的前后执行一段代码，也可以在一个Action  
执行前阻止其执行，同时也提供了一种可以提取Action中可重用部分代码的方式。在AOP中，拦截器用于在某个方法或者字段被访问之前，进行拦截  
然后再之前或者之后加入某些操作。

### 实现方式 ###

*  第一步 定义注解

package com.alibaba.gov.kamala2.service.annotation;
    
    import com.alibaba.gov.kamala2.enums.userrole.Kamala2LevelEnum;
    
    import java.lang.annotation.*;
    
    @Target({ElementType.TYPE, ElementType.METHOD})
    @Retention(RetentionPolicy.RUNTIME)
    @Documented
    public @interface AclUrlAutoHandle {
        /**
         * 权限名称
         * @return
         */
        String name();
        /**
         * 默认值 暂无意义
         * @return
         */
        String value() default "";
        /**
         * 唯一码
         * @return
         */
        String code();
    
        /**
         * 优先级
         * @return
         */
        Kamala2LevelEnum level() default Kamala2LevelEnum.LOW;
    
        /**
         * 是否需要校验node权限
         * @return
         */
        boolean nodeFlg() default false;
    
        /**
         * 描述
         * @return
         */
        String description() default "系统添加的权限";
    }

*  第二步 定义拦截器类，该类实现HandlerInterceptor接口，并重写preHandle、postHandle、afterCompletion方法（如有需要）；此处只需要重写preHandler即可；

import org.springframework.stereotype.Component;
    import org.springframework.web.servlet.HandlerInterceptor;
    import org.springframework.web.servlet.ModelAndView;
     
    import javax.servlet.http.HttpServletRequest;
    import javax.servlet.http.HttpServletResponse;
     
    @Component
    public class MyInterceptor implements HandlerInterceptor {
     
     //注入上下文
    	@Autowired
        private ApplicationContext applicationContext;
        /**
         * 该方法是在执行执行servlet的 service方法之前执行的
         * 即在进入controller之前调用
         * @return 如果返回true表示继续执行下一个拦截器的PreHandle方法；如果没有拦截器了,则执行controller
         */
        @Override
        public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception{
          RequestMappingHandlerMapping rmhp = applicationContext.getBean(RequestMappingHandlerMapping.class);
            Map<RequestMappingInfo, HandlerMethod> map = rmhp.getHandlerMethods();
            for(RequestMappingInfo info : map.keySet()){
                HandlerMethod hm=map.get(info);
                Method method = hm.getMethod();
               Object o = method.getAnnotation(AclUrlAutoHandle.class);
                if(o!= null){
                    System.out.println(info.getPatternsCondition());
                }
            }
            return true;
        }
     
        /**
         *在执行完controller之后，返回视图之前执行，我们可以对controller返回的结果做处理
         * 执行顺序：先执行最后一个拦截器的postHandle方法，一次向前
         */
        @Override
        public void postHandle(HttpServletRequest request, HttpServletResponse response, Object handler,
                               ModelAndView modelAndView) throws Exception{
     
            System.out.println("解析视图之前.....");
        }
     
        /**
         * 整个请求结束之后，即返回视图之后执行
         *该方法需要同一拦截器的preHandle返回true时执行，
         * 如果该拦截器preHandle返回false，则该拦截器的afterCompletion不执行
         * 执行顺序：先执行最后一个返回true的拦截器的afterCompletion，在依次向前
         */
        @Override
        public void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler,
                          Exception ex) throws Exception{
            System.out.println("视图解析完成...");
        }
    }

*  第三步：重写preHandler方法，在该方法体中获取含有自定义注解的url，并保存到set中；

@Override
        public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception{
          RequestMappingHandlerMapping rmhp = applicationContext.getBean(RequestMappingHandlerMapping.class);
           Set<String> set = new HashSet<>();
            Map<RequestMappingInfo, HandlerMethod> map = rmhp.getHandlerMethods();
            for(RequestMappingInfo info : map.keySet()){
                HandlerMethod hm=map.get(info);
                Method method = hm.getMethod();
               Object o = method.getAnnotation(AclUrlAutoHandle.class);
                if(o!= null){
                    System.out.println(info.getPatternsCondition());
                    set.add(info.getPatternsCondition().toString());
                }
            }
            return true;
        }

*  根据用户、角色、权限、授权记录四张表中的逻辑关系，来确定该用户是否有某一路径的权限（此处不做详细展开）
 *