背景
关于为什么做这个代码生成器,其实主要有两点:
- 参与的项目中有很多分析报表需要展示给业务部门,公司使用的商用产品,或多或少有些问题,这部分可能是历史选型导致的,这里撇开不不谈;
- 项目里面也有很多CRUD的功能,而这些功能的实现代码基本上差不多,这些功能都去手写,也比较浪费时间而且效率很低,还可能会写错;
针对这两个问题,就思考做一个简单的半自动的代码生成器,帮助团队的提升开发效率,只要在数据库中设计好表结构,就可以一键生成前后端代码、Swagger接口文档 、Validator参数校验。
设计
此篇主要是聊聊关于单表CRUD代码生成器的实现,后续大家可以扩展到树表、主子表,甚至还可以接入一些AI来扩展一下。整体技术方案,后端采用Spring Boot、Freemarker、MybatisPlus、Swagger等,前端使用Vue3、Element Plus。看到这里我想很多后端就知道我是如何做的了,整体核心实现就是通过Freemarker模版,增加一些配置项,动态生成CRUD的代码。这里做了一些细化设计,首先来看下整体的一个界面效果。
表结构设计
核心设计不变,只是增加三张表,可以进行自定义配置,让整个流程更加丝滑,整体表结构设计如下:
三张表分别是数据源配置表(tool_data_source_config)、表定义(tool_generator_table)以及列定义(tool_generator_column),数据源配置表主要是连接数据库相关的配置,表定义主要是类的抽象,包括整个类的包结构目录,列定义主要类字段的定义,并且包含一些CRUD条件定义。通过这三张表的定义,就可以实现半自动化代码生成。
代码设计
整体设计分为两部分,一部分通过MyBatis Plus Generator读取数据源配置,获取对应表的TableInfo信息,最终映射到表定义和列定义的表中,另外一部分就是通过Freemarker定义模板生成对应前端和后端代码。这里需要注意的是每个表名定义都是模块名加自身的业务抽象。
给大家看下核心代码,以下代码是通过MyBatis Plus Generator获取TableInfo,TableInfo包含数据库表的定义,后需要主要使用tableName、columns、tableComment等相关的属性,转换为tool_generator_table和tool_generator_column表相关的内容;
java
private List<TableInfo> getAllTable(Long dataSourceConfigId, String tableName) {
// 获得数据源配置
ToolDataSourceConfig config =
dataSourceConfigService.getDataSourceConfig(dataSourceConfigId);
// 使用 MyBatis Plus Generator 解析表结构
DataSourceConfig dataSourceConfig = new DataSourceConfig.Builder(config.getUrl(), config.getUsername(),
config.getPassword()).build();
StrategyConfig.Builder strategyConfig = new StrategyConfig.Builder();
if (StringUtils.isNotEmpty(tableName)) {
strategyConfig.addInclude(tableName);
}
GlobalConfig globalConfig = new GlobalConfig.Builder().dateType(DateType.TIME_PACK).build();
ConfigBuilder builder = new ConfigBuilder(null, dataSourceConfig, strategyConfig.build(),
null, globalConfig, null);
List<TableInfo> tables = builder.getTableInfoList();
tables.sort(Comparator.comparing(TableInfo::getName));
return tables;
}
以下代码是通过Freemarker生成对应的模版,具体的细节部分,可以在GitHub上查看;
java
public Map<String, String> execute(ToolGeneratorTable table, List<ToolGeneratorColumn> columns) {
//初始化
Map<String, Object> bindingMap = initBindingMap(table, columns);
//获取对应模版
Map<String, String> templates = getTemplates();
Map<String, String> result = Maps.newLinkedHashMapWithExpectedSize(templates.size());
templates.forEach((key, value) -> {
generateCode(result, key, value, bindingMap);
});
return result;
}
private void generateCode(Map<String, String> result, String vmPath, String filePath, Map<String, Object> bindingMap) {
filePath = formatFilePath(filePath, bindingMap);
try {
StringWriter writer = new StringWriter();
Template template = configuration.getTemplate(vmPath);
template.process(bindingMap, writer);
result.put(filePath, writer.toString());
} catch (Exception exception) {
log.error(vmPath + "模版加载异常" + exception);
}
}
结束
GIt地址 欢迎大家Star,下图截图是核心代码的重要实现的类。