javapoet-android generate Builder

javapoet-android#View on GitHub

JavaPoet generate Android common codes

Source code

@PoetCode
public class Person {
    private String name;
    private int age;
}

Generate Builder

1.Builder

        public static void createBuilder(String pkgName, String newClazzName, TypeSpec.Builder typeSpecBuilder, HashMap<String, TypeName> fieldMap) {
            ClassName newClassName = PoetCodeUtil.createNewClazzType(pkgName, newClazzName);
            ClassName builderClassName = PoetCodeUtil.createNewClazzType(pkgName, newClazzName, "Builder");
            TypeSpec.Builder typeSpecB = TypeSpec.classBuilder("Builder")
                    .addModifiers(Modifier.PUBLIC, Modifier.STATIC);
            for (String field : fieldMap.keySet()) {
                typeSpecB.addField(fieldMap.get(field), field, Modifier.PRIVATE)
                        .addMethod(PoetCodeUtil.createSetReturn(builderClassName, field, fieldMap.get(field), Modifier.PUBLIC));
            }
            MethodSpec.Builder methodSpecB = MethodSpec.methodBuilder("create")
                    .addModifiers(Modifier.PUBLIC)
                    .returns(newClassName)
                    .addStatement("$T outClazz = new $T()", newClassName, newClassName);
            for (String field : fieldMap.keySet()) {
                methodSpecB.addStatement("outClazz.$L = $L", field, field);
            }
            methodSpecB.addStatement("return outClazz");
            typeSpecB.addMethod(methodSpecB.build());
            typeSpecBuilder.addType(typeSpecB.build());
        }
        --------------------------------------------------------------------------------
  public static class Builder {
    private String name;

    private int age;

    public Builder setName(String name) {
      this.name = name;
      return this;
    }

    public Builder setAge(int age) {
      this.age = age;
      return this;
    }

    public PersonBuilderPattern create() {
      PersonBuilderPattern outClazz = new PersonBuilderPattern();
      outClazz.name = name;
      outClazz.age = age;
      return outClazz;
    }
  }

猜你喜欢

转载自blog.csdn.net/wcs542882916/article/details/82959710