JAVAPOET小试牛刀

惯例先放一下JAVAPOET的简介 源码在此

前几日其他项目突然出现了60多个表单的开发任务，开发任务之大令人发指，虽然手头项目暂时没有如此巨额的表单开发需求，但是为了防患于未然，加上最近确实没有了梳理业务的兴趣，就想弄一个代码生成器

翻遍了无数资料甚至还为此花了大洋去看所谓的名师教程（被骗，心痛），基本上能用的寥寥无几，而且生成的代码和项目架构不符，也没法应用。

找到一篇写的好详细的源码分析 链接

一咬牙 一跺脚，自己来！

放代码。

ublic class CodeCreate {

private String fileSrc;

public CodeCreate(){

this.fileSrc = "D:/TEMP/";

}

public CodeCreate(String fileSrc){

this.fileSrc = fileSrc;

}

/*

    * 生成所有层代码并输出到对应的文件位置中

    *

    */

    public void createAll(String tablename,String classname){

    try {

List tableInfoList =

ConnectDatabase.sysoutStrTablePdmCloumns(tablename);

    //生成实体测试

    createModel(tableInfoList,classname);

    //生成Dao层测试

    createDao(tablename,classname);

    //生成Service层测试

    createService(classname);

    //生成ServiceImpl层测试

    createServiceImpl(tableInfoList,classname);

    //生成Controller层测试

    createController(classname);

} catch (SQLException e) {

e.printStackTrace();

}

    }

    //生成model层代码

    private void createModel(List list,String className){

    Builder builder = TypeSpec.classBuilder(className);

    for (int i = 0; i < list.size(); i++) {

    TableInfo infotemp = list.get(i);

    String column = PoetUtils.getFieldName(infotemp.getColumnname());

    Class datatype = PoetUtils.getClassType(infotemp.getColumntype());

    builder = PoetUtils.addGetAndSet(builder,datatype,column);

}

    builder.addModifiers(Modifier.PUBLIC);

    TypeSpec relust = builder.build();

        JavaFile javaFile = JavaFile.builder("persistence.entity", relust).build();

        File outputFile = new File(fileSrc+className+"/"); //输出文件

        try {

            javaFile.writeTo(outputFile);

            javaFile.writeTo(System.out);

        } catch (IOException e) {

            // TODO Auto-generated catch block

            e.printStackTrace();

        }

    }

    /*

    * 生成dao层代码

    */

    private void createDao(String tablename,String className){

    Builder builder = TypeSpec.classBuilder(className+"Dao");

    //定义动态类型

    ClassName baseJdbcDao = ClassName.get("cn.jasgroup.jasframework.dataaccess.base", "BaseJdbcDao");

    ClassName bo = ClassName.get("business.bo", className+"Bo");

    ClassName entity = ClassName.get("persistence.entity", className);

    ClassName pageRequest = ClassName.get("com.jasgroup.platform.common.utils.page", "PageRequest");

    ClassName page = ClassName.get("com.jasgroup.platform.common.utils.page", "Page");

    ClassName abstractExcutiveDaoImpl = ClassName.get("com.jasgroup.platform.common.persistence.dao.impl", "AbstractExcutiveDaoImpl");

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

    //添加baseJdbcDao的get和set方法

    builder = PoetUtils.addGetAndSet(builder,baseJdbcDao,"baseJdbcDao");

builder.addModifiers(Modifier.PUBLIC);

//添加getAll方法

MethodSpec getAll = MethodSpec.methodBuilder("getAll")

          .addModifiers(Modifier.PUBLIC)

          .returns(ParameterizedTypeName.get(page, entity))

          .addParameter(bo,"bo")

          .addParameter(pageRequest,"pageRequest")

          .addStatement("$T hql = new $T($S)",StringBuffer.class,StringBuffer.class,"from "+className+" x where x.active = 1")

          .addStatement("return this.findPage(pageRequest, hql.toString(), new $T