JPA如何批量保存saveAll

這篇文章將為大家詳細講解有關JPA如何批量保存saveAll，小編覺得挺實用的，因此分享給大家做個參考，希望大家閱讀完這篇文章后可以有所收獲。

使用了JPA的saveAll方法批量保存一千多條數據的問題，但是發現日志打印卻是一條一條的insert語句。

查看saveAll的源碼，發現里面是使用了一個for循環然后一條條的執行save方法....

好吧，手動寫一個批量執行的方法

@Component
public 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”這篇文章就分享到這里了，希望以上內容可以對大家有一定的幫助，使各位可以學到更多知識，如果覺得文章不錯，請把它分享出去讓更多的人看到。