spring boot集成mybatis-plus——Mybatis Plus 新增数据并返回主键 ID（图文讲解）

妖狐艹你老母 2023-09-29 18:48 18阅读 0赞

## Mybatis Plus 新增数据并返回主键 ID(图文讲解) ##

更新时间 2023-01-10 15:37:37

大家好，我是小哈。

本小节中，我们将学习如何通过 Mybatis Plus 框架给数据库表新增数据，主要内容思维导图如下：

![Mybatis Plus 新增数据思维导图][Mybatis Plus]Mybatis Plus 新增数据思维导图

### 表结构 ###

为了演示新增数据，在前面小节中，我们已经定义好了一个用于测试的用户表， 执行脚本如下：

DROP TABLE IF EXISTS t_user;
    
    CREATE TABLE `t_user` ( `id` bigint(20) UNSIGNED NOT NULL AUTO_INCREMENT COMMENT '主键ID', `name` varchar(30) NOT NULL DEFAULT '' COMMENT '姓名', `age` int(11) NULL DEFAULT NULL COMMENT '年龄', `gender` tinyint(2) NOT NULL DEFAULT 0 COMMENT '性别，0：女 1：男', PRIMARY KEY (`id`) ) COMMENT = '用户表';

### 定义实体类 ###

定义一个名为 `User` 实体类：

@Data
    @TableName("t_user")
    public class User {
        /**
         * 主键 ID, @TableId 注解定义字段为表的主键，type 表示主键类型，IdType.AUTO 表示随着数据库 ID 自增
         */
        @TableId(type = IdType.AUTO) private Long id; /** * 姓名 */ private String name; /** * 年龄 */ private Integer age; /** * 性别 */ private Integer gender; }

讲解一下实体类中用到的注解：

#### @TableName 表名注解 ####

作用：标识实体类对应的表。

> TIP ：
> 
>  *  当实体类名称和实际表名一致时，如实体名为 `User`， 表名为 `user` ，可不用添加该注解，Mybatis Plus 会自动识别并映射到该表。
>  *  当实体类名称和实际表名不一致时，如实体名为 `User`， 表名为 `t_user`，需手动添加该注解，并填写实际表名称。

#### @TableId 主键注解 ####

作用：声明实体类中的主键对应的字段。

<table> 
 <thead> 
  <tr> 
   <th align="left">属性</th> 
   <th align="left">类型</th> 
   <th align="left">必须指定</th> 
   <th align="left">默认值</th> 
   <th align="left">描述</th> 
  </tr> 
 </thead> 
 <tbody> 
  <tr> 
   <td align="left">value</td> 
   <td align="left">String</td> 
   <td align="left">否</td> 
   <td align="left">""</td> 
   <td align="left">主键字段名</td> 
  </tr> 
  <tr> 
   <td align="left">type</td> 
   <td align="left">Enum</td> 
   <td align="left">否</td> 
   <td align="left">IdType.NONE</td> 
   <td align="left">指定主键类型</td> 
  </tr> 
 </tbody> 
</table>

#### IdType 主键类型 ####

<table> 
 <thead> 
  <tr> 
   <th align="left">值</th> 
   <th align="left">描述</th> 
  </tr> 
 </thead> 
 <tbody> 
  <tr> 
   <td align="left">AUTO</td> 
   <td align="left">数据库 ID 自增</td> 
  </tr> 
  <tr> 
   <td align="left">NONE</td> 
   <td align="left">无状态，该类型为未设置主键类型(默认)</td> 
  </tr> 
  <tr> 
   <td align="left">INPUT</td> 
   <td align="left">插入数据前，需自行设置主键的值</td> 
  </tr> 
  <tr> 
   <td align="left">ASSIGN_ID</td> 
   <td align="left">分配 ID(主键类型为 Number(Long 和 Integer)或 String)(since 3.3.0),使用接口<code>IdentifierGenerator</code>的方法<code>nextId</code>(默认实现类为<code>DefaultIdentifierGenerator</code>雪花算法)</td> 
  </tr> 
  <tr> 
   <td align="left">ASSIGN_UUID</td> 
   <td align="left">分配 UUID,主键类型为 String(since 3.3.0),使用接口<code>IdentifierGenerator</code>的方法<code>nextUUID</code>&nbsp;(默认 default 方法)</td> 
  </tr> 
  <tr> 
   <td align="left">ID_WORKER</td> 
   <td align="left">分布式全局唯一 ID 长整型类型 (推荐使用&nbsp;<code>ASSIGN_ID</code>)</td> 
  </tr> 
  <tr> 
   <td align="left">UUID</td> 
   <td align="left">32 位 UUID 字符串 (推荐使用&nbsp;<code>ASSIGN_UUID</code>)</td> 
  </tr> 
  <tr> 
   <td align="left">ID_WORKER_STR</td> 
   <td align="left">分布式全局唯一 ID 字符串类型 (推荐使用&nbsp;<code>ASSIGN_ID</code>)</td> 
  </tr> 
 </tbody> 
