转载

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

概述

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

注：本文首发于 My Personal Blog：CodeSheep·程序羊 ，欢迎光临小站

数据库准备与工程搭建

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

里面插入了若干条数据：

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

新建一个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插件工具栏中看到新增的插件选项，类似下图：

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

准备 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 />
<sqlMapGenerator />
<javaClientGenerator />
<table />

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

运行 MyBatis Generator

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

Spring Boot项目利用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语句，因此还是十分直观和容易理解的！