首页 > 编程语言 > Mybatis insert方法主键回填和自定义操作
2021
10-17

Mybatis insert方法主键回填和自定义操作

在数据库插入的时候,有很多属性需要我们自己处理,如主键自增字段。

MYSQL中主键根据一定规则生成后,需要我们在插入后去主动获取,以便后面的操作,Mybatis为我们提供了处理的方法。

主键回填

  • keyProperty:指定哪个字段是主键
  • useGeneratedKeys:这个主键是否使用数据库内置生成策略

我们可以在XML文件中进行如下配置:

<insert id="insertUser" parameterType="user" useGeneratedKeys="true" keProperty="id">
 insert into t_user(user_name,age) values (#${userName},#{age})
</insert>

这样我们传入的user对象无需设置id字段,Mybatis会自动进行设置处理,插入成功后id字段会回填到user对象中。

使用自定义规则生成主键

实际工作上有时候主键的生成规则并不是这么简单,比如我们取消t_user的主键自增规则。我们的要求是:

如果t_user表没有数据,我们也要设置主键为1,否则就取主键最大值加2,来设置新的主键。

对此,Mybatis也提供了处理方法,使用SelectKey元素进行处理

<insert id="insertUser" parameterType="user" useGeneratedKeys="true" keProperty="id">
 <SelectKey keProperty="id" resultType="int" order="BEFORE>
  select if(max(id) is null, 1, max(id) + 2) as newId from t_user
 </SelectKey>
 insert into t_user(user_name,age) values (#${userName},#{age})
</insert>

这样我们就可以根据自定义的规则来处理生成主键了。

MyBatis中配置主键回填

主键回填是一个很常见的需求,我们经常需要添加完数据之后,需要获取刚刚添加的数据 id,数据库主键回填在 MyBatis 中的简单实现如下:

useGeneratedKeys设置为true,keyProperty为要回填的属性名。

当插入完成后,这个对象的 id 就会被自动赋值,值就是刚刚插入成功的id。

以上为个人经验,希望能给大家一个参考,也希望大家多多支持自学编程网。

编程技巧