Mybatis中updateBatch實現批次更新

2022-03-11 10:00:03

一.更新多條資料,每條資料都不一樣

背景描述：通常如果需要一次更新多條資料有兩個方式，（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>

        <trim prefix="type =case" suffix="end," >
           <foreach collection="list" item="i" index="index">
                   <if test="i.type!=null">
                    when id=#{i.id} then #{i.type}
                   </if>
           </foreach>
        </trim>
        <trim prefix="unitsid =case" suffix="end," >
            <foreach collection="list" item="i" index="index">
                    <if test="i.unitsid!=null">
                     when id=#{i.id} then #{i.unitsid}
                    </if>
            </foreach>
     </trim>
    </trim>
    where
    <foreach collection="list" separator="or" item="i" index="index" >
        id=#{i.id}
    </foreach>
</update>

(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>