JavaPoet 基本用法

JavaPeot笔记
https://www.jianshu.com/p/d6dc11a816d2
Android中，怎么优雅的生成代码？
https://www.jianshu.com/p/4701538edd21?utm_campaign=maleskine&utm_content=note&utm_medium=seo_notes&utm_source=recommendation
javapoet项目github地址
https://github.com/square/javapoet

MethodSpec 方法构造

使用示例：

MethodSpec.Builder builder = MethodSpec.methodBuilder("get"+typeElementClassName.simpleName()+"Bind")
        // 指定方法修饰符
        .addModifiers(Modifier.PUBLIC)
        //静态方法
        .addModifiers(Modifier.STATIC)
        // 指定返回类型
        .returns(listEventMethodInfo);

生成示例：

public static ArrayList getLoginActivityBind() {
  ArrayList eventMethodInfoList = new ArrayList<>();
  eventMethodInfoList.add(new EventInfo("com.pf.eb3_apt_app.ui.LoginActivity", "message1", "com.pf.eb3_apt_app.bean.EventMessage"));
  eventMethodInfoList.add(new EventInfo("com.pf.eb3_apt_app.ui.LoginActivity", "message2", "com.pf.eb3_apt_app.bean.EventMessage"));
  eventMethodInfoList.add(new EventInfo("com.pf.eb3_apt_app.ui.LoginActivity", "message3", "com.pf.eb3_apt_app.bean.EventMessage"));
  return eventMethodInfoList;
}

returns 返回值
addStatement 添加语句
beginControlFlow 控制语句（if、while、for）
endControlFlow 结束控制语句

addStatement占位符
占位符作用备注
$L 万用占位符
$S 字符串占位符
$T内置类型映射支持，会自动 import 声明可以更方便的结合ClassName使用
$N 当引用一个通过名字生成的声明时，可以用当前这个占位符进行占位

控制流（if、while、for）

MethodSpec main = MethodSpec.methodBuilder("main")
    .addCode(""
        + "int total = 0;\n"
        + "for (int i = 0; i < 10; i++) {\n"
        + "  total += i;\n"
        + "}\n")
    .build();

生成如下代码

void main() {
  int total = 0;
  for (int i = 0; i < 10; i++) {
    total += i;
  }
}

MethodSpec提供beginControlFlow和endControlFlow控制流程事务。

MethodSpec main = MethodSpec.methodBuilder("main")
    .addStatement("int total = 0")
    .beginControlFlow("for (int i = 0; i < 10; i++)")
    .addStatement("total += i")
    .endControlFlow()
    .build();

TypeSpec 类、接口、枚举构造

使用示例：

TypeSpec typeElementSimpleNameBindingView = TypeSpec.classBuilder(typeElementClassName.simpleName()+"TcEventBusManager")
        // 指定类修饰符
        .addModifiers(Modifier.PUBLIC)
        // 添加方法
        .addMethod(main)
        //添加静态代码块
        .addStaticBlock(cBuilder)
        .build();

生成示例：

public class TcEventBusIndex {
  public static HashMap> cacheMap = new HashMap<>();

  public static List get(String obj) {
    return cacheMap.get(obj);
  }
}

FieldSpec 成员变量、字段构造

使用示例：

ClassName type = ClassName.get("com.pf.eb3_apt_bean", "EventInfo");
ClassName arrayList = ClassName.get("java.util", "ArrayList");
TypeName listType = ParameterizedTypeName.get(arrayList, type);

ClassName string = ClassName.get("java.lang", "String");

ClassName hashMap = ClassName.get("java.util", "HashMap");
TypeName mapType = ParameterizedTypeName.get(hashMap, string, listType);

FieldSpec fieldSpec = FieldSpec.builder(mapType, "cacheMap", Modifier.PUBLIC, Modifier.STATIC)
        .initializer("new HashMap<>()").build();

生成示例：

public static HashMap> cacheMap = new HashMap<>();

List

使用代码：

package com.example;

import com.example.all_product.ForInteger;
import com.squareup.javapoet.ClassName;
import com.squareup.javapoet.JavaFile;
import com.squareup.javapoet.MethodSpec;
import com.squareup.javapoet.ParameterizedTypeName;
import com.squareup.javapoet.TypeName;
import com.squareup.javapoet.TypeSpec;

import java.io.File;

import javax.lang.model.element.Modifier;

/**
 * 使用ClassName(重要)，生成List代码 2017/4/24 16:04
 */
public class PoetList {

