AI专业者开发社区 1 号成员
  •  0 回帖  •  2 浏览  •  5 个月前
   

Mybatis中进行批量更新（updateBatch）

一. 更新多条数据, 每条数据都不一样

背景描述：通常如果需要一次更新多条数据有两个方式，（1）在业务代码中循环遍历逐条更新。（2）一次性更新所有数据（更准确的说是一条 sql 语句来更新所有数据，逐条更新的操作放到数据库端，在业务代码端展现的就是一次性更新所有数据）。两种方式各有利弊，下面将会对两种方式的利弊做简要分析，主要介绍第二种方式在 mybatis 中的实现。

1. 逐条更新 (java 实现)

这种方式显然是最简单，也最不容易出错的，即便出错也只是影响到当条出错的数据，而且可以对每条数据都比较可控，更新失败或成功，从什么内容更新到什么内容，都可以在逻辑代码中获取。代码可能像下面这个样子：

updateBatch(List<MyData> datas){
    for(MyData data : datas){
        try{
            myDataDao.update(data);// 更新一条数据，mybatis 中如下面的 xml 文件的 update
        }
        catch(Exception e){
            ...// 如果更新失败可以做一些其他的操作，比如说打印出错日志等
        }
    }
}

//mybatis 中 update 操作的实现
<update>
update mydata
set   ...
where ...
</update>

这种方式最大的问题就是效率问题，逐条更新，每次都会连接数据库，然后更新，再释放连接资源（虽然通过连接池可以将频繁连接数据的效率大大提高，抗不住数据量大），这中损耗在数据量较大的时候便会体现出效率问题。这也是在满足业务需求的时候，通常会使用上述提到的第二种批量更新的实现（当然这种方式也有数据规模的限制，后面会提到）。

2. 逐条更新 (mybatis 实现)

通过循环，依次执行多条 update 的 sql

前提条件:

要实现批量更新，首先得设置mysql支持批量操作，在jdbc链接中需要附加&allowMultiQueries=true属性才行 
例如： 
jdbc:mysql://localhost:3306/dbname?characterEncoding=utf8&zeroDateTimeBehavior=convertToNull&allowMultiQueries=true


<update id="updateBatch"  parameterType="java.util.List">  
    <foreach collection="list" item="item" index="index" open="" close="" separator=";">
        update course
        <set>
            name=${item.name}
        </set>
        where id = ${item.id}
    </foreach>      
</update>

一条记录 update 一次，性能比较差，容易造成阻塞。

3.sql 批量更新 (主力实现)

(1)、实际实践 (传入的是 List<Map<String, Object>>)

务必注意: 一定要加 where 条件, 里面的 id 为需要更新的数据的 id; 如果不加 where 条件, 则会全部更新, 但是需要更新且有数据的更新为传递的数据, 没有数据的则更新为 null, 此时更新出错

<update id="updateChartParamByAccountAndChartid" parameterType="list">
        update followme_parameters
        <trim prefix="set" suffixOverrides=",">
            <trim prefix="signal_source =case" suffix="end,">
                <foreach collection="list" item="item" index="index">
                    <if test="item.signalSource!=null">
                        when account=#{item.account} and chart_id=#{item.chartId}
                         then #{item.signalSource}
                    </if>
                </foreach>
            </trim>
            <trim prefix="rate =case" suffix="end,">
                <foreach collection="list" item="item" index="index">
                    <if test="item.rate!=null">
                        when account=#{item.account} and chart_id=#{item.chartId}
                        then #{item.rate}
                    </if>
                </foreach>
            </trim>
        </trim>
        where id in
        <foreach collection="list" item="item" index="index" separator="," open="(" close=")">
            #{item.id}
        </foreach>
    </update>

另外文章的样板

