Spring Boot 项目利用 MyBatis Generator 进行数据层代码自动生成

概 述

MyBatis Generator (简称 MBG) 是一个用于 MyBatis 和 iBATIS 的代码生成器。它可以为 MyBatis 的所有版本以及 2.2.0 之后的 iBATIS 版本自动生成 ORM 层代码，典型地包括我们日常需要手写的 POJO、mapper xml 以及 mapper 接口等。MyBatis Generator 自动生成的 ORM 层代码几乎可以应对大部分 CRUD 数据表操作场景，可谓是一个生产力工具啊！

数据库准备与工程搭建

• 首先我们准备一个 MySQL 数据表 user_info用于下文实验

里面插入了若干条数据：

• 新建一个 Spring Boot 工程

• 引入 MyBatis Generator 依赖

<dependency> <groupId>org.mybatis.generator</groupId> <artifactId>mybatis-generator-core</artifactId> <version>1.3.7</version> <scope>provided</scope> </dependency> 

• 引入 MyBatis Generator Maven 插件

<plugin> <groupId>org.mybatis.generator</groupId> <artifactId>mybatis-generator-maven-plugin</artifactId> <version>1.3.7</version> <configuration> <configurationFile>src/main/resources/mybatis-generator.xml</configurationFile> <overwrite>true</overwrite> </configuration> <dependencies> <dependency> <groupId>mysql</groupId> <artifactId>mysql-connector-java</artifactId> <version>8.0.12</version> </dependency> </dependencies> </plugin> 

MyBatis Generator Maven 插件引入以后，我们可以在 Spring Boot 工程的 Maven 插件工具栏中看到新增的插件选项，类似下图：

准备 MyBatis Generator 配置文件

MyBatis Generator 也需要一个 xml 格式的配置文件，该文件的位置配在了上文 引入 MyBatis Generator Maven 插件的 xml 配置里，即src/main/resources/mybatis-generator.xml，下面给出本文所使用的配置：

<?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> <context id="MySql" defaultModelType="flat"> <plugin type="org.mybatis.generator.plugins.SerializablePlugin" /> <jdbcConnection driverClass="com.mysql.jdbc.Driver" cOnnectionURL="jdbc:mysql://121.196.123.245:3306/demo" userId="root" password="xxxxxx" /> <javaModelGenerator targetPackage="cn.codesheep.springbt_mybatis_generator.entity" targetProject="src/main/java"></javaModelGenerator> <sqlMapGenerator targetPackage="mapper" targetProject="src/main/resources"></sqlMapGenerator> <javaClientGenerator targetPackage="cn.codesheep.springbt_mybatis_generator.mapper" targetProject="src/main/java" type="XMLMAPPER"></javaClientGenerator> <table tableName="user_info"> <property name="modelOnly" value="false"/> </table> </context> </generatorConfiguration> 

上面 xml 中几个关键的配置简介如下：

• < jdbcConnection /> 数据库连接配置，至关重要
• <javaModelGenerator /> 指定自动生成的 POJO 置于哪个包下
• <sqlMapGenerator /> 指定自动生成的 mapper.xml 置于哪个包下
• <javaClientGenerator /> 指定自动生成的 DAO 接口置于哪个包下
• <table /> 指定数据表名，可以使用_和%通配符

更多关于 MyBatis Generator 配置的内容，可以移步 官方文档。

运行 MyBatis Generator

直接通过 IDEA 的 Maven 图形化插件来运行 MyBatis Generator，其自动生成的过程 和 生成的结果如下图所示：

很明显，通过 MyBatis Generator，已经很方便的帮我们自动生成了 POJO、mapper xml 以及 mapper接口，接下来我们看看自动生成的代码怎么用！

自动生成的代码如何使用

我们发现通过 MyBatis Generator 自动生成的代码中带有一个 Example 文件，比如上文中的 UserInfoExample，其实 Example 文件对于平时快速开发还是有很大好处的，它能节省很多写 sql 语句的时间，举几个实际的例子吧：

• 单条件模糊搜索 + 排序

在我们的例子中，假如我想通过用户名 user_name来在 MySQL 数据表 user_info中进行模糊搜索，并对结果进行排序，此时利用UserInfoExample可以方便快速的实现：

@Autowired private UserInfoMapper userInfoMapper; public List<UserInfo> searchUserByUserName( String userName ) { UserInfoExample userInfoExample = new UserInfoExample(); userInfoExample.createCriteria().andUserNameLike( '%'+ userName +'%' ); // 设置模糊搜索的条件 String orderByClause = "user_name DESC"; userInfoExample.setOrderByClause( orderByClause ); // 设置通过某个字段排序的条件 return userInfoMapper.selectByExample( userInfoExample ); } 

• 多条件精确搜索

再比如，我们想通过电话号码 phone和用户名 user_name 两个字段来在数据表 user_info中实现精确搜索，则可以如下实现：

public List<UserInfo> multiConditionsSearch( UserInfo userInfo ) { UserInfoExample userInfoExample = new UserInfoExample(); UserInfoExample.Criteria criteria = userInfoExample.createCriteria(); if( !"".equals(userInfo.getPhone()) ) criteria.andPhoneEqualTo( userInfo.getPhone() ); if( !"".equals(userInfo.getUserName()) ) criteria.andUserNameEqualTo( userInfo.getUserName() ); return userInfoMapper.selectByExample( userInfoExample ); } 

很明显可以看出的是，我们是通过直接编写代码逻辑来取代编写 SQL 语句，因此还是十分直观和容易理解的！

后 记

由于能力有限，若有错误或者不当之处，还请大家批评指正，一起学习交流！

• My Personal Blog：CodeSheep 程序羊