关于JAVA中源码级注解的编写及使用
少年阿涛 2020-02-18
一、注解简介：
1.1.什么是“注解”：
? 在我们编写代码时，一定看到过这样的代码：
class Student {
private String name;
@Override
public String toString(String str) {//编译错误！
return "Student name = " + name;
}
}? 其中的@Override，就是一个“注解”，@Override一般出现在重写equals()或者toString()方法的上边，意思是告诉编译器：下边的代码是重写父类方法的。这时编译器会按照“重写”的语法严格检查下面的方法，如果不符合重写语法，将会编译错误。
? "注解"作为一种“标记”，被写在源码中，不会改变程序的执行流程。它通常由“注解解析工具”来解析，而“注解解析器”可以随Java编译器启动，也可以独立启动，来解析注解，并以此可以做一些事情。
1.2.注解的分类：
源码注解：
? 注解只在源码中，编译成class文件后就不存在了。
编译时注解：
? 注解在源码和.class文件中都存在（如：JDK内置系统注解）
运行时注解：
? 在运行阶段还起作用，甚至会影响运行逻辑的注解（如：JUnit的@Test）
1.3.注解的作用
? 注解的作用非常广泛，注解可以被用在类、属性、构造方法、成员方法、局部变量等位置，用于对这些元素进行说明。由“注解解析工具”解析后，可以生成文档、进行代码分析、编译检查等。
? 本例将会实现一个用作"编译检查“的注解，以及一个"注解解析器"。"注解解析器"将会随着javac编译器一同启动来对使用了注解的类进行编译，并检查类名、字段名、方法名是否以大写、小写字符开头，如果违反了规则，编译时将会报错。
二、自定义注解：
2.1.定义注解的基本语法
? “注解”本质上是一个“类”，我们可以根据自己的需要定义自己的注解。
? 定义注解的语法很简单：
public @interface CheckWord{
...
}? "注解”编译后会生成.class文件。但这是一个非常简单的注解，它可以被用在任何位置，而且编译器遇到这种注解也不做任何事情。例如：
@CheckWord
public class Student {
@CheckWord
public Student() {
}
@CheckWord
private String name;
@CheckWord
public void study() {
}
}下面我们先使用“元注解”来规定这个注解可以被用在哪里。
2.2.元注解
? “元注解”也是一种“注解”，它是已经实现好的。必须用在“注解”的定义上，它可以规定注解可以用在哪里，以及可以存在于源码中，或者class中，或者运行时。
? 常用的“元注解”有两个：
? 1) : 规定注解可以用在哪里。常用的取值被定义在枚举java.lang.annotation.ElementType中：
? ElementType.TYPE：类和接口上
? ElementType.FIELD： 用在成员变量上
? ElementType.METHOD： 用在方法上
? ElementType.PARAMETER： 用在参数上
? ElementType.CONSTRUCTOR： 用在构造方法上
? ElementType.LOCAL_VARIABLE： 用在局部变量上
? 2) : 规定注解可以存在于哪里。常用的取值被定义在枚举java.lang.annotation.RetentionPolicy中：
? RetentionPolicy.SOURCE： 规定注解只存在于Java源代码中， 编译生成的字节码文件中就不存在了。
? RetentionPolicy.CLASS： 规定注解存在于Java源代码、 编译以后的字节码文件中， 但JVM运行时，不会被加载到内存。
? RetentionPolicy.RUNTIME： 规定注解存在于Java源代码中、 编译以后的字节码文件中、 运行时内存中， 程序可以通过反射获取该注解。
? 例如：修改我们的注解，规定它只能用在"类","字段"，“方法”上，并且可以存在于“源码中”：
import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;
@Target({ElementType.TYPE, ElementType.FIELD, ElementType.METHOD})
@Retention(RetentionPolicy.SOURCE)
public @interface CheckWord {
}? 如果再编译之前的Student类，会发现用在"构造方法"上的@CheckWord会编译错误，因为我们规定了它只能用在"类","字段","方法"上。
2.3.定义注解的属性：
? 1.“注解”中可以定义一些属性，“注解解析器”可以根据“属性”的不同，分别做不同的事情。
? 例如@Target注解中的ElementType.TYPE就是此注解的一个属性，它是一个"枚举"类型。
? 下面让我们来看看怎样定义属性，然后再解析这些属性。
? 注解中定义属性的语法：数据类型 属性名() [deafult 值];
? 1.其中“数据类型”可以是：
? 1).所有基本类型;
? 2).String;
? 3).Class;
? 4).枚举;
? 5).注解；
? 6).以上任一类型的数组
? 2.属性名()：属性名可以自由设定，要遵循Java标识符的命名规则；其中的一对()是必须的。
? 3.[default 值]：为此属性设置的默认值。
? 2.本例中由于只检查大小写，为了规范取值，所以定义一个"枚举"类型的属性。
? 1).先定义枚举：
public enum StartsWith {
UPPERCASE, LOWERCASE
}? 此枚举定义了两个值：UPPERCASE表示：大写；LOWERCASE表示：小写。
? 2).修改"CheckWord"注解的代码：
@Target({ElementType.TYPE, ElementType.FIELD, ElementType.METHOD})
@Retention(RetentionPolicy.SOURCE)
public @interface CheckWord {
StartsWith value();
}? 说明：
? a.StartsWith表示"数据类型"，是一个"枚举"类型。
? b.value表示"属性名"，在使用此注解时，此属性的可取值只有StartsWith.UPPERCASE和StartsWith.LOWERCASE两个。
? c.此属性没有设置"默认值"，在使用此注解时必须要设置此属性的值。如下面的代码：
? 3).修改"Student"类的代码：
@CheckWord(StartsWith.UPPERCASE)
public class Student {
@CheckWord(StartsWith.LOWERCASE)
private String stuName;
@CheckWord(StartsWith.LOWERCASE)
public void show() {
}
}2.4注解解析器：
? 1."注解解析器"通常是随着注解一起定义的，用于解析"注解"，并做一些事情。本例的"注解解析器"用于与javac编译器一起启动，编译Student类时，检查各元素的名称是否按要求以指定的大写、小写字母开头。
? 2.自定义"注解解析器"需要继承AbstractProcessor类，并重写process()方法，完整的"注解解析器"代码如下：
import javax.annotation.processing.AbstractProcessor;
import javax.annotation.processing.RoundEnvironment;
import javax.annotation.processing.SupportedAnnotationTypes;
import javax.annotation.processing.SupportedSourceVersion;
import javax.lang.model.SourceVersion;
import javax.lang.model.element.Element;
import javax.lang.model.element.TypeElement;
import javax.tools.Diagnostic;
import java.util.Set;
@SupportedAnnotationTypes("CheckWord")
@SupportedSourceVersion(SourceVersion.RELEASE_8)
public class MyProcessor extends AbstractProcessor {
@Override
public boolean process(Set<? extends TypeElement> annotations, RoundEnvironment roundEnv) {
//获取所有使用了@CheckWord注解的元素
Set<? extends Element> annoEle = roundEnv.getElementsAnnotatedWith(CheckWord.class);
// 遍历这些元素
for (Element e : annoEle) {
//获取元素名称，可能是：类名、属性名、方法名
String name = e.getSimpleName().toString();
//获取这个名字的第一个字母
char c = name.charAt(0);
//获取这个元素上的@CheckWord注解对象
CheckWord anno = e.getAnnotation(CheckWord.class);
//获取这个注解的value属性的值，它是一个StartsWith枚举类型
StartsWith sw = anno.value();
//判断属性值是否设置为：StartsWith.UPPERCASE，但名字的首字母是小写
if (sw == StartsWith.UPPERCASE && Character.isLowerCase(c)) {
//向控制台打印异常信息
this.processingEnv.getMessager().printMessage(Diagnostic.Kind.ERROR, "名称：" + name + " 首字母应该大写！");
return false;
}
//判断属性值是否设置为：StartsWith.LOWERCASE，但名字的首字母是大写
if (sw == StartsWith.LOWERCASE && Character.isUpperCase(c)) {
this.processingEnv.getMessager().printMessage(Diagnostic.Kind.ERROR, "名称：" + name + " 首字母应该小写！");
return false;
}
}
return true;
}
}? 此代码的细节大家可以根据注释一点一点研究。一些类：TypeElement，RoundEnvironment，Element等的一些方法大家可以在API手册中查找。
? 其它说明：
? @SupportedAnnotationTypes("CheckWord") : 表示只处理CheckWord注解。
? @SupportedSourceVersion(SourceVersion.RELEASE_8) ： 表示支持JDK1.8。
2.5.编译和测试：
? 1.在编译前，我们看一下完整的代码清单：请确保以下的四个类在同一个目录下
? 1).枚举类：
public enum StartsWith {
UPPERCASE, LOWERCASE
}? 2).自定义注解类：
import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;
@Target({ElementType.TYPE, ElementType.FIELD, ElementType.METHOD})
@Retention(RetentionPolicy.SOURCE)
public @interface CheckWord {
StartsWith value();
}? 3).使用了CheckWord注解的Student类：
@CheckWord(StartsWith.UPPERCASE)
public class Student {
@CheckWord(StartsWith.LOWERCASE)
private String StuName;
@CheckWord(StartsWith.LOWERCASE)
public void show() {
}
}? 4).注解解析器类：
?
import javax.annotation.processing.AbstractProcessor;
import javax.annotation.processing.RoundEnvironment;
import javax.annotation.processing.SupportedAnnotationTypes;
import javax.annotation.processing.SupportedSourceVersion;
import javax.lang.model.SourceVersion;
import javax.lang.model.element.Element;
import javax.lang.model.element.ElementKind;
import javax.lang.model.element.TypeElement;
import javax.tools.Diagnostic;
import java.util.Set;
@SupportedAnnotationTypes("CheckWord")
@SupportedSourceVersion(SourceVersion.RELEASE_8)
public class MyProcessor extends AbstractProcessor {
@Override
public boolean process(Set<? extends TypeElement> annotations, RoundEnvironment roundEnv) {
Set<? extends Element> annoEle = roundEnv.getElementsAnnotatedWith(CheckWord.class);
for (Element e : annoEle) {
String name = e.getSimpleName().toString();
char c = name.charAt(0);
CheckWord anno = e.getAnnotation(CheckWord.class);
StartsWith sw = anno.value();
if (sw == StartsWith.UPPERCASE) {
if (Character.isLowerCase(c)) {
this.processingEnv.getMessager().printMessage(Diagnostic.Kind.ERROR, "名称：" + name + " 首字母应该大写！");
return false;
}
}
if (sw == StartsWith.LOWERCASE) {
if (Character.isUpperCase(c)) {
this.processingEnv.getMessager().printMessage(Diagnostic.Kind.ERROR, "名称：" + name + " 首字母应该小写！");
return false;
}
}
}
return true;
}
}? 2.启动命令行，使用javac依次进行编译：
? javac StartsWith.java
? javac CheckWord.java
? javac MyProcessor.java（如果报错： 编码GBK的不可映射字符，是因为代码中的中文，可以使用javac -encoding UTF-8 MyProcessor.java进行编译）
? 接下来使用MyProcessor解析器编译Student:
? javac -processor MyProcessor Student.java
? 执行命令后，会有错误提示：
? 错误: 名称：StuName 首字母应该小写！
? 1 个错误
三、总结：
? 源码级注解的应用非常广泛，例如：进行代码检查、生成新类、生成文件。本文实现了基本的代码检查，用于检查类中的元素是否按照要求进行首字母大写或者小写。也可以根据需要，验证是否全部大写，或者全部小写。希望大家通过本案例能够了解源码级注解的编写及使用。
