1.前言
我们在开发传统的Maven项目或者SpringBoot项目过程中,最终处理的就是对于数据的读与存,这时就免不了写对象关系映射代码,实体类、mapper、mapper.xml等代码,若存在大量的增删改查功能的话,我们开发时就会不断重复相同的操作,大大降低了工作效率。
为此便但成了Mybatis “逆向工程” ,所谓逆向工程,就是从数据库表结构出发,反向去生成对应的实体和映射代码,今天我们在这里提供2种方案进行实现。
2.实现方案
2.1. mybatis-generator生成
2.1.1. 环境说明
1)springboot 项目(JDK1.8) 2)apache-maven-3.6.1 3)MySQL 8.0.30
2.1.2. 数据库表
逆向工程的第一步,需要先有数据库表结构!我们在这里先创建一个表,作为示例使用,sql脚本如下:
--若已存在,先删除 drop table user -- 创建数据库用户表 create table user ( id int NOT NULL AUTO_INCREMENT COMMENT '主键', name varchar(100) DEFAULT NULL COMMENT '员工姓名', phone_num varchar(20) DEFAULT NULL COMMENT '联系方式', address varchar(200) DEFAULT NULL COMMENT '住址', PRIMARY KEY (`id`) ) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_general_ci;
2.1.3 添加Maven依赖
逆向工程的实现,需要先引入mybatis相关依赖,如果项目中已经引入过,可自动忽略这一步。我们主要需要mybatis,mysql基础依赖,以及mybatis提供的生成器依赖
<dependencies> <!-- mybatis-generator --> <dependency> <groupId>org.mybatis.generator</groupId> <artifactId>mybatis-generator-core</artifactId> <version>1.3.7</version> </dependency> <dependency> <groupId>org.mybatis.spring.boot</groupId> <artifactId>mybatis-spring-boot-starter</artifactId> <version>2.2.2</version> </dependency> <dependency> <groupId>mysql</groupId> <artifactId>mysql-connector-java</artifactId> <scope>runtime</scope> </dependency> <dependency> <groupId>org.springframework.boot</groupId> <artifactId>spring-boot-starter-test</artifactId> <scope>test</scope> </dependency> </dependencies>
2.1.4 数据库连接配置
依赖与数据库表创建好之后,我们就正式进入了springboot项目中进行逆向工程开发啦,首先在resources目录下创建一个数据库连接的配置文件:generator.properties
jdbc.driverClass=com.mysql.cj.jdbc.Driver jdbc.connectionURL=jdbc:mysql://localhost:3306/vhr?useUnicode=true&characterEncoding=UTF-8&serverTimezone=Asia/Shanghai jdbc.userId=root jdbc.password=123456
2.1.5 配置生成规则
数据库配置完成后,我们紧接着来设置一个生成规则,包括编码,生成文件路径、是否设置自定义注释等等,这个配置可以从网上搜,有很多generatorConfiguration的配置。直接CV稍作修改即可
<?xml version="1.0" encoding="UTF-8"?> <!DOCTYPE generatorConfiguration PUBLIC "-//mybatis.org//DTD MyBatis Generator Configuration 1.0//EN" "http://mybatis.org/dtd/mybatis-generator-config_1_0.dtd"> <generatorConfiguration> <!--根据配置文件读取数据库表信息--> <properties resource="generator.properties"/> <context id="MySqlContext" targetRuntime="MyBatis3" defaultModelType="flat"> <!--前置分隔符--> <property name="beginningDelimiter" value="`"/> <!--后置分隔符--> <property name="endingDelimiter" value="`"/> <!--设置编码--> <property name="javaFileEncoding" value="UTF-8"/> <!-- 为模型生成序列化方法--> <plugin type="org.mybatis.generator.plugins.SerializablePlugin"/> <!-- 为生成的Java模型创建一个toString方法 --> <plugin type="org.mybatis.generator.plugins.ToStringPlugin"/> <!--生成mapper.xml时覆盖原文件--> <plugin type="org.mybatis.generator.plugins.UnmergeableXmlMappersPlugin" /> <!--可以自定义生成model的代码注释--> <commentGenerator type="org.javaboy.vhr.utils.CommentGenerator"> <!-- 是否去除自动生成的注释 true:是 : false:否 --> <property name="suppressAllComments" value="true"/> <property name="suppressDate" value="true"/> <property name="addRemarkComments" value="true"/> </commentGenerator> <!--配置数据库连接--> <jdbcConnection driverClass="${jdbc.driverClass}" connectionURL="${jdbc.connectionURL}" userId="${jdbc.userId}" password="${jdbc.password}"> <!--解决mysql驱动升级到8.0后不生成指定数据库代码的问题--> <property name="nullCatalogMeansCurrent" value="true" /> </jdbcConnection> <!--指定生成model的路径--> <javaModelGenerator targetPackage="org.javaboy.vhr.pojo" targetProject="vhr-web\src\main\java"/> <!--指定生成mapper.xml的路径--> <sqlMapGenerator targetPackage="mybatis.mapper" targetProject="vhr-web\src\main\resources"/> <!--指定生成mapper接口的的路径--> <javaClientGenerator type="XMLMAPPER" targetPackage="org.javaboy.vhr.mapper" targetProject="vhr-web\src\main\java"/> <!--单独生成某个表,写具体表名,否则生成全部表tableName设为%--> <table tableName="user" enableCountByExample="false" enableUpdateByExample="false" enableDeleteByExample="false" enableSelectByExample="false" selectByExampleQueryId="false"> <property name="useActualColumnNames" value="false"/> <property name="constructorBased" value="false"/> <generatedKey column="id" sqlStatement="JDBC" identity="true"/> </table> </context> </generatorConfiguration>
2.1.6 自定义实体类注释配置
由于在上一步我们配置了注释生成器,因此,我们要在相应的路径下,创建一个类,来自定义注释的生成规则。路径:org.javaboy.vhr.utils.CommentGenerator
<commentGenerator type="org.javaboy.vhr.utils.CommentGenerator"> <!-- 是否去除自动生成的注释 true:是 : false:否 --> <property name="suppressAllComments" value="true"/> <property name="suppressDate" value="true"/> <property name="addRemarkComments" value="true"/> </commentGenerator>
CommentGenerator代码
** * @author 86138 * @date 2024-05-01 13:27:15 * @description 添加自定义注释 */ public class CommentGenerator extends DefaultCommentGenerator { private boolean addRemarkComments = false; /** * 设置用户配置的参数 */ @Override public void addConfigurationProperties(Properties properties) { super.addConfigurationProperties(properties); this.addRemarkComments = StringUtility.isTrue(properties.getProperty("addRemarkComments")); } /** * 给字段添加注释 */ @Override public void addFieldComment(Field field, IntrospectedTable introspectedTable, IntrospectedColumn introspectedColumn) { String remarks = introspectedColumn.getRemarks(); //根据参数和备注信息判断是否添加备注信息 if (addRemarkComments && StringUtility.stringHasValue(remarks)) { addFieldJavaDoc(field, remarks); } } /** * 给model的字段添加注释 */ private void addFieldJavaDoc(Field field, String remarks) { //文档注释开始 field.addJavaDocLine("/**"); //获取数据库字段的备注信息 String[] remarkLines = remarks.split(System.getProperty("line.separator")); for (String remarkLine : remarkLines) { field.addJavaDocLine(" * " + remarkLine); } addJavadocTag(field, false); field.addJavaDocLine(" */"); } }
2.1.7 使用生成器
以上步骤做完后,我们就要写我们的逆向工程生成器啦,调用org.mybatis.generator.api中的MyBatisGenerator生成器进行实现。注意:当生成的代码重复时,覆盖原代码!直接执行下述Generator类的main方法,就会自动读取数据库中的表,并生成对应的实体类与映射
public class Generator { public static void main(String[] args) throws Exception { //MBG 执行过程中的警告信息 List<String> warnings = new ArrayList<String>(); //当生成的代码重复时,覆盖原代码 boolean overwrite = true; //读取我们的 MBG 配置文件 InputStream is = Generator.class.getResourceAsStream("/generatorConfig.xml"); ConfigurationParser cp = new ConfigurationParser(warnings); Configuration config = cp.parseConfiguration(is); is.close(); DefaultShellCallback callback = new DefaultShellCallback(overwrite); //创建 MBG MyBatisGenerator myBatisGenerator = new MyBatisGenerator(config, callback, warnings); //执行生成代码 myBatisGenerator.generate(null); //输出警告信息 for (String warning : warnings) { System.out.println(warning); } } }
2.2. MybatisX插件生成(推荐)
基于方案一,我们可以快速的根据数据库中的表进行逆向工程的创建,大量的减少了我们的重复性工作,但是!它的实现还是比较复杂的,generatorConfiguration配置中的内容丰富且复杂,比较难记,而且在原有代码中需要写一定的代码量才能实现。
接下来我们要说的这种方案二,几乎不用写任何代码,只需要点点点就可以实现逆向工程啦,如果你使用idea的话,这种方案尤其方便。
2.2.1 下载IDEA插件
在插件管理中下载MybatisX插件
2.2.2 配置数据源
在idea的database中配置数据源,这里以MySQL为例,需要注意时区的配置。
2.2.3 配置maven依赖
<!--lombok依赖--> <dependency> <groupId>org.projectlombok</groupId> <artifactId>lombok</artifactId> <optional>true</optional> </dependency> <!--MyBatis Plus依赖--> <dependency> <groupId>com.baomidou</groupId> <artifactId>mybatis-plus-boot-starter</artifactId> <version>3.1.0</version> </dependency>
2.2.4 配置生成规则
打开刚刚配置好的本地数据库,选择你想构建的数据表,右键 -> MybatisX-Generator。
选择项目包路径、实体包名、表名;
next后继续配置mapper、xml、service的生成路径以及需要的注释功能,hashCode/equals、Lombk注解等等。这个要根据你项目的实际情况来选择
在上一步的最后配置完成,点击Finish按钮,查看结果,在对应的路径下成功的生成了我们需要的文件。
3.总结
这次我们简单介绍了Mybatis逆向工程的两种实现方案,方案一实现复杂,但可以满足基本的增删改查的实现,博主还是更偏向于第二种哈哈,可视化操作起来比较简单,大家可以根据自己的日常开发场景,进行合理选择。