Batch inserts (including associations)

Author: AnghelLeonard

HibernateSpringBootBatchInsertOrder/pom.xml

@@ -4,17 +4,17 @@
     <modelVersion>4.0.0</modelVersion>
 
     <groupId>com.jpa</groupId>
-    <artifactId>jpa</artifactId>
+    <artifactId>HibernateSpringBootBatchInsertOrder</artifactId>
     <version>1.0</version>
     <packaging>jar</packaging>
 
-    <name>jpa</name>
+    <name>HibernateSpringBootBatchInsertOrder</name>
     <description>JPA project for Spring Boot</description>
 
     <parent>
         <groupId>org.springframework.boot</groupId>
         <artifactId>spring-boot-starter-parent</artifactId>
-        <version>2.0.5.RELEASE</version>
+        <version>2.1.4.RELEASE</version>
         <relativePath/> <!-- lookup parent from repository -->
     </parent>
 
@@ -37,12 +37,7 @@
         <dependency>
             <groupId>org.springframework.boot</groupId>
             <artifactId>spring-boot-starter-web</artifactId>
-        </dependency>
-        <dependency>
-            <groupId>net.ttddyy</groupId>
-            <artifactId>datasource-proxy</artifactId>
-            <version>${datasource-proxy.version}</version>
-        </dependency>
+        </dependency>        
         <dependency>
             <groupId>mysql</groupId>
             <artifactId>mysql-connector-java</artifactId>

HibernateSpringBootBatchInsertOrder/src/main/java/com/bookstore/MainApplication.java

@@ -0,0 +1,31 @@
+package com.bookstore;
+
+import com.bookstore.impl.BatchRepositoryImpl;
+import com.bookstore.service.BookstoreService;
+import org.springframework.boot.ApplicationRunner;
+import org.springframework.boot.SpringApplication;
+import org.springframework.boot.autoconfigure.SpringBootApplication;
+import org.springframework.context.annotation.Bean;
+import org.springframework.data.jpa.repository.config.EnableJpaRepositories;
+
+@SpringBootApplication
+@EnableJpaRepositories(repositoryBaseClass = BatchRepositoryImpl.class)
+public class MainApplication {
+
+    private final BookstoreService authorService;
+
+    public MainApplication(BookstoreService authorService) {
+        this.authorService = authorService;
+    }
+
+    public static void main(String[] args) {
+        SpringApplication.run(MainApplication.class, args);
+    }
+
+    @Bean
+    public ApplicationRunner init() {
+        return args -> {
+            authorService.batchAuthorsAndBooks();
+        };
+    }
+}

HibernateSpringBootBatchInsertOrder/src/main/java/com/bookstore/entity/Author.java

@@ -0,0 +1,95 @@
+package com.bookstore.entity;
+
+import java.io.Serializable;
+import java.util.ArrayList;
+import java.util.Iterator;
+import java.util.List;
+import javax.persistence.CascadeType;
+import javax.persistence.Entity;
+import javax.persistence.Id;
+import javax.persistence.OneToMany;
+
+@Entity
+public class Author implements Serializable {
+
+    private static final long serialVersionUID = 1L;
+
+    @Id   
+    private Long id;
+
+    private String name;
+    private String genre;
+    private int age;
+
+    @OneToMany(cascade = CascadeType.ALL,
+            mappedBy = "author", orphanRemoval = true)
+    private List<Book> books = new ArrayList<>();
+
+    public void addBook(Book book) {
+        this.books.add(book);
+        book.setAuthor(this);
+    }
+
+    public void removeBook(Book book) {
+        book.setAuthor(null);
+        this.books.remove(book);
+    }
+
+    public void removeBooks() {
+        Iterator<Book> iterator = this.books.iterator();
+
+        while (iterator.hasNext()) {
+            Book book = iterator.next();
+
+            book.setAuthor(null);
+            iterator.remove();
+        }
+    }
+
+    public Long getId() {
+        return id;
+    }
+
+    public void setId(Long id) {
+        this.id = id;
+    }
+
+    public String getName() {
+        return name;
+    }
+
+    public void setName(String name) {
+        this.name = name;
+    }
+
+    public String getGenre() {
+        return genre;
+    }
+
+    public void setGenre(String genre) {
+        this.genre = genre;
+    }  
+
+    public int getAge() {
+        return age;
+    }
+
+    public void setAge(int age) {
+        this.age = age;
+    }
+
+    public List<Book> getBooks() {
+        return books;
+    }
+
+    public void setBooks(List<Book> books) {
+        this.books = books;
+    }
+
+    @Override
+    public String toString() {
+        return "Author{" + "id=" + id + ", name=" + name 
+                + ", genre=" + genre + ", age=" + age + '}';
+    }
+    
+}

