这篇文章将为大家详细讲解有关JPA如何批量保存saveAll,小编觉得挺实用的,因此分享给大家做个参考,希望大家阅读完这篇文章后可以有所收获。
使用了JPA的saveAll方法批量保存一千多条数据的问题,但是发现日志打印却是一条一条的insert语句。
查看saveAll的源码,发现里面是使用了一个for循环然后一条条的执行save方法....
好吧,手动写一个批量执行的方法
@Componentpublic class DbUtils { private static EntityManager entityManager; public static final int BATCH_SIZE = 1000; public static EntityManager getEntityManager() { if (entityManager == null) { entityManager = SpringUtils.getBean(EntityManager.class); } return entityManager; } public static void setEntityManager(EntityManager entityManager) { //DbUtils.entityManager = entityManager; } public static void batchExecute(String sql, List<Object[]> params, Integer batchSize){ if (batchSize == null) { batchSize = BATCH_SIZE; } if(params.size() == 0){ return; } Connection conn = null; Session session = null; PreparedStatement stmt = null; try { session = (Session) getEntityManager().getDelegate(); SessionFactoryImpl sf = (SessionFactoryImpl)session.getSessionFactory(); conn = sf.getJdbcServices().getBootstrapJdbcConnectionAccess().obtainConnection(); stmt = conn.prepareStatement(sql); conn.setAutoCommit(false); int index = 0; for (Object[] objects : params) { for(int i=0 ; i<objects.length ; i++){ stmt.setObject(i+1, objects[i]); } stmt.addBatch(); index++; if (index % batchSize == 0){ stmt.executeBatch(); stmt.clearBatch(); } } if (index % batchSize != 0){ stmt.executeBatch(); stmt.clearBatch(); } conn.commit(); }catch (Exception e){ throw new RuntimeException(e.getMessage()); }finally { try { stmt.close(); conn.close(); }catch (SQLException e){ throw new RuntimeException(e.getMessage()); } } }}
关于“JPA如何批量保存saveAll”这篇文章就分享到这里了,希望以上内容可以对大家有一定的帮助,使各位可以学到更多知识,如果觉得文章不错,请把它分享出去让更多的人看到。