Spring Boot使用自定義註解和攔截器進行權限控制

最近工作中需要在微信管理端的應用中使用自定義註解實現權限控制,必須保持和後台權限一致,因為賬戶系統分為partner、alliance、merchant三種,頁面和權限也分為三套所以微信後台也要分別對著三種權限進行控制。

实现HandlerInterceptor

public class AuthenticationInterceptor extends HandlerInterceptorAdapter{
    @Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler)
            throws Exception {
        // 将handler强转为HandlerMethod, 前面已经证实这个handler就是HandlerMethod
        HandlerMethod handlerMethod = (HandlerMethod) handler;
        // 从方法处理器中获取出要调用的方法
        Method method = handlerMethod.getMethod();
        // 获取出方法上的RequiresPermissions注解
        RequiresPermissions permissions = method.getAnnotation(RequiresPermissions.class);
        if (permissions == null) {
            // 如果注解为null, 说明不需要拦截, 直接放过
            return true;
        }
        
        String accountType = SessionUtil.getAccountType(request);
        // 获取用户所有权限
        Set<String> permsSet = SessionUtil.permissions(request, response);
        String[] values = null;
        if (accountType.equals(USER_TYPE.PARTNER.key)) {
            values = permissions.partner();
        } else if (accountType.equals(USER_TYPE.ALLIANCE.key)) {
            values = permissions.alliance();
        } else if (accountType.equals(USER_TYPE.MERCHANT.key)) {
            values = permissions.merchant();
        }
        
        if (values.length > 0) {
            // 权限列表为空
            if (permsSet.size() == 0) {
                if (SessionUtil.isSuperAdmin(request.getSession())) {
                    // 如果是超级管理员
                    return true;
                } else {
                    return false;
                }
            }
            for (String permission : values) {
                if (!permsSet.contains(permission)) {
                    return false;
                }
            }
        }
        
        // 拦截之后应该返回公共结果, 这里没做处理
        return true;
    }
}

配置拦截器

@Configuration
public class InterceptorConfig extends WebMvcConfigurerAdapter{
    @Override
    public void addInterceptors(InterceptorRegistry registry) {
        // TODO Auto-generated method stub
        registry.addInterceptor(new AuthenticationInterceptor()).addPathPatterns("/**");
    }
}

自定义权限注解

@Target({ElementType.TYPE, ElementType.METHOD})
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface RequiresPermissions {
    String[] partner() default {};
    String[] alliance() default {};
    String[] merchant() default {};
}

參考文章:SpringBoot使用自定义注解实现权限拦截

©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • 程序員創業白皮書 作者:Paul Graham Paul Graham是程序員,專欄作家。他在1995年創建了第一...
    刘立山John阅读 1,944评论 0 20
  • 在4月份中,简书(非官方)QQ群共18位作者分享了61篇文章。 1.作者劉漢皇 ■邯郸学步:别用你的无知装高尚 ■...
    康查舒阅读 316评论 1 1
  • 期中考试结束了(就在今天) 自我感觉不好 好怕父母斥责! 我也努力学习了,成绩还提不上去 怎么办?我初三了!求助啊
    黑夜不懂爷的黑阅读 124评论 1 1
  • 上虞的老同学,从小学结下的姐妹,一早发消息给我,说寄出了杨梅,明日中午到。 上虞盛产水果,这些年打造的“四季鲜果”...
    涓子Fiona阅读 584评论 1 50