Batch Insert Support 批量插入
在开发中如果遇到需要批量insert的需求,可以使用Mybatis 的 Batch Insert Support 提高插入效率。
代码实例(开发的项目中截取的片段)
@Autowired private SqlSessionTemplate sqlSessionTemplate; public int insertFolder(List<IpsCatalogFolderDetail> ips) { //获取sql会话 SqlSession session = sqlSessionTemplate.getSqlSessionFactory().openSession(ExecutorType.BATCH, false); //通过新的session获取mapper,而不是常规的spring管理注入 IipsCatalogFolderDetailDao folderDetailDao = session.getMapper(IipsCatalogFolderDetailDao.class); int size = ips.size(); //如果有父类子类两层都需要批量插入也可 try { //外层循环 for (int i = 0; i < size; i ) { ips.get(i).setType("folder"); //用上面在session中获取的mapper进行插入操作 folderDetailDao.insertFolder(ips.get(i)); //内层循环 String cs = ips.get(i).getContentIds(); if (StringUtils.isNotBlank(cs)){ List<String> con = JSON.parseArray(cs,String.class); if (cs != null && con.size() > 0) { for (int j = 0; j < con.size(); j ) { IpsCatalogFolderDetail ifd = new IpsCatalogFolderDetail(); ifd.setParentCode(ips.get(i).getCode()); ifd.setContentId(con.get(j)); ifd.setType("contents"); //同样用上面在session中获取的mapper进行插入操作 folderDetailDao.insertFolder(ifd); } } } //最后批量提交 if (i % 200 == 0 || i == size - 1) { session.commit();//200个提交一次,手动提交,提交后无法回滚 session.clearCache(); //清理缓存,防止溢出 } } }catch (Exception e) { System.out.println(e.toString()); session.rollback(); //没有提交的数据可以回滚 } finally { session.close(); } return 0; }
另外有时我们在插入的时候需要先查询数据是否已存在,如果也需要批量操作可将insert和update语句合并,然后就可以继续使用Batch Insert了
ORACLE数据库sql示例
@Insert("merge into ips_catalog_folder_detail fd " "using(select #{code,jdbcType=VARCHAR} c from dual)t " "on(fd.FOLDERID = t.c)" "when matched then" "update set " ...(省略)... "where ..." "when not matched then insert(" "fd.PROD_LINE," "fd.TYPE," "fd.PARENTFOLDERCODE," "fd.FOLDERID," "fd.FOLDERCODE," "fd.FOLDERNAME," "fd.COLUMN_SORTINDEX," "fd.DESCRIPTION," "fd.CONTENTID," "fd.CREATETIME" ")" "VALUES" "(" "#{prod_line}," "#{type,jdbcType=VARCHAR}," "#{parentCode,jdbcType=VARCHAR}," "#{code,jdbcType=VARCHAR}," "#{aliasCode,jdbcType=VARCHAR}," "#{name,jdbcType=VARCHAR}," "#{sortIndex,jdbcType=VARCHAR}," "#{desc,jdbcType=VARCHAR}," "#{contentId,jdbcType=VARCHAR}," "#{createTime,jdbcType=VARCHAR}" ")") int insertFolder(IpsCatalogFolderDetail fd);
MYSQL示例:
REPLACE INTO users (id,name,age) VALUES(1, '张雨绮', 32);
批量插入几千条数据优化(foreach)
项目中有一个耗时较长的Job存在CPU占用过高的问题
经排查发现,主要时间消耗在往MyBatis中批量插入数据。mapper configuration是用foreach循环做的,差不多是这样。
<insert id="batchInsert" parameterType="java.util.List"> insert into USER (id, name) values <foreach collection="list" item="model" index="index" separator=","> (#{model.id}, #{model.name}) </foreach> </insert>
优化代码
可以看 http://www.mybatis.org/mybatis-dynamic-sql/docs/insert.html 中 Batch Insert Support 标题里的内容)
SqlSession session = sqlSessionFactory.openSession(ExecutorType.BATCH); try { SimpleTableMapper mapper = session.getMapper(SimpleTableMapper.class); List<SimpleTableRecord> records = getRecordsToInsert(); // not shown BatchInsert<SimpleTableRecord> batchInsert = insert(records) .into(simpleTable) .map(id).toProperty("id") .map(firstName).toProperty("firstName") .map(lastName).toProperty("lastName") .map(birthDate).toProperty("birthDate") .map(employed).toProperty("employed") .map(occupation).toProperty("occupation") .build() .render(RenderingStrategy.MYBATIS3); batchInsert.insertStatements().stream().forEach(mapper::insert); session.commit(); } finally { session.close(); }
总结一下,如果MyBatis需要进行批量插入,推荐使用 ExecutorType.BATCH 的插入方式,如果非要使用 <foreach>的插入的话,需要将每次插入的记录控制在 20~50 左右
以上为个人经验,希望能给大家一个参考,也希望大家多多支持Devmax。