代码生成器

    1. public static void main(String[] args) {
    2. //配置数据量的连接信息,可以通过 JbootApplication.setBootArg 来配置
    3. //也可以在 jboot.properties 里配置
    4. JbootApplication.setBootArg("jboot.datasource.url", "jdbc:mysql://127.0.0.1:3306/jbootdemo");
    5. JbootApplication.setBootArg("jboot.datasource.user", "root");
    6. JbootApplication.setBootArg("jboot.datasource.password", "123456");
    7. String modelPackage = "io.jboot.test.codegen.model"; //生成的Model的包名
    8. String baseModelPackage = "io.jboot.test.codegen.modelbase"; //生成的BaseModel的包名
    9. //Model存放的路径,一般情况下是 /src/main/java 下,如下是放在 test 目录下
    10. String modelDir = PathKit.getWebRootPath() + "/src/test/java/" + modelPackage.replace(".", "/");
    11. String baseModelDir = PathKit.getWebRootPath() + "/src/test/java/" + baseModelPackage.replace(".", "/");
    12. System.out.println("start generate...");
    13. System.out.println("generate dir:" + modelDir);
    14. //开始生成 Model 和 BaseModel 的代码
    15. new JbootBaseModelGenerator(baseModelPackage, baseModelDir).setGenerateRemarks(true).generate();
    16. new JbootModelGenerator(modelPackage, baseModelPackage, modelDir).generate();
    17. //设置 service 层代码的存放目录
    18. String serviceOutputDir = PathKit.getWebRootPath() + "/src/test/java/" + servicePackage.replace(".", "/");
    19. String serviceImplOutputDir = PathKit.getWebRootPath() + "/src/test/java/" + serviceImplPackage.replace(".", "/");
    20. //开始生成代码
    21. new JbootServiceInterfaceGenerator(servicePackage, serviceOutputDir, modelPackage).generate();
    22. new JbootServiceImplGenerator(servicePackage, serviceImplOutputDir, modelPackage).setImplName("provider").generate();
    23. }