更新于:

代码生成(新增功能)

目录

代码生成(新增功能)

大部分项目里,其实有很多代码是重复的,几乎每个模块都有 CRUD 增删改查的功能,而这些功能的实现代码往往是大同小异的。如果这些功能都要自己去手写,非常无聊枯燥,浪费时间且效率很低,还可能会写错。

所以这种重复性的代码,项目提供了 codegen (opens new window) 代码生成器,只需要在数据库中设计好表结构,就可以一键生成前后端代码 + 单元测试 + Swagger 接口文档 + Validator 参数校验。

下面,我们使用代码生成器,在 yudao-module-system 模块中,开发一个【用户组】的功能。

# 👍 相关视频教程

# 1. 数据库表结构设计

设计用户组的数据库表名为 system_group,其建表语句如下:

1
2
3
4
5
6
7
8
9
10
11
12
13
CREATE TABLE `system_group` (
`id` bigint NOT NULL AUTO_INCREMENT COMMENT '编号',
`name` varchar(255) COLLATE utf8mb4_unicode_ci NOT NULL COMMENT '名字',
`description` varchar(512) COLLATE utf8mb4_unicode_ci DEFAULT NULL COMMENT '描述',
`status` tinyint NOT NULL COMMENT '状态',
`creator` varchar(64) CHARACTER SET utf8mb4 COLLATE utf8mb4_unicode_ci DEFAULT '' COMMENT '创建者',
`create_time` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP COMMENT '创建时间',
`updater` varchar(64) CHARACTER SET utf8mb4 COLLATE utf8mb4_unicode_ci DEFAULT '' COMMENT '更新者',
`update_time` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP COMMENT '更新时间',
`deleted` bit(1) NOT NULL DEFAULT b'0' COMMENT '是否删除',
`tenant_id` bigint NOT NULL DEFAULT '0' COMMENT '租户编号',
PRIMARY KEY (`id`) USING BTREE
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_unicode_ci COMMENT='用户组';

建表语句

① 表名的前缀,要和 Maven Module 的模块名保持一致。例如说,用户组在 yudao-module-system 模块,所以表名的前缀是 system_

疑问:为什么要保持一致?

代码生成器会自动解析表名的前缀,获得其所属的 Maven Module 模块,简化配置过程。

② 设置 ID 主键,一般推荐使用 bigint 长整形,并设置自增长。

③ 正确设置每个字段是否允许空,代码生成器会根据它生成参数是否允许空的校验规则。

④ 正确设置注释,代码生成器会根据它生成字段名与提示等信息。

⑤ 添加 creatorcreate_timeupdaterupdate_timedeleted 是必须设置的系统字段;如果开启多租户的功能,并且该表需要多租户的隔离,则需要添加 tenant_id 字段。

# 2. 代码生成

① 点击 [基础设施 -> 代码生成] 菜单,点击 [基于 DB 导入] 按钮,选择 system_group 表,后点击 [确认] 按钮。

基于 DB 导入

代码实现?

可见 CodegenBuilder (opens new window) 类,自动解析数据库的表结构,生成默认的配置。

② 点击 system_group 所在行的 [编辑] 按钮,修改生成配置。后操作如下:

基本信息

字段信息

  • status 字段的显示类型为【下拉框】,字典类型为【系统状态】。
  • description 字段的【查询】取消。
  • idnamedescriptionstatus 字段的【示例】填写上。

字段信息

  • 插入:新增时,是否传递该字段。
  • 编辑:修改时,是否传递该字段。
  • 列表:Table 表格,是否展示该字段。
  • 查询:搜索框,是否支持该字段查询,查询的条件是什么。
  • 允许空:新增或修改时,是否必须传递该字段,用于 Validator 参数校验。
  • 字典类型:在显示类型是下拉框、单选框、复选框时,选择使用的字典。
  • 示例:参数示例,用于 Swagger 接口文档的 example 示例。

生成信息

  • 将【上级菜单】设置为【系统管理】。
  • 将【前端类型】设置为【Vue2 Element UI 标准模版】或【Vue3 Element Plus 标准模版】,具体根据你使用哪种管理后台。

生成信息

  • 生成场景:分成管理后台、用户 App 两种,用于生成 Controller 放在 admin 还是 app 包。
  • 上级菜单:生成场景是管理后台时,需要设置其所属的上级菜单。
  • 前端类型: 提供多种 UI 模版。
    • 【Vue3 Element Plus Schema 模版】,对应 《前端手册 Vue 3.X —— CRUD 组件》 说明。
    • 后端的 application.yaml 配置文件中的 yudao.codegen.front-type 配置项,设置默认的 UI 模版,避免每次都需要设置。

完成后,点击 [提交] 按钮,保存生成配置。

③ 点击 system_group 所在行的 [预览] 按钮,在线预览生成的代码,检查是否符合预期。

生成信息

④ 点击 system_group 所在行的 [生成代码] 按钮,下载生成代码的压缩包,双击进行解压。

生成代码

代码实现?

可见 CodegenEngine (opens new window) 类,基于 Velocity 模板引擎,生成具体代码。模板文件,可见 resources/codegen (opens new window) 目录。

# 3. 代码运行

本小节,我们将生成的代码,复制到项目中,并进行运行。

# 3.1 后端运行

① 将生成的后端代码,复制到项目中。操作如下图所示:

复制后端

② 将 ErrorCodeConstants.java_手动操作 文件的错误码,复制到该模块 ErrorCodeConstants 类中,并设置对应的错误码编号,之后进行删除。操作如下图所示:

添加错误码

③ 将 h2.sql 的 CREATE 语句复制到该模块的 create_tables.sql 文件,DELETE 语句复制到该模块的 clean.sql。操作如下图:

复制 h2 SQL

疑问:`create_tables.sql` 和 `clean.sql` 文件的作用是什么?

项目的单元测试,需要使用到 H2 内存数据库,create_tables.sql 用于创建所有的表结构,clean.sql 用于每个单元测试的方法跑完后清理数据。

然后,运行 GroupServiceImplTest 单元测试,执行通过。

④ 打开数据库工具,运行代码生成的 sql/sql.sql 文件,用于菜单的初始化。

初始化菜单

⑤ Debug 运行 YudaoServerApplication 类,启动后端项目。通过 IDEA 的 [Actuator -> Mappings] 菜单,可以看到代码生成的 GroupController 的 RESTful API 接口已经生效。

RESTful API 接口

# 3.2 前端运行

① 将生成的前端代码,复制到项目中。操作如下图所示:

复制前端

重新执行 npm run dev 命令,启动前端项目。点击 [系统管理 -> 用户组管理] 菜单,就可以看到用户组的 UI 界面。

前端界面


至此,我们已经完成了【用户组】功能的代码生成,基本节省了你 80% 左右的开发任务,后续可以根据自己的需求,进行剩余的 20% 的开发!

# 4. 后续变更

随着业务的发展,已经生成代码的功能需要变更。继续以【用户组】举例子,它的 system_group 表需要新增一个分类 category 字段,此时不建议使用代码生成器,而是直接修改已经生成的代码:

① 后端:修改 GroupDO 数据实体类、GroupBaseVO 基础 VO 类、GroupExcelVO 导出结果 VO 类,新增 category 字段。

② 前端:修改 index.vue 界面的列表和表单组件,新增 category 字段。

③ 重新编译后后端,并进行启动。

over!非常简单方便,即保证了代码的整洁规范,又不增加过多的开发量。

上次更新: 2023/04/13, 08:15:12

本站由 钟意 使用 Stellar 1.25.0 主题创建。
又拍云 提供CDN加速/云存储服务
vercel 提供托管服务
湘ICP备2023019799号-1
总访问 次 | 本页访问