【本文版权归微信公众号 ” 代码艺术 ”（ID：onblog）所有，若是转载请务必保留本段原创声明，违者必究。若是文章有不足之处，欢迎关注微信公众号私信与我进行交流！】

昨天无意中，逛到了 lombok 的网站，并看到了首页的 5 分钟视频，视频中的作者只是在实体类中写了几个字段，就可以自动编译为含 setter、getter、toString()等方法的 class 文件。看着挺新奇的，于是自己研究了一下原理，整理下发出来。

1. 何处下手

视频中作者的流程为：

(1)编写 Java 文件，在类上写 @Data 注解

@Data
public class Demo {
    private String name;
    private double abc;
}

(2)javac 编译，lombok.jar 是 lombok 的 jar 包。

javac -cp lombok.jar Demo.java

(3)javap 查看 Demo.class 类文件

javap Demo

Demo.class：

public class Demo {public Demo();
  public java.lang.String getName();
  public void setName(java.lang.String);
  public double getAbc();
  public void setAbc(double);
}

可以看到 Demo.class 内部竟然多了很多未定义的 setter、getter 方法，而视频作者主要使用的就是注解 + 编译，那么我们就从这方面入手。

2. 必备知识

2.1 注解

注解，相信大部分人都用过，不少人还会自定义注解，并会利用反射等搞点小东西。但本文所讲的并非是利用注解加反射在运行期自定义行为，而是在编译期。

自定义注解离不开四大元注解。

@Retention：注解保留时期

@Target：用于标记可以应用于哪些类型上

@Documented：作用是能够将注解中的元素包含到 Javadoc 中

@Inherited：继承。假设注解 A 使用了此注解，那么类 B 使用了注解 A，类 C 继承了类 B，那么类 C 也使用了注解 A。（这里的 使用 是为了区分易理解，实际为 被注解）

2.1 注解处理器

注解处理器就是 javac 包中专门用来处理注解的工具。所有的注解处理器都必须继承抽象类 AbstractProcessor 然后重写它的几个方法。

注解处理器是运行在它自己的 JVM 中。javac 启动一个完整 Java 虚拟机来运行注解处理器，这意味着你可以使用任何你在其他 java 应用中使用的的东西。其中抽象方法 process 是必须要重写的，再该方法中注解处理器可以遍历所有的源文件，然后通过 RoundEnvironment 类获取我们需要处理的注解所标注的所有的元素，这里的元素可以代表包，类，接口，方法，属性等。再处理的过程中可以利用特定的工具类自动生成特定的.java 文件或者.class 文件，来帮助我们处理自定义注解。

一个普通的注解处理器文件如下：

package com.example;

import java.util.LinkedHashSet;
import java.util.Set;
import javax.annotation.processing.AbstractProcessor;
import javax.annotation.processing.ProcessingEnvironment;
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.TypeElement;

public class MyProcessor extends AbstractProcessor {

    @Override
    public boolean process(Set<? extends TypeElement> annoations,
            RoundEnvironment env) {return false;}

    @Override
    public Set<String> getSupportedAnnotationTypes() {Set<String> annotataions = new LinkedHashSet<String>();
        annotataions.add("com.example.MyAnnotation");
        return annotataions;
    }

    @Override
    public SourceVersion getSupportedSourceVersion() {return SourceVersion.latestSupported();
    }

    @Override
    public synchronized void init(ProcessingEnvironment processingEnv) {super.init(processingEnv);
    }

}

• init(ProcessingEnvironment processingEnv)：所有的注解处理器类都必须有一个无参构造函数。然而，有一个特殊的方法 init()，它会被注解处理工具调用，以 ProcessingEnvironment 作为参数。ProcessingEnvironment 提供了一些实用的工具类 Elements, Types 和 Filer。
• process(Set<? extends TypeElement> annoations, RoundEnvironment env)：这类似于每个处理器的 main()方法。你可以在这个方法里面编码实现扫描，处理注解，生成 java 文件。使用 RoundEnvironment 参数，你可以查询被特定注解标注的元素。
• getSupportedAnnotationTypes()：在这个方法里面你必须指定哪些注解应该被注解处理器注册。注意，它的返回值是一个 String 集合，包含了你的注解处理器想要处理的注解类型的全称。换句话说，你在这里定义你的注解处理器要处理哪些注解。
• getSupportedSourceVersion()：用来指定你使用的 java 版本。通常你应该返回SourceVersion.latestSupported()。不过，如果你有足够的理由坚持用 java 6 的话，你也可以返回SourceVersion.RELEASE_6。

关于 getSupportedAnnotationTypes() 和getSupportedSourceVersion()这两个方法，你也可以使用相应注解进行代替。代码如下：

