谈谈 Java 中自定义注解及使用场景

点击上方“全栈程序员社区”，星标公众号
java

重磅干货，第一时间送达

做者：快给我饭吃程序员

www.jianshu.com/p/a7bedc771204web

Java自定义注解通常使用场景为：自定义注解+拦截器或者AOP，使用自定义注解来本身设计框架，使得代码看起来很是优雅。本文将先从自定义注解的基础概念提及，而后开始实战，写小段代码实现自定义注解+拦截器，自定义注解+AOP。面试

一. 什么是注解(Annotation)

Java注解是什么，如下是引用自维基百科的内容spring

Java注解又称Java标注，是JDK5.0版本开始支持加入源代码的特殊语法元数据。json

Java语言中的类、方法、变量、参数和包等均可以被标注。和Javadoc不一样，Java标注能够经过反射获取标注内容。在编译器生成类文件时，标注能够被嵌入到字节码中。Java虚拟机能够保留标注内容，在运行时能够获取到标注内容。固然它也支持自定义Java标注。后端

二. 注解体系图

元注解：java.lang.annotation中提供了元注解，可使用这些注解来定义本身的注解。主要使用的是Target和Retention注解springboot

注解处理类：既然上面定义了注解，那得有办法拿到咱们定义的注解啊。java.lang.reflect.AnnotationElement接口则提供了该功能。注解的处理是经过java反射来处理的。微信

以下，反射相关的类Class, Method, Field都实现了AnnotationElement接口。app

所以，只要咱们经过反射拿到Class, Method, Field类，就可以经过getAnnotation(Class<T>)拿到咱们想要的注解并取值。

搜索Java知音公众号，回复“后端面试”，送你一份Java面试题宝典

三. 经常使用元注解

Target：描述了注解修饰的对象范围，取值在java.lang.annotation.ElementType定义，经常使用的包括：

METHOD：用于描述方法
PACKAGE：用于描述包
PARAMETER：用于描述方法变量
TYPE：用于描述类、接口或enum类型

Retention: 表示注解保留时间长短。取值在java.lang.annotation.RetentionPolicy中，取值为：

SOURCE：在源文件中有效，编译过程当中会被忽略
CLASS：随源文件一块儿编译在class文件中，运行时忽略
RUNTIME：在运行时有效

只有定义为RetentionPolicy.RUNTIME时，咱们才能经过注解反射获取到注解。

因此，假设咱们要自定义一个注解，它用在字段上，而且能够经过反射获取到，功能是用来描述字段的长度和做用。

@Target(ElementType.FIELD)  //  注解用于字段上
@Retention(RetentionPolicy.RUNTIME)  // 保留到运行时，可经过注解获取
public @interface MyField {
    String description();
    int length();
}

四. 示例-反射获取注解

先定义一个注解：

@Target(ElementType.FIELD)
@Retention(RetentionPolicy.RUNTIME)
public @interface MyField {
    String description();
    int length();
}

经过反射获取注解

public class MyFieldTest {

    //使用咱们的自定义注解
    @MyField(description = "用户名", length = 12)
    private String username;

    @Test
    public void testMyField(){

        // 获取类模板
        Class c = MyFieldTest.class;

        // 获取全部字段
        for(Field f : c.getDeclaredFields()){
            // 判断这个字段是否有MyField注解
            if(f.isAnnotationPresent(MyField.class)){
                MyField annotation = f.getAnnotation(MyField.class);
                System.out.println("字段:[" + f.getName() + "], 描述:[" + annotation.description() + "], 长度:[" + annotation.length() +"]");
            }
        }

    }
}

运行结果

应用场景一：自定义注解+拦截器实现登陆校验

接下来，咱们使用springboot拦截器实现这样一个功能，若是方法上加了@LoginRequired，则提示用户该接口须要登陆才能访问，不然不须要登陆。

首先定义一个LoginRequired注解

@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
public @interface LoginRequired {
    
}

而后写两个简单的接口，访问sourceA，sourceB资源

@RestController
public class IndexController {