    /**
     * 生成器 2017/4/24 16:04
     */
    public static void product() {
        try {

            // list泛型的类型 2017/4/24 16:06
            ClassName type = ClassName.get("com.example.all_product", "ForInteger");

            // List类型
            ClassName list = ClassName.get("java.util", "List");

            // ArrayList类型
            ClassName arrayList = ClassName.get("java.util", "ArrayList");

            // 生成泛型类型，类型的名称、参数的名称 2017/4/24 16:08
            TypeName listType = ParameterizedTypeName.get(list, type);

            MethodSpec methodSpec = MethodSpec.methodBuilder("listType")
                    .addModifiers(Modifier.PUBLIC, Modifier.STATIC)
                    .returns(listType)
                    .addStatement("$T lstHello = new $T<>()", listType, arrayList)

                    // ClassName类的绝对路径，内部利用该路径进行反射获取实体类，使用场景是当该类也是生成类时，可以硬指定 2017/4/24 16:18
                    .addStatement("lstHello.add(new $T())", type) // ClassName指定
                    .addStatement("lstHello.add(new $T())", ForInteger.class) // 引用指定
                    .addStatement("return lstHello")
                    .build();

            TypeSpec typeSpec = TypeSpec.classBuilder("ListType")
                    .addModifiers(Modifier.PUBLIC)
                    .addMethod(methodSpec)
                    .build();

            JavaFile javaFile = JavaFile.builder("com.example.all_product", typeSpec)
                    .build();

            javaFile.writeTo(new File("poet/src/main/java"));
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    public static void main(String[] args) {
        PoetList.product();
    }

}

生成代码：

package com.example.all_product;

import java.util.ArrayList;
import java.util.List;

public class ListType {
  public static List listType() {
    List lstHello = new ArrayList<>();
    lstHello.add(new ForInteger());
    lstHello.add(new ForInteger());
    return lstHello;
  }
}

Map

使用代码：

package com.example;

import com.squareup.javapoet.ClassName;
import com.squareup.javapoet.JavaFile;
import com.squareup.javapoet.MethodSpec;
import com.squareup.javapoet.ParameterizedTypeName;
import com.squareup.javapoet.TypeName;
import com.squareup.javapoet.TypeSpec;

import java.io.File;

import javax.lang.model.element.Modifier;

/**
 * 使用ClassName(重要)，生成Map代码 2017/4/25 13:39
 */
public class PoetMap {

    /**
     * 生成器 2017/4/25 13:39
     */
    public static void product() {
        try {

            // list泛型的类型 2017/4/24 16:06
            ClassName type = ClassName.get("com.example.all_product", "ForInteger");

            // key 2017/4/25 13:38
            ClassName string = ClassName.get("java.land", "String");

            // map类型
            ClassName map = ClassName.get("java.util", "Map");

            // HashMap类型
            ClassName hashMap = ClassName.get("java.util", "HashMap");

            // 生成Map类型，类型的名称、Key、Value 2017/4/25 14:08
            TypeName listType = ParameterizedTypeName.get(map, string, type);

            MethodSpec methodSpec = MethodSpec.methodBuilder("listType")
                    .addModifiers(Modifier.PUBLIC, Modifier.STATIC)
                    .returns(listType)
                    .addStatement("$T mapHello = new $T<>()", listType, hashMap)
                    .addStatement("mapHello.put($S, new $T())", "key", type)
                    .addStatement("return mapHello")
                    .build();

            TypeSpec typeSpec = TypeSpec.classBuilder("MapType")
                    .addModifiers(Modifier.PUBLIC)
                    .addMethod(methodSpec)
                    .build();

            JavaFile javaFile = JavaFile.builder("com.example.all_product", typeSpec)
                    .build();

            javaFile.writeTo(new File("poet/src/main/java"));
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    public static void main(String[] args) {
        PoetMap.product();
    }

}

生成代码：

package com.example.all_product;

import java.util.HashMap;
import java.util.Map;

public class MapType {
  public static Map listType() {
    Map mapHello = new HashMap<>();
    mapHello.put("key", new ForInteger());
    return mapHello;
  }
}

JavaFile 顶级Java文件

使用示例：

//创建类
TypeSpec typeElementSimpleNameBindingView = TypeSpec.classBuilder("TcEventBusIndex")
        // 指定类修饰符
        .addModifiers(Modifier.PUBLIC)
        //添加变量
        .addField(fieldSpec)
        // 添加方法
        .addMethod(main)
        .build();

JavaFile javaFile = JavaFile.builder("com.pf.eb3_apt_app", typeElementSimpleNameBindingView).build();
    try {
            // 创建文件
            javaFile.writeTo(filer);
            } catch (IOException e) {
            e.printStackTrace();
            }

生成示例：

public class TcEventBusIndex {
  public static HashMap> cacheMap = new HashMap<>();

  public static List get(String obj) {
    return cacheMap.get(obj);
  }
}

ParameterSpec 参数构造

暂未使用过

AnnotationSpec 注解构造

暂未使用过

文章待优化

JavaPoet 基本用法

相关文章：

MethodSpec 方法构造

TypeSpec 类、接口、枚举构造

FieldSpec 成员变量、字段构造

JavaFile 顶级Java文件

ParameterSpec 参数构造

AnnotationSpec 注解构造

你可能感兴趣的:(JavaPoet 基本用法)