@SupportedSourceVersion(SourceVersion.RELEASE_8)
@SupportedAnnotationTypes("com.example.MyAnnotation")
public class MyProcessor extends AbstractProcessor {....

不过为了兼容 Java6，最好是重载这俩方法。

3. 开始编码

知识我们已经学会，现在开始实战。

3.1 自定义注解

@Retention(RetentionPolicy.CLASS)
@Target(ElementType.TYPE)
public @interface Data {}

3.2 自定义注解处理器

public class DataAnnotationProcessor extends AbstractProcessor {
    private Messager messager; // 用于打印日志
    private Elements elementUtils; // 用于处理元素
    private Types typeUtils;
    private Filer filer;  // 用来创建 java 文件或者 class 文件

    @Override
    public synchronized void init(ProcessingEnvironment processingEnv) {super.init(processingEnv);
        messager = processingEnv.getMessager();
        elementUtils = processingEnv.getElementUtils();
        filer = processingEnv.getFiler();
        typeUtils = processingEnvironment.getTypeUtils();}
    
    @Override
    public SourceVersion getSupportedSourceVersion() {return SourceVersion.latestSupported();
    }

    @Override
    public Set<String> getSupportedAnnotationTypes(){Set<String> set = new HashSet<>();
        set.add(Data.class.getCanonicalName());
        return Collections.unmodifiableSet(set);
    }

    @Override
    public boolean process(Set<? extends TypeElement> annotations, RoundEnvironment roundEnv) {messager.printMessage(Diagnostic.Kind.NOTE,"----- 开始自动生成源代码");
        try {
            // 标识符
            boolean isClass = false;
            // 类的全限定名
            String classAllName = null;
            // 返回被注释的节点
            Set<? extends Element> elements = roundEnv.getElementsAnnotatedWith(Data.class);
            Element element = null;
            for (Element e : elements) {
                // 如果注释在类上
                if (e.getKind() == ElementKind.CLASS && e instanceof TypeElement) {TypeElement t = (TypeElement) e;
                    isClass = true;
                    classAllName = t.getQualifiedName().toString();
                    element = t;
                    break;
                }
            }
            // 未在类上使用注释则直接返回，返回 false 停止编译
            if (!isClass) {return true;}
            // 返回类内的所有节点
            List<? extends Element> enclosedElements = element.getEnclosedElements();
            // 保存字段的集合
            Map<TypeMirror, Name> fieldMap = new HashMap<>();
            for (Element ele : enclosedElements) {if (ele.getKind() == ElementKind.FIELD) {
                    // 字段的类型
                    TypeMirror typeMirror = ele.asType();
                    // 字段的名称
                    Name simpleName = ele.getSimpleName();
                    fieldMap.put(typeMirror, simpleName);
                }
            }
            // 生成一个 Java 源文件
            JavaFileObject sourceFile = filer.createSourceFile(getClassName(classAllName));
            // 写入代码
            createSourceFile(classAllName, fieldMap, sourceFile.openWriter());
            // 手动编译
            compile(sourceFile.toUri().getPath());
        } catch (IOException e) {messager.printMessage(Diagnostic.Kind.ERROR,e.getMessage());
        }
        messager.printMessage(Diagnostic.Kind.NOTE,"----- 完成自动生成源代码");
        return true;
    }

    private void createSourceFile(String className, Map<TypeMirror, Name> fieldMap, Writer writer) throws IOException {
        // 生成源代码
        JavaWriter jw = new JavaWriter(writer);
        jw.emitPackage(getPackage(className));
        jw.beginType(getClassName(className), "class", EnumSet.of(Modifier.PUBLIC));
        for (Map.Entry<TypeMirror, Name> map : fieldMap.entrySet()) {String type = map.getKey().toString();
            String name = map.getValue().toString();
            // 字段
            jw.emitField(type, name, EnumSet.of(Modifier.PRIVATE));
        }
        for (Map.Entry<TypeMirror, Name> map : fieldMap.entrySet()) {String type = map.getKey().toString();
            String name = map.getValue().toString();
            //getter
            jw.beginMethod(type, "get" + humpString(name), EnumSet.of(Modifier.PUBLIC))
                    .emitStatement("return" + name)
                    .endMethod();
            //setter
            jw.beginMethod("void", "set" + humpString(name), EnumSet.of(Modifier.PUBLIC), type, "arg")
                    .emitStatement("this." + name + "= arg")
                    .endMethod();}
        jw.endType().close();
    }

    /**
     * 编译文件
     * @param path
     * @throws IOException
     */
    private void compile(String path) throws IOException {
        // 拿到编译器
        JavaCompiler complier = ToolProvider.getSystemJavaCompiler();
        // 文件管理者
        StandardJavaFileManager fileMgr =
                complier.getStandardFileManager(null, null, null);
        // 获取文件
        Iterable units = fileMgr.getJavaFileObjects(path);
        // 编译任务
        JavaCompiler.CompilationTask t = complier.getTask(null, fileMgr, null, null, null, units);
        // 进行编译
        t.call();
        fileMgr.close();}