    @GetMapping("/sourceA")
    public String sourceA(){
        return "你正在访问sourceA资源";
    }

    @GetMapping("/sourceB")
    public String sourceB(){
        return "你正在访问sourceB资源";
    }

}

没添加拦截器以前成功访问

实现spring的HandlerInterceptor 类先实现拦截器，但不拦截，只是简单打印日志，以下：

public class SourceAccessInterceptor implements HandlerInterceptor {
    @Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
        System.out.println("进入拦截器了");
        return true;
    }

    @Override
    public void postHandle(HttpServletRequest request, HttpServletResponse response, Object handler, ModelAndView modelAndView) throws Exception {

    }

    @Override
    public void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler, Exception ex) throws Exception {

    }
}

实现spring类WebMvcConfigurer，建立配置类把拦截器添加到拦截器链中

@Configuration
public class InterceptorTrainConfigurer implements WebMvcConfigurer {
    @Override
    public void addInterceptors(InterceptorRegistry registry) {
        registry.addInterceptor(new SourceAccessInterceptor()).addPathPatterns("/**");
    }
}

拦截成功以下

在sourceB方法上添加咱们的登陆注解@LoginRequired

@RestController
public class IndexController {

    @GetMapping("/sourceA")
    public String sourceA(){
        return "你正在访问sourceA资源";
    }

    @LoginRequired
    @GetMapping("/sourceB")
    public String sourceB(){
        return "你正在访问sourceB资源";
    }

}

简单实现登陆拦截逻辑

@Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
        System.out.println("进入拦截器了");

        // 反射获取方法上的LoginRequred注解
        HandlerMethod handlerMethod = (HandlerMethod)handler;
        LoginRequired loginRequired = handlerMethod.getMethod().getAnnotation(LoginRequired.class);
        if(loginRequired == null){
            return true;
        }

        // 有LoginRequired注解说明须要登陆，提示用户登陆
        response.setContentType("application/json; charset=utf-8");
        response.getWriter().print("你访问的资源须要登陆");
        return false;
    }

运行成功，访问sourceB时须要登陆了，访问sourceA则不用登陆

应用场景二：自定义注解+AOP 实现日志打印

先导入切面须要的依赖包

<dependency>
      <groupId>org.springframework.boot</groupId>
     <artifactId>spring-boot-starter-aop</artifactId>
</dependency>

定义一个注解@MyLog

@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
public @interface MyLog {
    
}

定义一个切面类，见以下代码注释理解：

@Aspect // 1.代表这是一个切面类
@Component
public class MyLogAspect {

    // 2. PointCut表示这是一个切点，@annotation表示这个切点切到一个注解上，后面带该注解的全类名
    // 切面最主要的就是切点，全部的故事都围绕切点发生
    // logPointCut()表明切点名称
    @Pointcut("@annotation(me.zebin.demo.annotationdemo.aoplog.MyLog)")
    public void logPointCut(){};

    // 3. 环绕通知
    @Around("logPointCut()")
    public void logAround(ProceedingJoinPoint joinPoint){
        // 获取方法名称
        String methodName = joinPoint.getSignature().getName();
        // 获取入参
        Object[] param = joinPoint.getArgs();

        StringBuilder sb = new StringBuilder();
        for(Object o : param){
            sb.append(o + "; ");
        }
        System.out.println("进入[" + methodName + "]方法,参数为:" + sb.toString());

        // 继续执行方法
        try {
            joinPoint.proceed();
        } catch (Throwable throwable) {
            throwable.printStackTrace();
        }
        System.out.println(methodName + "方法执行结束");

    }
}

在步骤二中的IndexController写一个sourceC进行测试，加上咱们的自定义注解：

    @MyLog
    @GetMapping("/sourceC/{source_name}")
    public String sourceC(@PathVariable("source_name") String sourceName){
        return "你正在访问sourceC资源";
    }

启动springboot web项目，输入访问地址

本文分享自微信公众号 - 全栈程序员社区（mush_it）。
若有侵权，请联系 support@oschina.cn 删除。
本文参与“OSC源创计划”，欢迎正在阅读的你也加入，一块儿分享。