天天看点

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>
           

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