作為CRUD程序員,大部分JAVA開發者應該都在用MyBatis Plus來操作數據庫。但是BaseMApper默認僅提供了int insert(T entity)這個單條插入的方法。那么我們想批量插入數據該怎么辦呢?
在以前用Mybatis的時候,我們會在Mapper.xml里面去寫foreach循環:
|
|
如果批量的實體對象較多的話,我們就會寫很多這樣的Mapper,如果表字段較多,這工作量的也不小的,當然,如果你用了代碼生成工具,那就當我沒說。Mybatis Plus作為Mybatis的增強版,也為我們考慮到了這個問題。使用Mybatis Plus批量插入數據有兩種方式,第一種是Service層繼承IService,第二種便是insertBatchSomeColumn。先來看看IService的方式
- 新建Mapper繼承BaseMapper<T>
public interface UserMapper extends BaseMapper<UserDO> {} - 新建一個Service接口,繼承IService<T>
public interface UserService extends IService<UserDO> { void batchAddUser();} - 新建一個Service實現類,繼承ServiceImpl<M,T>
public class UserServiceImpl extends ServiceImpl<UserMapper,UserDO> implements UserService { @Autowired private UserMapper userMapper; @Override public void batchAddUser() { List<UserDO> userDOList=new ArrayList<>(); UserDO user1=new UserDO(); user1.setAge(99); user1.setName("99_1"); userDOList.add(user1); UserDO user2=new UserDO(); user2.setAge(999); user2.setName("999_1"); userDOList.add(user2); this.saveBatch(userDOList,2); }}
這種方式,會生成多條INSERT語句一條條執行,然后按設置的batchSize來提交事務。
我們再來看看insertBatchSomeColumn是怎么做的呢。
1.新建一個InsertBatchInjector類來擴展DefaultSqlInjector
|
|
2.新建一個Mapper接口,繼承BaseMapper<T>并添加insertBatchSomeColumn方法,
|
|
3.在要批量寫的地方直接調用該Mapper的insertBatchSomeColumn即可
|
|
這種方式則是拼接成一條SQL,一次性提交到數據庫執行
這里,SQL語句的長度受MySQL服務端的max_allowed_packet參數限制,默認為1M,如果拼接的字符串過長則會報錯??梢赃M來減少批量提交的記錄,或者是調整max_allowed_packet參數的大小。除了受到max_allowed_packet的影響,innodb_buffer_pool_size的大小也會影響批量的記錄數大小,innodb_buffer_pool_size的默認值為128M,我們可以根據情況適當調整,合理地配置innodb_buffer_pool_size參數,可以提高MySQL的讀寫性能。