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方式。