HibernateSpringBootBatchInsertOrder/src/main/java/com/bookstore/entity/Book.java

@@ -0,0 +1,81 @@
+package com.bookstore.entity;
+
+import java.io.Serializable;
+import javax.persistence.Entity;
+import javax.persistence.FetchType;
+import javax.persistence.Id;
+import javax.persistence.JoinColumn;
+import javax.persistence.ManyToOne;
+
+@Entity
+public class Book implements Serializable {
+
+    private static final long serialVersionUID = 1L;
+
+    @Id   
+    private Long id;
+
+    private String title;
+    private String isbn;
+
+    @ManyToOne(fetch = FetchType.LAZY)
+    @JoinColumn(name = "author_id")
+    private Author author;
+
+    public Long getId() {
+        return id;
+    }
+
+    public void setId(Long id) {
+        this.id = id;
+    }
+
+    public String getTitle() {
+        return title;
+    }
+
+    public void setTitle(String title) {
+        this.title = title;
+    }
+
+    public String getIsbn() {
+        return isbn;
+    }
+
+    public void setIsbn(String isbn) {
+        this.isbn = isbn;
+    }
+
+    public Author getAuthor() {
+        return author;
+    }
+
+    public void setAuthor(Author author) {
+        this.author = author;
+    }
+
+    @Override
+    public boolean equals(Object obj) {
+        
+        if (this == obj) {
+            return true;
+        }
+        
+        if (getClass() != obj.getClass()) {
+            return false;
+        }
+        
+        return id != null && id.equals(((Book) obj).id);
+    }
+
+    @Override
+    public int hashCode() {
+        return 2021;
+    }
+
+    @Override
+    public String toString() {
+        return "Book{" + "id=" + id + ", title=" + title + ", isbn=" + isbn + '}';
+    }
+    
+}

HibernateSpringBootBatchInsertOrder/src/main/java/com/bookstore/impl/BatchRepository.java

@@ -0,0 +1,12 @@
+package com.bookstore.impl;
+
+import java.io.Serializable;
+import org.springframework.data.jpa.repository.JpaRepository;
+import org.springframework.data.repository.NoRepositoryBean;
+
+@NoRepositoryBean
+public interface BatchRepository<T, ID extends Serializable> extends JpaRepository<T, ID> {
+    
+    <S extends T> S persist(S entity);
+    <S extends T> Iterable<S> saveInBatch(Iterable<S> entites);
+}

HibernateSpringBootBatchInsertOrder/src/main/java/com/bookstore/impl/BatchRepositoryImpl.java

@@ -0,0 +1,99 @@
+package com.bookstore.impl;
+
+import java.io.IOException;
+import java.io.InputStream;
+import java.io.Serializable;
+import java.util.ArrayList;
+import java.util.List;
+import java.util.Properties;
+import java.util.logging.Level;
+import java.util.logging.Logger;
+import javax.persistence.EntityManager;
+import org.hibernate.dialect.Dialect;
+import org.springframework.data.jpa.repository.support.JpaEntityInformation;
+import org.springframework.data.jpa.repository.support.SimpleJpaRepository;
+import org.springframework.transaction.annotation.Transactional;
+
+@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> S persist(S entity) {
+        entityManager.persist(entity);
+
+        return entity;
+    }
+
+    @Override
+    @Transactional
+    public <S extends T> Iterable<S> saveInBatch(Iterable<S> entities) {
+
+        if (entities == null) {
+            throw new IllegalArgumentException("The given Iterable of entities cannot be null!");
+        }
+
+        int i = 0;
+        List<S> result = new ArrayList<>();
+        for (S entity : entities) {
+            result.add(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();
+        }
+
+        return result;
+    }
+
+    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;
+    }
+}

HibernateSpringBootBatchInsertOrder/src/main/java/com/bookstore/repository/AuthorRepository.java

@@ -0,0 +1,10 @@
+package com.bookstore.repository;
+
+import com.bookstore.impl.BatchRepository;
+import com.bookstore.entity.Author;
+import org.springframework.stereotype.Repository;
+
+@Repository
+public interface AuthorRepository extends BatchRepository<Author, Long> {        
+}
+