使用注解和AOP实现接口权限控制

最新推荐文章于 2025-03-18 16:12:24 发布

lhc66666

最新推荐文章于 2025-03-18 16:12:24 发布

阅读量969

点赞数

分类专栏： java spring 文章标签： java

本文链接：https://blog.youkuaiyun.com/lhc66666/article/details/124868393

版权

java 同时被 2 个专栏收录

9 篇文章

订阅专栏

spring

4 篇文章

订阅专栏

本文介绍了一种通过在Java方法上添加自定义注解`@RequirePermission`和创建切面类`PermissionAspect`来实现运行时权限检查的机制。在`@RequirePermission`注解中定义了需要的权限码，在切面类中，通过AOP拦截注解标记的方法，检查当前用户是否具备所需权限，从而实现细粒度的权限控制。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1. 首先先写一个注解

@Documented
@Retention(value = RetentionPolicy.RUNTIME)
@Target({ElementType.METHOD, ElementType.TYPE})
public @interface RequirePermission {
    PermissionCode[] value();
}

2.在需要权限控制的方法上加上注解

@PostMapping("/add")
    @RequirePermission(PermissionCode.SANCUS_ADMIN)
    public Result<Long> add(@RequestBody MarketingRules marketingRules) {
        OneLogUtil.createLog(LOG_ITEM_NAME, JSON.toJSONString(marketingRules));
        return RESULT_PROVIDER.getResult(marketingRulesManager.add(marketingRules));
    }

3. 写一个切面类

@Aspect
@Component
@Slf4j
public class PermissonAspect {

    @Autowired
    protected HttpServletRequest request;

    @Autowired
    private PermissionManager permissionManager;

    private static final ResultProvider RESULT_PROVIDER = ResultProvider.init(ModuleCode.NOT_USE);

    @Pointcut("@annotation(com.taobao.datasecurity.annotation.RequirePermission)")
    public void pointCut() {

    }

    @Around("pointCut()")
    public Object exceptionHandler(ProceedingJoinPoint joinPoint) throws Throwable {
        MethodSignature signature = (MethodSignature) joinPoint.getSignature();
        RequirePermission requirePermission = signature.getMethod().getAnnotation(RequirePermission.class);
        boolean requireAdmin = Arrays.stream(requirePermission.value()).anyMatch(it -> it.equals(PermissionCode.SANCUS_ADMIN));
        User currentUser = BucUserUtil.getCurrentUser(request);
        Boolean sancusAdmin = permissionManager.isSancusAdmin(currentUser);
        if (requireAdmin && sancusAdmin) {
            return joinPoint.proceed();
        }
        return RESULT_PROVIDER.getResult(UNAUTHORIZED);
    }
}