主键回填的两种方法

业务场景

主键回填听上去很高级，不妨用一个业务场景来引入概念吧。

在增加操作中，如果设置了主键id是自增的，那么插入数据时id的值一般写null，因为存入数据库时id都会自动+1，看似操作没毛病，非常完美。

但是在购物交易的添加模块这可不兴这样，细想一下在上网买东西时，用户付完钱之后一般都会返回一个订单编号，其实这个订单编号在数据库中就是主键，但是因为在插入数据时id的值写了null，所以订单编号也为null，也就是说返回不了订单编号的数据。

那还是老老实实挨个挨个在插入数据的时候写id呗，如果是批量添加数据呢，面对千万级的插入数据，这显然是不科学滴~

那要不先插入数据再进行查询操作吧，保证能查到数据，emm...有点麻烦，可以但没必要，有需求就有解决方案，开发者早就想到了，我们只要坐享其成就行啦~

主键回填就是用来干这个的，讲完这个业务场景在来思考一下这个专有名词，"主键"+"回填"，谜底就在谜面上，作用就是把插入数据的null主键给填回去，完美实现插入数据为null主键的同时可以查询到主键信息的需求~~

主键回填是什么？

主键回填一般用于增加操作中，把插入数据时插入为null的主键id数据填回去，存入到java对象和主键对应的属性中（数据库主键字段为id则回填的是实体类的id属性），实现添加+查询主键一步到位。

主键回填其实是一个非常常见的需求，特别是在数据添加的过程中，我们经常需要添加完数据之后，需要获取刚刚添加的数据 id，无论是 Jdbc 还是各种各样的数据库框架都对此提供了相关的支持，本文我就来和和大家分享下数据库主键回填在 MyBatis 中的两种实现思路。

主键回填的两种方法

方式一

这种方式比较简单，就是在插入节点上添加useGeneratedKeys属性，同时设置接收回传主键的属性。配置完成后，我们执行一个插入操作，插入时传入一个对象，插入完成后，这个对象的 id 就会被自动赋值，值就是刚刚插入成功的id。

sql 复制代码

<!--useGeneratedKeys="true":使用数据库生成的主键-->
<!--keyProperty="id":返回的主键回填给pojo的id属性-->
<insert id="add" useGeneratedKeys="true" keyProperty="id" parameterType="com.by.pojo.CheckGroup">
<!--sql语句-->
</insert>

方式二

第二种方式则是利用MySQL自带的last_insert_id()函数查询刚刚插入的id，示例代码如下：

sql 复制代码

<insert id="insertBook">
    <selectKey keyProperty="id" resultType="java.lang.Integer">
        SELECT LAST_INSERT_ID()
    </selectKey>
    insert into t_book (b_name,author) values (#{name},#{author});
</insert>

这种方式是在insert节点中添加selectKey来实现主键回填，实际上这种方式的功能更加丰富，因为 selectKey 节点中的 SQL 我们既可以在插入之前执行，也可以在插入之后执行（通过设置节点的Order属性为AFTER或者BEFORE可以实现），具体什么时候执行，还是要看具体的需求，如果是做主键回填，我们当然需要在插入 SQL 执行之后执行selectKey节点中的 SQL。

注意第二种方式一样也要通过设置keyProperty来指定将查询到的数据绑定到哪个属性上。