数据库
首页 > 数据库> > 关于 JPA PostgreSQL中使用 GenerationType.IDENTITY 批处理失效

关于 JPA PostgreSQL中使用 GenerationType.IDENTITY 批处理失效

作者:互联网

@NoRepositoryBean
public interface BatchRepository<T, ID extends Serializable> extends JpaRepository<T, ID> {
        
    <S extends T> void saveInBatch(Iterable<S> entites);
}
@Transactional(readOnly = true)
public class BatchRepositoryImpl<T, ID extends Serializable>
        extends SimpleJpaRepository<T, ID> implements BatchRepository<T, ID> {

    private static final Logger logger = Logger.getLogger(BatchRepositoryImpl.class.getName());

    private final EntityManager entityManager;

    public BatchRepositoryImpl(JpaEntityInformation entityInformation,
            EntityManager entityManager) {
        super(entityInformation, entityManager);

        this.entityManager = entityManager;
    }

    @Override
    @Transactional
    public <S extends T> void saveInBatch(Iterable<S> entities) {

        if (entities == null) {
            throw new IllegalArgumentException("The given Iterable of entities cannot be null!");
        }

        int i = 0;
        for (S entity : entities) {
            entityManager.persist(entity);

            i++;

            // Flush a batch of inserts and release memory
            if (i % batchSize() == 0 && i > 0) {
                logger.log(Level.INFO,
                        "Flushing the EntityManager containing {0} entities ...", i);

                entityManager.flush();
                entityManager.clear();
                i = 0;
            }
        }

        if (i > 0) {
            logger.log(Level.INFO,
                    "Flushing the remaining {0} entities ...", i);

            entityManager.flush();
            entityManager.clear();
        }
    }

    private static int batchSize() {

        int batchsize = Integer.valueOf(Dialect.DEFAULT_BATCH_SIZE); // default batch size

        Properties configuration = new Properties();
        try (InputStream inputStream = BatchRepositoryImpl.class.getClassLoader()
                .getResourceAsStream("application.properties")) {
            configuration.load(inputStream);
        } catch (IOException ex) {
            logger.log(Level.SEVERE,
                    "Cannot fetch batch size. Using further Dialect.DEFAULT_BATCH_SIZE{0}", ex);
            return batchsize;
        }

        String batchsizestr = configuration.getProperty(
                "spring.jpa.properties.hibernate.jdbc.batch_size");
        if (batchsizestr != null) {
            batchsize = Integer.valueOf(batchsizestr);
        }

        return batchsize;
    }
}

 

Why To Avoid PostgreSQL (BIG)SERIAL In Batching Inserts Via Hibernate

https://github.com/AnghelLeonard/Hibernate-SpringBoot/tree/master/HibernateSpringBootBatchingAndSerial

自增列 EntityManager (MySQL)

https://github.com/AnghelLeonard/Hibernate-SpringBoot/tree/master/HibernateSpringBootBatchInsertsEntityManagerBatchPerTransaction

标签:entityManager,PostgreSQL,JPA,BatchRepositoryImpl,entities,EntityManager,logger,G
来源: https://www.cnblogs.com/Irving/p/14855248.html