    /**
     * 驼峰命名
     *
     * @param name
     * @return
     */
    private String humpString(String name) {
        String result = name;
        if (name.length() == 1) {result = name.toUpperCase();
        }
        if (name.length() > 1) {result = name.substring(0, 1).toUpperCase() + name.substring(1);
        }
        return result;
    }

    /**
     * 读取类名
     * @param name
     * @return
     */
    private String getClassName(String name) {
        String result = name;
        if (name.contains(".")) {result = name.substring(name.lastIndexOf(".") + 1);
        }
        return result;
    }

    /**
     * 读取包名
     * @param name
     * @return
     */
    private String getPackage(String name) {
        String result = name;
        if (name.contains(".")) {result = name.substring(0, name.lastIndexOf("."));
        }else {result = "";}
        return result;
    }
}

在自定义注解处理器中，注释非常详细的说明了每一步的思路，首先是读取被注释的节点，判断是否是类节点，然后生成 Java 源文件，并使用 javawriter 框架写入 Java 代码，最后手动编译该 java 源文件。

javawriter 框架引用如下：

compile 'com.squareup:javawriter:2.5.1'

3.3 注解处理器的注册

【本文版权归微信公众号 ” 代码艺术 ”（ID：onblog）所有，若是转载请务必保留本段原创声明，违者必究。若是文章有不足之处，欢迎关注微信公众号私信与我进行交流！】

编码结束后，还需要把注解处理器注册到 javac 编译器，所以需要提供一个 .jar 文件。就像其他 .jar 文件一样，你将你已经编译好的注解处理器打包到此文件中。并且，在你的 .jar 文件中，你必须打包一个特殊的文件 javax.annotation.processing.Processor 到 META-INF/services 目录下。因此你的 .jar 文件目录结构看起来就你这样：

MyProcess.jar
    -com
        -example
            -MyProcess.class
    -META-INF
        -services
            -javax.annotation.processing.Processor

javax.annotation.processing.Processor 文件的内容是一个列表，每一行是一个注解处理器的全称。例如：

com.example.MyProcess

在 IDE 中，只需在 resources 目录下新建 META-INF/services/javax.annotation.processing.Processor 文件即可。

其它注册方式

前面的注册方式很底层，个人推荐使用。当处理的注解处理器过多时，这种方式不免过于繁琐，所以另一种方式就是使用自动注册注解处理器的框架。

添加对谷歌自动注册注解库的引用

implementation‘com.google.auto.service:auto-service:1.0-rc4’

在注解处理器类前面声明

@AutoService(Processor.class)

4. 打包使用

此时我们把项目打包为 jar 包即可使用，下面演示下使用过程。

(1)写个 Demo.java

import cn.zyzpp.annotation.Data;

@Data
public class Demo {
    private String name;
    private double abc;
}

(2)编译 java 文件，在该 Demo.java 文件夹下打开控制台窗口，记得把打包的 jar 包一起放在此目录。

javac -cp annotation-1.0-SNAPSHOT.jar Demo.java

(3)使用 javap 查看编译后的 Demo.class

Compiled from "Demo.java"
public class Demo {public Demo();
  public double getAbc();
  public void setAbc(double);
  public java.lang.String getName();
  public void setName(java.lang.String);
}

再看此时的 Demo.java 代码

public class Demo {
  private double abc;
  private String name;
  public double getAbc() {return abc;}
  public void setAbc(double arg) {this.abc = arg;}
  public String getName() {return name;}
  public void setName(String arg) {this.name = arg;}
}

到此，我们正式开发出了自动生成 getter、setter 方法的插件。有的小伙伴可能觉得这个并没有多大作用，用 IDE 快捷键就能非常轻易的办到。其实，知识已经学会，能做出什么多姿多彩的框架就要靠小伙伴们的智慧了。比如，我们一般都会新建 Entity 类，然后基于此新建 Dao 层，Service 层代码，用本文所述知识足可以打造一款适合自己的代码生成器，节约时间，提高开发效率。

附赠

IntelliJ IDEA lombok 插件的安装和使用

补充

处理器最好作为 Jar 包依赖，打包失败可以指定版本。

<plugins>
<plugin>
<groupId>org.apache.maven.plugins</groupId>
<artifactId>maven-compiler-plugin</artifactId>
<version>2.0.2</version>
<executions>
<execution>
<id>default-compile</id>
<configuration>
<compilerArgument>-proc:none</compilerArgument>
<source>1.8</source>
<target>1.8</target>
</configuration>
</execution>
<execution>
<id>default-testCompile</id>
<configuration>
<source>1.8</source>
<target>1.8</target>
</configuration>
</execution>
</executions>
</plugin>
</plugins>

版权声明

【本文版权归微信公众号 ” 代码艺术 ”（ID：onblog）所有，若是转载请务必保留本段原创声明，违者必究。若是文章有不足之处，欢迎关注微信公众号私信与我进行交流！】