快捷導(dǎo)航

Mybatis批量更新對象數(shù)據(jù)的兩種實現(xiàn)方式

更新時間：2024年08月05日 09:28:02 作者：何中應(yīng)

這篇文章主要介紹了Mybatis批量更新對象數(shù)據(jù)的兩種實現(xiàn)方式,具有很好的參考價值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教

創(chuàng)建Demo

首先，創(chuàng)建一個簡單的Demo；

（User，用戶對象）

import lombok.Data;
import java.io.Serializable;

@Data
public class User implements Serializable {
    
    private String id;

    private String username;

    private String password;
}

（UserController，用戶控制器）

@RestController
@RequestMapping("/user")
public class UserController {

    @Autowired
    private UserMapper userMapper;

    /**
     * 根據(jù)ID查詢用戶
     * @param id
     * @return
     */
    @GetMapping("/getUser/{id}")
    public String getUser(@PathVariable("id") String id){
        return userMapper.getUser(id).toString();
    }
}

（UserMapper，用戶數(shù)據(jù)訪問接口）

import com.hezy.pojo.User;
import org.apache.ibatis.annotations.Mapper;

@Mapper
public interface UserMapper {

    User getUser(String id);
}

數(shù)據(jù)庫數(shù)據(jù)，tb_user表

項目啟動，測試一下接口，沒得問題；

批量更新

創(chuàng)建一個新的接口，用于觸發(fā)批量更新對象的代碼，內(nèi)容是傳遞一個List集合對象給Mapper處理；

    @GetMapping("/updateUser")
    public void updateUser(){
        ArrayList<User> users = new ArrayList<>();

        User user1 = new User();
        user1.setId("1");
        user1.setUsername("zhangsan_fix");

        User user2 = new User();
        user2.setId("2");
        user2.setUsername("lisi_fix");
        user2.setPassword("654321_fix");

        users.add(user1);
        users.add(user2);
        userMapper.updateUser(users);
    }

接下來，重點是Mapper.xml中的Statement要怎么寫，繼續(xù)往下看

方式一

首先，我們可以使用動態(tài)SQL，如下：

    <update id="updateUser">
        <foreach collection="users" item="user" separator=";">
            update tb_user
            <set>
                <if test="user.username != null and user.username != ''">
                    username = #{user.username},
                </if>
                <if test="user.password != null and user.password != ''">
                    password = #{user.password}
                </if>
            </set>
            where id = #{user.id}
        </foreach>
    </update>

我們把拼接后的SQL打印出來，會發(fā)現(xiàn)一個問題。SQL中，分號（;）表示一條語句的結(jié)束，使用上面的方式拼接出來的方式，是多條SQL。

因此產(chǎn)生了一個問題，Mybatis中一個Statement標簽中可以有多條SQL嗎？

答案是默認不可以，所以上面的代碼報錯了。

需要在數(shù)據(jù)庫連接后面加上&allowMultiQueries=true配置，如下：

再次執(zhí)行，修改成功了，這是第一種方式；

查看日志發(fā)現(xiàn)，Updates：1，就是說如果需要返回更新的記錄條數(shù)，那么這種方式返回的更新條數(shù)會是1，不能體現(xiàn)出數(shù)據(jù)庫真實發(fā)生變化的記錄條數(shù)；

另外，將多條SQL合成一條執(zhí)行，有SQL注入的風(fēng)險

方式二

第二種方式是用一條SQL的方式來實現(xiàn)，如下：

	<update id="updateUser">
	    update tb_user
	    set
	    
	    username = case
	    <foreach collection="users" item="user">
	        when id = #{user.id}
	        <choose>
	            <when test="user.username != null and user.username != ''">then #{user.username}</when>
	            <otherwise>then username</otherwise>
	        </choose>
	    </foreach>
	    end,
	
	    password = case
	    <foreach collection="users" item="user">
	        when id = #{user.id}
	        <choose>
	            <when test="user.password != null and user.password != ''">then #{user.password}</when>
	            <otherwise>then password</otherwise>
	        </choose>
	    </foreach>
	    end
	
	    where
	    <foreach collection="users" item="user" separator="or">
	        id = #{user.id}
	    </foreach>
	</update>

看著有些復(fù)雜，拼接后的SQL如下：

	update tb_user
	set
	    username = case
	                   when id = '1' then 'zhangsan_fix'
	                   when id = '2' then 'lisi_fix'
	    end,
	
	    password = case
	                   when id = '1' then password
	                   when id = '2' then '654321_fix'
	    end
	where id = '1'
	   or id = '2';

這種方式不需要加額外的配置，執(zhí)行測試；