004-构建Auth集合

读取标注了@Controller的类的所有方法,如果方法上标注了@SourceAuth注解,则解析成为AuthVO对象

@Component
public class AuthorityConfig implements ApplicationContextAware, ApplicationRunner {

	private ApplicationContext applicationContext;

	@Override
    public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
        this.applicationContext = applicationContext;
    }

	@Override
    public void run(ApplicationArguments args) throws Exception {
        // 读取菜单配置
        MenuContext menuContext = buildMenu();
		// 构建权限
        List<AuthVO> authList = buildAuth();
    }

	/**
     * 构建权限
     */
    private List<AuthVO> buildAuth() {
        List<AuthVO> authList = new ArrayList<>();
        // 读取权限配置
        Map<String, Object> beansWithAnnotation = applicationContext.getBeansWithAnnotation(Controller.class);
        beansWithAnnotation.forEach((className, controller) -> {
            Class<?> clazz = controller.getClass();

            Method[] methods = controller.getClass().getMethods();
            for (Method method : methods) {
                Method mostSpecificMethod = AopUtils.getMostSpecificMethod(method, clazz);
                SourceAuth sourceAuth = mostSpecificMethod.getAnnotation(SourceAuth.class);
                if (Objects.isNull(sourceAuth)) {
                    continue;
                }
                AuthVO authVO;
                String[] menuKeys = sourceAuth.menuKeys();
                String[] authKeys = sourceAuth.authKeys();
                for (String menuKey : menuKeys) {
                    for (String authKey : authKeys) {
                        authVO = new AuthVO();
                        authVO.setMenuKey(menuKey);
                        authVO.setAuthKey(menuKey + "." + authKey);
                        authVO.setAuthName(sourceAuth.authName());
                        authList.add(authVO);
                    }
                }
            }
        });
        return authList.stream().distinct().collect(Collectors.toList());
    }

}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值