Springboot自定义注解

---

  需求：通过Springboot自定义注解的方式来实现权限和日志模块的引用（这里侧重应用层面作用）。

---

1、工程描述

---

a1.png

  说明：InAuth.java为要定义的注解接口，AuthAspect.java为切面实现，UserAuthEnum.java为用户权限枚举，AuthServiceImpl.java为业务实现，Auth.java为对外提供接口（控制层）。

---

2、代码步骤

---

第一：定义注解接口

@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.METHOD)
public @interface InAuth {
    UserAuthEnum[] limit() default {};
}

第二：定义切面

/*****************************************
 ***说明：这里重点介绍权限,忽略登录会话等***
 *****************************************/
@Aspect
@Component
@Configuration
public class AuthAspect {
    @Around("@annotation(inAuth)")
    public Object around(ProceedingJoinPoint point, InAuth inAuth) throws Throwable {
        UserAuthEnum[] ifAuth = inAuth.limit();// 获取接口权限
        if (ifAuth == null || ifAuth.length == 0) {
            return point.proceed(); // 该接口不需要权限
        }
        int[] limit = {10,20,100};  //将ifAuth转成权限码
        // 1、获取用户的会话Token，我们是放在ThreadLocal中的
        // 2、根据Token会话中取出用户的权限列表，会话存在redis中
        // 3、上面两步略，用户权限直接用数据模拟，我们用户权限返回的是Set
        // 4、可以加一些逻辑判断，返回一些异常情况
        Set<Integer> userSet = new HashSet<Integer>();
        userSet.add(1000) ;
        for (int lt : limit) {
            if (userSet.contains(lt)) {
                return point.proceed();
            }
        }
        return "no acess auth!";
    }
}

第三：定义权限的枚举类

public enum UserAuthEnum {
    A_ADMIN(10), // 管理员权限
    A_READ(20), // 只读权限
    A_UPLOAD(30), // 上报权限
    A_CHARTS(40), // 报表权限
    A_LIST(50);// 名单管理权限
    private int code;
    private UserAuthEnum(int code) {
        this.code = code;
    }
    public int getCode() {
        return this.code;
    }
}

第四：定义业务接口(引用自定的注解)

@Service("authService")
public class AuthServiceImpl implements AuthService {
    @Override
    @InAuth(limit={UserAuthEnum.A_CHARTS,UserAuthEnum.A_ADMIN}) //要么有管理员权限，要么有charts权限可以访问
    public String testAuth() {
        return "---auth test---";
    }
}

第五：定义切面

public interface EacProxy {
     public String query();  //可以返回复杂对象类型
}

第六：定义测试接口

@RestController
public class Auth {
    @Autowired
    private AuthService authService ;

    @RequestMapping("/auth")
    public String sayHello() throws Exception {
        return authService.testAuth() ;
    }
}

  注解让代码代码世界更简洁、更简单了，但也存在缺点，让异常定位更麻烦，让理解spring执行原理更复杂（感觉从spring原理慢慢修理才是正道）。

---

3、打包下载

---

链接：https://pan.baidu.com/s/15upCKAvQerQr7zpuJNAy9Q 
提取码：i9dm 

  若链接莫名失效，可以留言联系，也可以按照上述的方式自行配置。

更多可关注