springboot：通过aop注解进行接口权限判断

你知道烟火吗

已于 2024-02-02 16:30:16 修改

阅读量510

点赞数 7

分类专栏： springboot 文章标签： java 后端 spring boot

于 2024-02-02 11:31:44 首次发布

本文链接：https://blog.csdn.net/Wanankl/article/details/135990179

版权

springboot 专栏收录该内容

4 篇文章 0 订阅

订阅专栏

本文介绍了如何使用Spring框架中的注解`@ActionRoles`实现接口级别的权限控制，包括创建注解、在接口上添加鉴权以及在`RoleAuthorizeAspect`中处理授权逻辑。

摘要由CSDN通过智能技术生成

1、创建注解

import org.springframework.core.annotation.Order;

import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;

/**
 * 操作权限注解
 * 可放到方法上，也可放到类上
 */
@Target({ElementType.METHOD,ElementType.TYPE})
@Retention(RetentionPolicy.RUNTIME)
@Order(3)
public @interface ActionRoles {
    String[] value();
}

2、在需要鉴权接口上添加直接

@PostMapping("test")
@ActionRoles(RoleType.ROLE_ADMIN)
public BiResponseDTO<String> test() {
    return BiResponseDTO.success();
}

3、创建AuthorizeAspect方法

@Slf4j
@Aspect
@Component
public class RoleAuthorizeAspect {

    @Around("@annotation(ActionRoles) || @within(ActionRoles)")
    public Object around(ProceedingJoinPoint point) throws Throwable {
        ActionRoles annotation = point.getTarget().getClass().getAnnotation(ActionRoles.class);
        if(annotation != null){
            checkRole(annotation.value());
            return point.proceed();
        }
        //获取连接点
        MethodSignature signature = (MethodSignature) point.getSignature();
        Method method = signature.getMethod();
        //获取方法ActionRoles注解
        annotation = method.getAnnotation(ActionRoles.class);
        if (annotation == null) {
            return point.proceed();
        }
        checkRole(annotation.value());
        return point.proceed();
    }
    
    private boolean checkRole(String[] roles) throws BaseException {
    	// 获取用户权限，与注解权限进行比较判断
        List<String> permissionList = user.getPermissionList();
        if(!ObjectUtil.isEmpty(roles) && ObjectUtil.isEmpty(permissionList)){
            log.warn("当前登录账号：{}不具备权限：{}，鉴权失败");
            return false;
        }
        return true;
    }
}