<update id="updateBatch" parameterType="list">
     update course
      <trim prefix="set" suffixOverrides=",">
       <trim prefix="peopleId =case" suffix="end,">
           <foreach collection="list" item="i" index="index">
                   <if test="i.peopleId!=null">
                    when id=#{i.id} then #{i.peopleId}
                   </if>
           </foreach>
        </trim>
        <trim prefix="roadgridid =case" suffix="end,">
           <foreach collection="list" item="i" index="index">
                   <if test="i.roadgridid!=null">
                    when id=#{i.id} then #{i.roadgridid}
                   </if>
           </foreach>
        </trim>

    &lt;trim prefix=<span class="hljs-string">"type =case"</span> suffix=<span class="hljs-string">"end,"</span> &gt;
       &lt;foreach collection=<span class="hljs-string">"list"</span> item=<span class="hljs-string">"i"</span> index=<span class="hljs-string">"index"</span>&gt;
               &lt;<span class="hljs-keyword">if</span> test=<span class="hljs-string">"i.type!=null"</span>&gt;
                when id=#{i.id} then #{i.type}
               &lt;/<span class="hljs-keyword">if</span>&gt;
       &lt;/foreach&gt;
    &lt;/trim&gt;
    &lt;trim prefix=<span class="hljs-string">"unitsid =case"</span> suffix=<span class="hljs-string">"end,"</span> &gt;
        &lt;foreach collection=<span class="hljs-string">"list"</span> item=<span class="hljs-string">"i"</span> index=<span class="hljs-string">"index"</span>&gt;
                &lt;<span class="hljs-keyword">if</span> test=<span class="hljs-string">"i.unitsid!=null"</span>&gt;
                 when id=#{i.id} then #{i.unitsid}
                &lt;/<span class="hljs-keyword">if</span>&gt;
        &lt;/foreach&gt;
 &lt;/trim&gt;
&lt;/trim&gt;
where
&lt;foreach collection=<span class="hljs-string">"list"</span> separator=<span class="hljs-string">"or"</span> item=<span class="hljs-string">"i"</span> index=<span class="hljs-string">"index"</span> &gt;
    id=#{i.id}
&lt;/foreach&gt;



</update>


作者：junehappylove
来源：CSDN
原文：https://blog.csdn.net/junehappylove/article/details/82215674 
版权声明：本文为博主原创文章，转载请附上博文链接！

(2)、下面逐步讲解

一条 sql 语句来批量更新所有数据，下面直接看一下在 mybatis 中通常是怎么写的（去掉 mybatis 语法就是原生的 sql 语句了，所有就没单独说 sql 是怎么写的）。

<update id="updateBatch" parameterType="java.util.List">
    update mydata_table 
    set  status=
    <foreach collection="list" item="item" index="index" 
        separator=" " open="case ID" close="end">
        when #{item.id} then #{item.status}
    </foreach>
    where id in
    <foreach collection="list" index="index" item="item" 
        separator="," open="(" close=")">
        #{item.id,jdbcType=BIGINT}
    </foreach>
 </update>

其中 when...then... 是 sql 中的 "switch" 语法。这里借助 mybatis 的语法来拼凑成了批量更新的 sql，上面的意思就是批量更新 id 在 updateBatch 参数所传递 List 中的数据的 status 字段。还可以使用实现同样的功能, 代码如下:

<update id="updateBatch" parameterType="java.util.List">
        update mydata_table
        <trim prefix="set" suffixOverrides=",">
            <trim prefix="status =case" suffix="end,">
                <foreach collection="list" item="item" index="index">
                     when id=#{item.id} then #{item.status}
                </foreach>
            </trim>
        </trim>
        where id in
        <foreach collection="list" index="index" item="item" separator="," open="(" close=")">
            #{item.id,jdbcType=BIGINT}
        </foreach>
    </update>

<trim>

属性说明

1.prefix,suffix 表示在trim标签包裹的部分的前面或者后面添加内容 
2.如果同时有prefixOverrides,suffixOverrides 表示会用prefix,suffix 覆盖 Overrides 中的内容。 
3.如果只有prefixOverrides,suffixOverrides 表示删除开头的或结尾的xxxOverides指定的内容。

上述代码转化成 sql 如下:

update mydata_table 
    set status = 
    case
        when id = #{item.id} then #{item.status}// 此处应该是 <foreach> 展开值
        ...
    end
    where id in (...);

当然这是最简单的批量更新实现, 有时候可能需要更新多个字段, 那就需要将

<trim prefix="status =case" suffix="end,">
     <foreach collection="list" item="item" index="index">
          when id=#{item.id} then #{item.status}
     </foreach>
</trim>

复制拷贝多次, 更改 prefix 和 when...then... 的内容即可. 而如果当需要为某个字段设置默认值的时候可以使用 else

<trim prefix="status =case" suffix="end,">
     <foreach collection="list" item="item" index="index">
          when id=#{item.id} then #{item.status}
     </foreach>
     else default_value
</trim>

还有更常见的情况就是需要对要更新的数据进行判断, 只有符合条件的数据才能进行更新, 这种情况可以这么做:

<trim prefix="status =case" suffix="end,">
     <foreach collection="list" item="item" index="index">
         <if test="item.status !=null and item.status != -1">
             when id=#{item.id} then #{item.status}
         </if>
     </foreach>
</trim>

