Mybatis插入时返回自增主键方式(selectKey和useGeneratedKeys)

MyBatis是一个支持自动生成SQL的持久层框架，可以将查询结果映射到Java对象上，而且支持多种自增主键返回方式，本文将重点讲解Mybatis插入时返回自增主键的两种方式：selectKey和useGeneratedKeys。

1. selectKey方式

1.1 selectKey方式概述

selectKey方式是通过SQL语句在执行INSERT命令时，返回插入后的主键值，并将其自动赋值给指定的javaBean或者Map对象中相应的属性值。

1.2 selectKey方式原理

使用SELECT LAST_INSERT_ID()函数来获取自增主键的值，同时利用MyBatis的ParameterMap功能将获取到的值作为参数注入到INSERT语句中。

1.3 selectKey方式示例

下面是一个使用selectKey方式向MySQL数据库中插入自增主键数据的示例：

<insert id="insertUser" parameterType="com.example.User">
  <selectKey keyProperty="id" resultType="int" order="AFTER">
    SELECT LAST_INSERT_ID()
  </selectKey>
  insert into user (username, age) values (#{username}, #{age})
</insert>

上述代码中，keyProperty属性表示将返回的主键值注入到User对象中的id属性中，order="AFTER"表示查询语句在INSERT语句之后执行。

2. useGeneratedKeys方式

2.1 useGeneratedKeys方式概述

useGeneratedKeys方式是通过JDBC驱动获取自增主键的值，并将其自动赋值给指定的javaBean或者Map对象中相应的属性值，不需要用户手动指定查询语句。

2.2 useGeneratedKeys方式原理

使用JDBC PreparedStatement的方法prepareStatement(String sql, int autoGeneratedKeys)设置autoGeneratedKeys参数为Statement.RETURN_GENERATED_KEYS，即可获取自增主键的值。MyBatis底层会自动识别是否使用了自增主键，并且获取相关信息。

2.3 useGeneratedKeys方式示例

下面是一个使用useGeneratedKeys方式向MySQL数据库中插入自增主键数据的示例：

<insert id="insertUser" parameterType="com.example.User" useGeneratedKeys="true" keyProperty="id">
  insert into user (username, age) values (#{username}, #{age})
</insert>

上述代码中，useGeneratedKeys属性表示启用自动获取自增主键值的功能，keyProperty属性表示将返回的主键值注入到User对象中的id属性中。

3. 总结

selectKey方式和useGeneratedKeys方式均可以实现Mybatis插入时返回自增主键的功能，但是两种方式适用的场景有所不同。

在一般情况下，useGeneratedKeys方式更加简单和安全，因此推荐使用。而在一些需要设置更加高级的特性的情况下，比如需要自定义一些查询逻辑或者处理一些异常情况，可以考虑使用selectKey方式。