Javapoet源码解析

Javapoet:是生成.java源文件的开源API

 github：https://github.com/square/javapoet

 

 以生成一个HelloWrold.java文件为例：

package com.example.helloworld;

public final class HelloWorld {
  public static void main(String[] args) {
    System.out.println("Hello, JavaPoet!");
  }
}

MethodSpec main = MethodSpec.methodBuilder("main")
    .addModifiers(Modifier.PUBLIC, Modifier.STATIC)
    .returns(void.class)
    .addParameter(String[].class, "args")
    .addStatement("$T.out.println($S)", System.class, "Hello, JavaPoet!")
    .build();

TypeSpec helloWorld = TypeSpec.classBuilder("HelloWorld")
    .addModifiers(Modifier.PUBLIC, Modifier.FINAL)
    .addMethod(main)
    .build();

JavaFile javaFile = JavaFile.builder("com.example.helloworld", helloWorld)
    .build();

///javaFile.writeTo(System.out);  //System.out print  to the console
 //generate a java file in user.dir
String directory = System.getProperty("user.dir");
   File file = new File(directory + SEPARATOR + "HelloWrold.java");
PrintStream ps = new PrintStream(new FileOutputStream(file));
javaFile.writeTo(ps);
ps.flush();
ps.close();

在这个例子中用到了三个类的对象 ：JavaFile  MethodSpec   TypeSpec,首先来看第一个JavaFile

 public static Builder builder(String packageName, TypeSpec typeSpec) {
    checkNotNull(packageName, "packageName == null");
    checkNotNull(typeSpec, "typeSpec == null");
    return new Builder(packageName, typeSpec);
  }

Builder是JavaFile中的一个静态内部类，由此可以看到 Javapoet使用的是建造者设计模式。通过packageName 和TypeSpec生成一个Builder对象，然后调用Builder类的build方法：

public JavaFile build() {
      return new JavaFile(this);
    }

返回当前的JavaFile对象，然后调用writeto方法将该对象写入流/文件。

接下来看看构造Builder方法时使用的TypeSpec是什么。

staticBlock：A generated class, interface, or enum declaration：对于要生成的类，接口或枚举的声明。就是定义 类 接口或者枚举，定义的时候需要给他们命名，添加权限修饰符 是否静态 是否抽象 是否final，添加方法等操作

所以这个类里面有静态方法classBuilder interfaceBuilder enumBuilder 以及匿名内部类anonymousClassBuilder，这些静态方法都返回对应的一个Builder对象

该Builder对象是TypeSpec中的静态内部类，用于构建当前的TypeSpec对象。在内部类Builder中应该有添加权限修饰符/注释/注解/变量等操作来构建一个类或者接口枚举

MethodSpec：A generated constructor or method declaration. 对方法的声明，包括构造方法。所以这个类里面有methodBuilder constructorBuilder,返回该类的一个Builder对象，这个builder也是该类的一个静态内部类。并且这个Builder内部类中同样包括添加权限修饰符/注释/注解等方法，另外方法有返回值，参数 Comment 语句来构建一个方法

同样的原理，FieldSpec是对成员变量的声明，而且里面也有一个可以返回当前对象的builder静态内部类，在这个builder中同样有添加注释注解访问修饰符等的操作，来构建一个变量。

类变量等已经定义完毕，接下来我们分析一下是如何生成java文件的，回到JavaFile的writeTo方法

这里有很多重载的writeTo方法，最终都会调用

 public void writeTo(Appendable out) throws IOException {
    // First pass: emit the entire class, just to collect the types we'll need to import.
    CodeWriter importsCollector = new CodeWriter(NULL_APPENDABLE, indent, staticImports);
    emit(importsCollector);
    Map<String, ClassName> suggestedImports = importsCollector.suggestedImports();

    // Second pass: write the code, taking advantage of the imports.
    CodeWriter codeWriter = new CodeWriter(out, indent, suggestedImports, staticImports);
    emit(codeWriter);
  }

CodeWriter是什么呢？CodeWriter将JavaFile转换为适用于人类和javac使用的字符串，实现了导入，缩进和延期变量名称。

以FieldSpec为例来说明CodeWriter的用法：

void emit(CodeWriter codeWriter, Set<Modifier> implicitModifiers) throws IOException {
    codeWriter.emitJavadoc(javadoc);
    codeWriter.emitAnnotations(annotations, false);
    codeWriter.emitModifiers(modifiers, implicitModifiers);
    codeWriter.emit("$T $L", type, name);
    if (!initializer.isEmpty()) {
      codeWriter.emit(" = ");
      codeWriter.emit(initializer);
    }
    codeWriter.emit(";
");
  }

这段代码是在生成变量时调用的。第二行是写javadoc注释，第三行是写注解，第四行是写访问修饰符，第五行是写该变量的类型和变量名，第6-9行是判断变量是否有初始化值，如果有就写初始化值

第10行；代表该变量写出完毕。至此，HelloWorld.java的生成过程分析完毕。

用到的一些集合类中的方法：

Collections.singletonList

Collections.emptyList()

EnumSet.copyOf(Set<Modifier> modifiers)

Collections.unmodifiableList(new ArrayList<>(collection));   返回指定列表的不可修改视图。 此方法允许模块为用户提供对内部列表的“只读”访问权限。