这样的话只有要更新的 list 中 status != null && status != -1 的数据才能进行 status 更新. 其他的将使用默认值更新, 而不会保持原数据不变. 如果要保持原数据不变呢? 即满足条件的更新, 不满足条件的保持原数据不变, 简单的来做就是再加一个, 因为 mybatis 中没有 if...else... 语法, 但可以通过多个实现同样的效果, 如下:

<trim prefix="status =case" suffix="end,">
     <foreach collection="list" item="item" index="index">
         <if test="item.status !=null and item.status != -1">
             when id=#{item.id} then #{item.status}
         </if>
         <if test="item.status == null or item.status == -1">
             when id=#{item.id} then mydata_table.status      // 这里就是原数据
         </if>
     </foreach>
</trim>

整体批量更新的写法如下:

<update id="updateBatch" parameterType="java.util.List">
    update mydata_table
    <trim prefix="set" suffixOverrides=",">
        <trim prefix="status =case" suffix="end,">
             <foreach collection="list" item="item" index="index">
                 <if test="item.status !=null and item.status != -1">
                     when id=#{item.id} then #{item.status}
                 </if>
                 <if test="item.status == null or item.status == -1">
                     when id=#{item.id} then mydata_table.status// 原数据
                 </if>
             </foreach>
        </trim>
    </trim>
    where id in
    <foreach collection="list" index="index" item="item" separator="," open="(" close=")">
        #{item.id,jdbcType=BIGINT}
    </foreach>
</update>

4. 批量更新 (单个字段, 传参 list), 实际是 sql 批量更新的简化版本而已

(1)、单个字段方法 1

<update id="updateByBatch" parameterType="java.util.List">
    update t_goods
    set NODE_ID=
    <foreach collection="list" item="item" index="index"
             separator=" " open="case" close="end">
      when GOODS_ID=#{item.goodsId} then #{item.nodeId}
    </foreach>
    where GOODS_ID in
    <foreach collection="list" index="index" item="item"
             separator="," open="(" close=")">
      #{item.goodsId,jdbcType=BIGINT}
    </foreach>
  </update>

(2)、单个字段方法 2

<update id="updateByBatch" parameterType="java.util.List">
    UPDATE
    t_goods
    SET NODE_ID = CASE
    <foreach collection="list" item="item" index="index">
      WHEN GOODS_ID = #{item.goodsId} THEN #{item.nodeId}
    </foreach>
    END
    WHERE GOODS_ID IN
    <foreach collection="list" index="index" item="item" open="(" separator="," close=")">
      #{item.goodsId}
    </foreach>
  </update>

以上单字段更新实际执行：

UPDATE t_goods SET NODE_ID = CASE WHEN GOODS_ID = ? THEN ? END WHERE GOODS_ID IN ( ? )

5.sql 批量更新 (通过 insert 实现)

传入的是 List<Map<String,Object>>

直接运行插入, 如果有插入的数据转为更新该条数据

<insert id="updateChartParamByAccountAndChartid">
    insert into followme_parameters
    (account,chart_id,signal_source,rate)
    values
    <foreach collection="list" separator="," index="index" item="item">
        (#{item.account},#{item.chartId},#{item.signalSource},#{item.rate})
    </foreach>
    ON duplicate KEY UPDATE
    signal_source=values(signal_source),rate=values(rate) 
</insert>

二. 更新多条数据, 更新的内容一样.

1. 传 map/ 传 String

NODE_ID 从 map 中取出来,goodsIdList 是字符串拼接好的 (如下面的 "1,2,5")

<update id="updateByBatchPrimaryKey" parameterType="java.util.Map">
    UPDATE t_goods
    SET NODE_ID = #{nodeId}
    WHERE GOODS_ID IN (${goodsIdList})
  </update>

实际的 sql

UPDATE t_goods SET NODE_ID = ? WHERE GOODS_ID IN (1,2,5);

2. 传 map/ 传 list

NODE_ID 从 map 中取出来,goodsIdList 是用 list 拼接出来的

<update id="updateByBatchPrimaryKey" parameterType="java.util.Map">
    UPDATE t_goods
    SET NODE_ID = #{nodeId}
    WHERE GOODS_ID IN 
    <foreach collection="list" index="index" item="item" open="(" separator="," close=")">
      #{item.goodsId}
    </foreach>
</update>

实际的 sql

UPDATE t_goods SET NODE_ID = ? WHERE GOODS_ID IN (1,2,5);

参考文章:

主力:https://blog.csdn.net/xyjawq1/article/details/74129316/
辅助:https://www.jianshu.com/p/041bec8ae6d3