手把手带你实现一个代码生成器

前言

不知各位看官在工作之中有没有陷入过疯狂CV代码、看着密密麻麻的类不想动手，或者把大把的时间花费在底层的情况。以笔者为例，会经常遇到以下两个问题：

• 隔一段时间就需要构建一个新应用，需要各种复制粘贴（缺乏定制化的脚手架）
• 新需求一堆的Entity、Bean、Request、Response、DTO、Dao、Service、Business需要写，看着都不想动手

很多时候甚至会在复制粘贴代码时漏掉一些关键的注解，比如：@Service，导致项目无法启动，再花费更多的精力去排查。因此本文将以实际工程代码为例，来构建一个可定制化，支持高度扩展的代码生成器。

项目目标

本项目将基于生成器项目生成一个可直接运行/可方便一键复制的SpringBoot成品项目，其中包括基本的数据库操作、业务操作及Web接口、视图层。

同时支持拔插式自定义实现，具备较高的拓展性，以下是项目基本结构：

Code-Generate├── pom.xml├── src│   ├── main│   │   ├── java│   │   │   └── 

编码

构建配置中心

由于本项目涉及数据库操作层等，因此除了目标目录，项目名，作者，根目录等基本参数外，还需要数据库相关配置等，配置文件如下：

# 数据库IP, 数据库Driver, 编码, 用户名, 密码ip=127.0.0.1port=3306driver=com.mysql.jdbc.DriverdataBase=school-miaoencoding=UTF-8loginName=rootpassWord=# 需要构建的表名  为* 默认包含全部, 以;号隔离比如:  a;b;c;d;include=*;# 项目名projectName=Demo# 包名packageName=com.demo# 作者authorName=Kerwin# 项目输出根目录rootPath=F:\\code# 自定义Handle包含项, 现有自定义模块：DataMdImpl，LayUiHtmlImpl, * 号默认包含所有，以;号隔离比如:  a;b;c;d;customHandleInclude=DataMdImpl;LayUiHtmlImpl;复制代码

考虑好基础的配置内容后，通过读取文件配置，将相关信息置入配置中心即可，具体代码便不再展示，需要的直接观看源码即可（文末链接）。

本阶段涉及的类：ConfigurationInfo.java、GlobleConfig.java、PropertiesFactory。

入口：
com.mysql.engine.AbstractEngine#init

基于数据库获取表字段信息

上文已获取到目标数据库的配置信息，本阶段即可连接数据库，通过通用SQL获取目标库的表、字段信息。

以下面的SQL为例，只需获取数据库连接加上库信息，即可获取所有的表名

SELECTtable_name FROMinformation_schema.TABLES WHEREtable_schema = "school-miao-demo"  # 库名AND table_type = "base table";# 响应# schools复制代码

同理，通过通用SQL也可以获取到指定数据表的所