本篇文章将会讲解ruoyi-vue系统下代码生成器的使用、原理分析以及将这部分代码抽离出来形成独立版的代码生成器。
四、代码生成器高级使用
在三的测试过程中,我们发现,商品是否上架这个字段对应的前端表单是个文本输入框,但是实际上这里应该是个下拉列表框或者单选按钮才对,因为它只有两个值:0或者1,ruoyi-vue代码生成器实际上是支持这种操作的。之前生成代码的时候导入表之后直接点击了下载按钮生成了代码,实际上跳过了一个重要的步骤,那就是编辑代码生成选项。
点击编辑按钮之后,跳转修改生成配置页面。
在这个页面中,可以修改字段在前端的显示类型以及字典类型,比如,要想“商品是否上架”在前端展示为单选框,就可以修改显示类型为“单选框”,字典类型设置为“业务是否(
需要新增数据字典
,0表示下架,1表示上架)”。
提交之后重新生成的代码样式:
可以看到,商品是否字段变成了下拉列表和单选框的样式。
另外,若依代码生成器支持三种数据格式模板的代码生成:单表、树表、主子表,这里默认使用的是单表模板,也是最常使用的模板。
五、代码生成器原理
1.Velocity
Velocity是一个基于Java的模板引擎,其提供了一个Context容器,在java代码里面我们可以往容器中存值,然后在vm文件中使用特定的语法获取,这是velocity基本的用法,其与jsp、freemarker并称为三大视图展现技术。作为一个模块引擎,除了作为前后端分离的MVC展现层,Velocity还有一些其他用途,比如源代码生成。
在若依Vue系统中,正是使用了Velocity技术实现的源代码生成。大体上,源代码生成只需三步走:
-
创建模板文件
-
准备上下文(变量值)
-
替换模板文件中的变量
三步走完之后源代码就生成了,说起来是很简单的,但是实际上做起来会比较麻烦,特别是第一步创建模板文件是最复杂的,以下为index.vue模板文件部分源代码:
#foreach($column in $columns)
#if($column.query)
#set($dictType=$column.dictType)
#set($AttrName=$column.javaField.substring(0,1).toUpperCase() + ${column.javaField.substring(1)})
#set($parentheseIndex=$column.columnComment.indexOf("("))
#if($parentheseIndex != -1)
#set($comment=$column.columnComment.substring(0, $parentheseIndex))
#else
#set($comment=$column.columnComment)
#end
#if($column.htmlType == "input")
...
可以看到,该vue模板文件中充斥着大量Velocity的if-else语法,嵌套在一起更是显得无比复杂。
总之,整体上来看,java代码的模板比较简单,vue和mybatis mapper的模板文件比较复杂。
2.information_schema 数据库
mysql数据库中有一个information_schema数据库,它是mysql的系统数据库之一,它里面存储着两个表
TABLES
以及
COLUMNS
,这两个表分别存储着所有的表信息以及所有表中的列信息,代码生成器正是以两张表的信息为核心实现的。
以goods表为例,
TABLES
表中的记录为
TABLE_CATALOG
|
TABLE_SCHEMA
|
TABLE_NAME
|
TABLE_TYPE
|
ENGINE
|
VERSION
|
ROW_FORMAT
|
TABLE_ROWS
|
AVG_ROW_LENGTH
|
DATA_LENGTH
|
MAX_DATA_LENGTH
|
INDEX_LENGTH
|
DATA_FREE
|
AUTO_INCREMENT
|
CREATE_TIME
|
UPDATE_TIME
|
CHECK_TIME
|
TABLE_COLLATION
|
CHECKSUM
|
CREATE_OPTIONS
|
TABLE_COMMENT
|
def
|
ruoyi
|
goods
|
BASE TABLE
|
InnoDB
|
10
|
Dynamic
|
1
|
16384
|
16384
|
0
|
0
|
0
|
4
|
2021-02-25 08:22:31
|
2021-02-26 05:18:20
|
(NULL)
|
utf8mb4_general_ci
|
(NULL)
|
|
商品表
|
COLUMNS
表中的记录为
TABLE_CATALOG
|
TABLE_SCHEMA
|
TABLE_NAME
|
COLUMN_NAME
|
ORDINAL_POSITION
|
COLUMN_DEFAULT
|
IS_NULLABLE
|
DATA_TYPE
|
CHARACTER_MAXIMUM_LENGTH
|
CHARACTER_OCTET_LENGTH
|
NUMERIC_PRECISION
|
NUMERIC_SCALE
|
DATETIME_PRECISION
|
CHARACTER_SET_NAME
|
COLLATION_NAME
|
COLUMN_TYPE
|
COLUMN_KEY
|
EXTRA
|
PRIVILEGES
|
COLUMN_COMMENT
|
GENERATION_EXPRESSION
|
def
|
ruoyi
|
goods
|
id
|
1
|
(NULL)
|
NO
|
int
|
(NULL)
|
(NULL)
|
10
|
0
|
(NULL)
|
(NULL)
|
(NULL)
|
int(11)
|
PRI
|
auto_increment
|
select,insert,update,references
|
主键
|
|
def
|
ruoyi
|
goods
|
GOODS_NAME
|
2
|
(NULL)
|
YES
|
varchar
|
255
|
1020
|
(NULL)
|
(NULL)
|
(NULL)
|
utf8mb4
|
utf8mb4_general_ci
|
varchar(255)
|
|
|
select,insert,update,references
|
商品名字
|
|
def
|
ruoyi
|
goods
|
put_way_flag
|
3
|
(NULL)
|
YES
|
tinyint
|
(NULL)
|
(NULL)
|
3
|
0
|
(NULL)
|
(NULL)
|
(NULL)
|
tinyint(1)
|
|
|
select,insert,update,references
|
商品是否上架,0:下架,1:上架
|
|
def
|
ruoyi
|
goods
|
create_time
|
4
|
(NULL)
|
YES
|
datetime
|
(NULL)
|
(NULL)
|
(NULL)
|
(NULL)
|
0
|
(NULL)
|
(NULL)
|
datetime
|
|
|
select,insert,update,references
|
创建时间
|
|
def
|
ruoyi
|
goods
|
create_by
|
5
|
(NULL)
|
YES
|
varchar
|
64
|
256
|
(NULL)
|
(NULL)
|
(NULL)
|
utf8mb4
|
utf8mb4_general_ci
|
varchar(64)
|
|
|
select,insert,update,references
|
创建人
|
|
def
|
ruoyi
|
goods
|
update_time
|
6
|
(NULL)
|
YES
|
datetime
|
(NULL)
|
(NULL)
|
(NULL)
|
(NULL)
|
0
|
(NULL)
|
(NULL)
|
datetime
|
|
|
select,insert,update,references
|
更新时间
|
|
def
|
ruoyi
|
goods
|
update_by
|
7
|
(NULL)
|
YES
|
varchar
|
64
|
256
|
(NULL)
|
(NULL)
|
(NULL)
|
utf8mb4
|
utf8mb4_general_ci
|
varchar(64)
|
|
|
select,insert,update,references
|
更新人
|
|
这里面详细记录着goods表和列的详细信息,这正是代码生成器实现的基石。
3.ruoyi-vue代码生成器源码分析
ruoyi-vue代码生成器相关代码均位于ruoyi-generator模块中,根据之前的实际操作体验上来看,最简单的情况,前端页面只需要两步即可完成代码生成
实际上这两步对应着后端的两个接口:com.ruoyi.generator.controller.GenController#importTableSave 和 com.ruoyi.generator.controller.GenController#batchGenCode ,生成源码的步骤就要从这两步下手。
首先看com.ruoyi.generator.controller.GenController#importTableSave 接口,它做了以下这些事情
-
从information_schema数据库的tables表中查询目标表的表明、标注释、创建时间和更新时间,但是忽略掉定时任务的表和已经生成过的表。
-
初始化表数据并将数据插入ruoyi数据库的gen_table表
-
从information_schema数据库的columns表中查询目标表的列信息,包含字段名、字段注释、字段类型、是否允许为null等详细信息
-
初始化列信息并将数据插入ruoyi数据库的gen_table_column表
接下来看下 com.ruoyi.generator.controller.GenController#batchGenCode 接口,它做了以下这些事情
-
从ruoyi数据库的gen_table、gen_table_column表查询出生成代码需要的表和列信息。
-
初始化Velocity
-
准备Velocity上下文信息(变量值信息)
-
读取模板、渲染模板,然后将渲染后的模板内容添加进如压缩流,之后前端就可以下载zip压缩文件了。
完毕。
六、扩展篇:封装独立版ruoyi-vue代码生成器
1.为什么要做这个
作为一个后端开发,我最经常做的事情不是搞啥系统架构,而是最简单的CRUD。。。若是能有一个代码一键生成工具自动根据已经创建的表信息生成CRUD后端代码,那岂不是能节省老鼻子功夫了——若依系统已经实现了这个代码生成器的工具,但是它依赖于前端页面,必须有权限访问“系统工具-代码生成”菜单才行,而这在企业中像我这种普通研发往往是是没有权限访问的。但是我有权限访问表,查看表结构。作为一个有追求的开发,既不肯开口问别人要权限,还想要实现代码生成器,该怎么做?
自己搞一个呗。
在这里封装的代码生成器不考虑前端页面调整功能,其实现的功能更加注重于后端代码,其作用和“系统工具-代码生成”页面中最简单的生成代码的两步(导入表和下载代码,无编辑)结果等效。
2.抽离ruoyi-vue代码生成器逻辑
ruoyi-vue中的ruoyi-generator模块有着完整的代码生成逻辑,但是它依托于ruoyi-admin的spring-boot框架才能运行,现在我要将ruoyi-generator模块的功能独立于spring-boot,让其作为一个普通的spring的程序,只有一个普通的main方法,实现和原来等效的功能。
这里的做法是直接修改ruoyi-generator模块,删除spring-boot的相关功能,但是保留spring、mybatis、druid等基础组件的依赖,然后将这些组件手动重新纳入spring容器中进行管理,最后通过main方法调用到相关模块。
具体就不展开讲了,有兴趣的可以看看我的源代码:https://gitee.com/kdyzm/ruoyi-vue-gen
3.独立版代码生成器使用方法
3.1 配置application.properties配置文件
该配置文件的内容如下:
mysql.username=${dbUserName}
mysql.password=${dbPassword}
mysql.connectionUrl=${dbUrl}
mysql.driverClass=${dbDriverClassName}
gen.author=kdyzm
gen.packageName=com.kdyzm.business
gen.autoRemovePre=false
gen.tableName=news
gen.tablePrefix=sys_
上半部分是数据库配置,连接的是ruoyi-vue数据库,正常配置即可;下半部分是生成配置,除了
gen.tableName
,其它配置和原ruoyi-vue代码生成器的配置相同。
要注意,代码生成结果仅使用用ruoyi-vue项目,如需自定义模板,需要修改源代码。
3.2 打包和运行
在项目根目录运行命令
mvn clean package
,打包完成之后切换到target目录,使用命令
java -jar ruoyi-vue-gen-1.0-SNAPSHOT.jar
运行jar包得到ruoyi.zip压缩文件