</table>

### 开始新增数据 ###

测试表准备好后，我们准备开始演示新增数据。实际上，Mybatis Plus 对 Mapper 层和 Service 层都将常见的增删改查操作都封装好了，只需简单的继承，即可轻松搞定对数据的增删改查，本文重点讲解新增数据这块。

#### Mapper 层 ####

定义一个 `UserMapper` , 让其继承 `BaseMapper` :

public interface UserMapper extends BaseMapper<User> { }

然后，注入 Mapper :

@Autowired
    private UserMapper userMapper;

`BaseMapper` 提供的新增方法仅一个 `insert()` 方法：

![071f1c47070814a9b495407866ec465e.jpeg][]

我们通过它测试一下添加数据，并获取主键 ID :

User user =new User();
    user.setName("犬小哈");
    user.setAge(30);
    user.setGender(1);
    
    userMapper.insert(user);
    
    // 获取插入数据的主键 ID Long id = user.getId(); System.out.println("id:" + id);

怎么样，是不是非常简单呢！

#### Service 层 ####

Mybatis Plus 同样也封装了通用的 Service 层 CRUD 操作，并且提供了更丰富的方法。接下来，我们上手看 Service 层的代码结构，如下图：

![定义 Service 层][Service]定义 Service 层

先定义 `UserService` 接口 ，让其继承自 `IService`:

public interface UserService extends IService<User> { }

再定义实现类 `UserServiceImpl`，让其继承自 `ServiceImpl`, 同时实现 `UserService` 接口，这样就可以让 `UserService` 拥有了基础通用的 CRUD 功能，当然，实际开发中，业务会更加复杂，就需要向 `IService` 接口自定义方法并实现:

@Service
    public class UserServiceImpl extends ServiceImpl<UserMapper, User> implements UserService { }

注入 `UserService` :

@Autowired
    private UserService userService;

与 Mapper 层不同的是，Service 层的新增方法均以 `save` 开头，并且功能更丰富，来看看都提供了哪些方法：

![新增数据共有 6 个方法][6]新增数据共有 6 个方法

简单解释下每个方法的作用，以作了解：

// 新增数据
    sava(T) : boolean
    // 伪批量插入，实际上是通过 for 循环一条一条的插入
    savaBatch(Collection<T>) : boolean
    // 伪批量插入，int 表示批量提交数，默认为 1000 savaBatch(Collection<T>, int) : boolean // 新增或更新(单条数据) saveOrUpdate(T) : boolean // 批量新增或更新 saveOrUpdateBatch(Collection<T>) : boolean // 批量新增或更新(可指定批量提交数) saveOrUpdateBatch(Collection<T>, int) : boolean

大致看完后，上手测试一下。

##### sava(T) #####

简单的新增数据，示例代码如下：

// 新增数据
    // 实际执行 SQL :INSERT INTO user ( name, age, gender ) VALUES ( '小哈 111', 30, 1 ) User user = new User(); user.setName("小哈 111"); user.setAge(30); user.setGender(1); boolean isSuccess = userService.save(user); // 返回主键ID Long id = user.getId(); System.out.println("isSuccess:" + isSuccess); System.out.println("主键 ID: " + id);

##### savaBatch(Collection) #####

伪批量插入，注意，命名虽然包含了批量的意思，但这不是真的批量插入，不信的话，我们来实际测试一下：

// 批量插入
    List<User> users = new ArrayList<>();
    for (int i = 0; i < 5; i++) { User user = new User(); user.setName("犬小哈" + i); user.setAge(i); user.setGender(1); users.add(user); } boolean isSuccess = userService.saveBatch(users); System.out.println("isSuccess:" + isSuccess);

执行代码，实际执行的 SQL 如下：

> TIP : 如何打印实际执行的 SQL, 可参考之前小节的 [《Mybatis Plus 打印 SQL 语句(包含执行耗时)》][Mybatis Plus _ SQL] 。

![批量插入输入 SQL 打印][SQL]批量插入输入 SQL 打印

可以看到，并不是 `insert into user (xxx) values (xxx),(xxx),(xxx)` 这种批量形式，还是一条一条插入的。

###### 批量新增源码分析 ######

我们来看下源码, 内部的`saveBatch()` 方法默认的批量提交阀值参数，数值为 1000， 即达到 1000 条批量提交一次，继续点进去看：：

![89254809f93cab01c96e2e330238de91.jpeg][]

![源码分析][cb9b6becb213484c5c33fa7450925bfa.jpeg]源码分析

