代码生成
About 452 wordsAbout 2 min
1984-01-24
BeetlSQL3重写了代码生成框架,使得代码生成更容易理解以及按照特定需求定制,可以直接查源码S7CodeGen了解基本用法
List<SourceBuilder> sourceBuilder = new ArrayList<>();
SourceBuilder entityBuilder = new EntitySourceBuilder();
SourceBuilder mapperBuilder = new MapperSourceBuilder();
SourceBuilder mdBuilder = new MDSourceBuilder();
sourceBuilder.add(entityBuilder);
sourceBuilder.add(mapperBuilder);
sourceBuilder.add(mdBuilder);
SourceConfig config = new SourceConfig(sqlManager,sourceBuilder);
//如果有错误,抛出异常而不是继续运行1
EntitySourceBuilder.getGroupTemplate().setErrorHandler(new ReThrowConsoleErrorHandler() );
ConsoleOnlyProject project = new ConsoleOnlyProject();
String tableName = "sys_user";
//可以在控制台看到生成的所有代码
config.gen(tableName,project);
SourceConfig是核心类,类通过SQLManager构造,另外一个参数是SourceBuilder列表。SourceBuilder代表了代码生成逻辑,比如生成POJO代码,生成Mapper代码,生成Service代码,甚至是数据库文档。BeetlSQL内置了如下生成
- EntitySourceBuilder 生成POJO代码
- MapperSourceBuilder 生成Mapper代码
- MDSourceBuilder 生成markdown文件
- MDDocBuilder,生成一个数据库表文档,markdown格式
你可以实现SourceBuilder,以生成特定的代码或者文档,或者其他类型文档。
配置好SourceConfig后,需要使用一个BaseProject的子类,告诉生成代码存放在哪里,BeetlSQL3提供如下Project
ConsoleOnlyProject , 仅仅输出到控制台,你可以把生成的代码复制到你需要的类里
SimpleMavenProject 当前工程是maven工程,输出成文件到maven工程里,不推荐,这样很容易覆盖以前生成好但有微调过的代码,除非你是第一次使用
StringOnlyProject 输出到字符串里
StringOnlyProject project = new StringOnlyProject(); String tableName = "sys_user"; config.gen(tableName,project); String content = project.getContent();
SourceConfig 类有很多配置选项,比如,使用BigDecimal还是Double,每个SourceBuilder实现也可以设置不同属性。比如,配置生成的java类的包名。可以阅读这俩个源码了解如生成配置