1、什么是JavaPoet
JavaPort是square公司推出的開源java代碼生成框架,提供Java API生成.java源文件,這個框架功能非常實用,也是我們習慣的java面向對象OOP語法,可以很方便的使用它根據注解生成代碼,通過這種自動化生成代碼的方式,可以讓我們用更加簡潔優雅的方式替換繁瑣冗雜的重復工作。
項目主頁及源碼:https://github.com/square/javapoet
2、開發環境配置
3、JavaPoet8個常用的類
4、JavaPoet字符串格式化規則
4、JavaPoet的使用
這里我們借助APT的介紹與使用中的例子,使用javapoet生成我們想要的文件。
// 幫助我們通過類調用的形式來生成Java代碼
implementation "com.squareup:javapoet:1.9.0"
@AutoService(Processor.class)
@SupportedAnnotationTypes("com.migill.annotation.ARouter")
@SupportedSourceVersion(SourceVersion.RELEASE_7)
@SupportedOptions("content")
public class ARouterProcessorNew extends AbstractProcessor {
private Elements elementUtils;
private Types typeUtils;
private Messager messager;
private Filer filer;
@Override
public synchronized void init(ProcessingEnvironment processingEnvironment) {
super.init(processingEnvironment);
elementUtils = processingEnvironment.getElementUtils();
typeUtils = processingEnvironment.getTypeUtils();
messager = processingEnvironment.getMessager();
filer = processingEnvironment.getFiler();
String content = processingEnvironment.getOptions().get("content");
messager.printMessage(Diagnostic.Kind.NOTE, content);
}
@Override
public boolean process(Set<? extends TypeElement> set, RoundEnvironment roundEnvironment) {
if (set.isEmpty()) return false;
Set<? extends Element> elements = roundEnvironment.getElementsAnnotatedWith(ARouter.class);
for (Element element : elements) {
String packageName = elementUtils.getPackageOf(element).getQualifiedName().toString();
String className = element.getSimpleName().toString();
String finalClassName = className + "$ARouter";
messager.printMessage(Diagnostic.Kind.NOTE, "生成的文件名:" + finalClassName);
try {
// 高級寫法,javapoet構建工具
ARouter aRouter = element.getAnnotation(ARouter.class);
//構造方法體
MethodSpec method = MethodSpec.methodBuilder("findTargetClass")//方法名
.addModifiers(Modifier.PUBLIC, Modifier.STATIC)
.returns(Class.class)
.addParameter(String.class, "path")
.addStatement("return path.equals($S) ? $T.class : null", aRouter.path(), ClassName.get((TypeElement) element))
.build();
//構建類
TypeSpec typeSpec = TypeSpec.classBuilder(finalClassName)
.addModifiers(Modifier.PUBLIC)
.addMethod(method)//添加方法體
.build();
//在指定包名下,生成Java類文件
JavaFile javaFile = JavaFile.builder(packageName, typeSpec).build();
javaFile.writeTo(filer);
} catch (IOException e) {
e.printStackTrace();
}
}
return false;
}
}
現在編譯,就可以生成對應的文件了,如下圖,就是生成的文件與文件的內容