public boolean saveBatch(Collection<T> entityList, int batchSize) { // 获取预编译的插入 SQL String sqlStatement = this.getSqlStatement(SqlMethod.INSERT_ONE); // for 循环执行 insert return this.executeBatch(entityList, batchSize, (sqlSession, entity) -> { sqlSession.insert(sqlStatement, entity); }); }

再看下 `SqlMethod.INSERT_ONE` 这个枚举，描述信息为*插入一条数据*：

![5454af13bb33682f1de0b0f4fdb16f35.jpeg][]

继续往 `executeBatch()` 方法里看，瞅瞅它这个批量到底是怎么处理的，具体每行代码的意思，小哈都加了注释：

public static <E> boolean executeBatch(Class<?> entityClass, Log log, Collection<E> list, int batchSize, BiConsumer<SqlSession, E> consumer) { // 断言需要批处理数据集大小不等于1 Assert.isFalse(batchSize < 1, "batchSize must not be less than one", new Object[0]); // 判空数据集，若不为空，则开始执行批量处理 return !CollectionUtils.isEmpty(list) && executeBatch(entityClass, log, (sqlSession) -> { int size = list.size(); // 将批处理大小与传入的操作集合大小进行比较，取最小的那个 int idxLimit = Math.min(batchSize, size); int i = 1; // 迭代器循环 for(Iterator var7 = list.iterator(); var7.hasNext(); ++i) { // 获取当前需要执行的数据库操作 E element = var7.next(); // 回调 sqlSession.insert() 方法 consumer.accept(sqlSession, element); // 判断是否达到需要批处理的阀值 if (i == idxLimit) { // 开始批处理，此方法执行并清除缓存在 JDBC 驱动类中的执行语句 sqlSession.flushStatements(); idxLimit = Math.min(idxLimit + batchSize, size); } } }); }

看完就明白了，相比较自己手动 `for` 循环执行插入，Mybatis Plus 这个伪批量插入性能会更好些，内部会将每次的插入语句缓存起来，等到达到 1000 条的时候，才会统一推给数据库，虽然最终在数据库那边还是一条一条的执行 INSERT，但还是在和数据库交互的 IO 上做了优化。

##### savaBatch(Collection, int) #####

多了个 `batchSize` 参数，可以手动指定批处理的大小，即多少 SQL 操作执行一次，默认为 1000。

##### saveOrUpdate(T) #####

保存或者更新。即当你需要执行的数据，数据库中不存在时，就执行插入操作：

// 实际执行 SQL :INSERT INTO user ( name, age, gender ) VALUES ( '小小哈', 60, 1 ) User user = new User(); user.setName("小小哈"); user.setAge(60); user.setGender(1); userService.saveOrUpdate(user);

当你需要执行的数据，数据库中已存在时，就执行更新操作。框架是如何判断该记录是否存在呢？ 如设置了主键 ID，因为主键 ID 必须是唯一的，Mybatis Plus 会先执行查询操作，判断数据是否存在，存在即执行更新，否则，执行插入操作：

User user =new User();
    // 设置了主键字段
    user.setId(21L);
    user.setName("小小哈");
    user.setAge(60); user.setGender(1); userService.saveOrUpdate(user);

具体执行 SQL 如下：

![b81badcdc0359dc0ede66601b1859691.jpeg][]

##### saveOrUpdateBatch(Collection) #####

批量保存或者更新，示例代码如下：

List<User> users =new ArrayList<>();
    for (int i = 0; i < 5; i++) { User user = new User(); user.setId(Long.valueOf(i)); user.setName("犬小哈" + i); user.setAge(i+1); user.setGender(1); users.add(user); } userService.saveOrUpdateBatch(users);

##### saveOrUpdateBatch(Collection, int) #####

批量保存或者更新(可手动指定批量大小)，示例代码如下：

[Mybatis Plus]: https://img-blog.csdnimg.cn/img_convert/3f9d5f1d6d047b0325ebeb8c81481bba.jpeg
[071f1c47070814a9b495407866ec465e.jpeg]: https://img-blog.csdnimg.cn/img_convert/071f1c47070814a9b495407866ec465e.jpeg
[Service]: https://img-blog.csdnimg.cn/img_convert/c7d18e883f8305bbe4c976ec0b5f958f.jpeg
[6]: https://img-blog.csdnimg.cn/img_convert/0157018d0bfbced7b26b04ef607fba27.jpeg
[Mybatis Plus _ SQL]: https://www.quanxiaoha.com/mybatis-plus/mybatisplus-print-sql.html
[SQL]: https://img-blog.csdnimg.cn/img_convert/2422c0109918035fb30f9b454128fbcb.jpeg
[89254809f93cab01c96e2e330238de91.jpeg]: https://img-blog.csdnimg.cn/img_convert/89254809f93cab01c96e2e330238de91.jpeg
[cb9b6becb213484c5c33fa7450925bfa.jpeg]: https://img-blog.csdnimg.cn/img_convert/cb9b6becb213484c5c33fa7450925bfa.jpeg
[5454af13bb33682f1de0b0f4fdb16f35.jpeg]: https://img-blog.csdnimg.cn/img_convert/5454af13bb33682f1de0b0f4fdb16f35.jpeg
[b81badcdc0359dc0ede66601b1859691.jpeg]: https://img-blog.csdnimg.cn/img_convert/b81badcdc0359dc0ede66601b1859691.jpeg