ไฮเบอร์เนต - การประมวลผลแบทช์

พิจารณาสถานการณ์เมื่อคุณต้องการอัปโหลดบันทึกจำนวนมากไปยังฐานข้อมูลของคุณโดยใช้ไฮเบอร์เนต ต่อไปนี้เป็นข้อมูลโค้ดเพื่อให้บรรลุสิ่งนี้โดยใช้ไฮเบอร์เนต -

Session session = SessionFactory.openSession();
Transaction tx = session.beginTransaction();
for ( int i=0; i<100000; i++ ) {
   Employee employee = new Employee(.....);
   session.save(employee);
}
tx.commit();
session.close();

ตามค่าเริ่มต้น Hibernate จะแคชวัตถุที่ยังคงอยู่ทั้งหมดในแคชระดับเซสชันและท้ายที่สุดแอปพลิเคชันของคุณจะล้มลงด้วย OutOfMemoryExceptionที่ไหนสักแห่งแถวที่ 50,000 คุณสามารถแก้ไขปัญหานี้ได้หากคุณใช้batch processing ด้วย Hibernate

ในการใช้คุณสมบัติการประมวลผลชุดงานให้ตั้งค่าก่อน hibernate.jdbc.batch_sizeเป็นขนาดแบทช์เป็นตัวเลขที่ 20 หรือ 50 ขึ้นอยู่กับขนาดวัตถุ สิ่งนี้จะบอกคอนเทนเนอร์ไฮเบอร์เนตว่าทุกแถว X จะถูกแทรกเป็นแบตช์ ในการติดตั้งโค้ดของคุณเราจะต้องทำการปรับเปลี่ยนเล็กน้อยดังนี้ -

Session session = SessionFactory.openSession();
Transaction tx = session.beginTransaction();
for ( int i=0; i<100000; i++ ) {
   Employee employee = new Employee(.....);
   session.save(employee);
   if( i % 50 == 0 ) { // Same as the JDBC batch size
      //flush a batch of inserts and release memory:
      session.flush();
      session.clear();
   }
}
tx.commit();
session.close();

โค้ดด้านบนจะทำงานได้ดีสำหรับการดำเนินการ INSERT แต่ถ้าคุณยินดีที่จะทำการ UPDATE คุณสามารถทำได้โดยใช้รหัสต่อไปนี้ -

Session session = sessionFactory.openSession();
Transaction tx = session.beginTransaction();

ScrollableResults employeeCursor = session.createQuery("FROM EMPLOYEE").scroll();
int count = 0;

while ( employeeCursor.next() ) {
   Employee employee = (Employee) employeeCursor.get(0);
   employee.updateEmployee();
   seession.update(employee); 
   if ( ++count % 50 == 0 ) {
      session.flush();
      session.clear();
   }
}
tx.commit();
session.close();

ตัวอย่างการประมวลผลแบทช์

ให้เราแก้ไขไฟล์คอนฟิกเพื่อเพิ่ม hibernate.jdbc.batch_size คุณสมบัติ -

<?xml version = "1.0" encoding = "utf-8"?>
<!DOCTYPE hibernate-configuration SYSTEM 
"http://www.hibernate.org/dtd/hibernate-configuration-3.0.dtd">

<hibernate-configuration>
   <session-factory>
   
      <property name = "hibernate.dialect">
         org.hibernate.dialect.MySQLDialect
      </property>
   
      <property name = "hibernate.connection.driver_class">
         com.mysql.jdbc.Driver
      </property>

      <!-- Assume students is the database name -->
   
      <property name = "hibernate.connection.url">
         jdbc:mysql://localhost/test
      </property>
   
      <property name = "hibernate.connection.username">
         root
      </property>
   
      <property name = "hibernate.connection.password">
         root123
      </property>
   
      <property name = "hibernate.jdbc.batch_size">
         50
      </property>

      <!-- List of XML mapping files -->
      <mapping resource = "Employee.hbm.xml"/>

   </session-factory>
</hibernate-configuration>

พิจารณาระดับพนักงาน POJO ต่อไปนี้ -

public class Employee {
   private int id;
   private String firstName; 
   private String lastName;   
   private int salary;  

   public Employee() {}
   
   public Employee(String fname, String lname, int salary) {
      this.firstName = fname;
      this.lastName = lname;
      this.salary = salary;
   }
   
   public int getId() {
      return id;
   }
   
   public void setId( int id ) {
      this.id = id;
   }
   
   public String getFirstName() {
      return firstName;
   }
   
   public void setFirstName( String first_name ) {
      this.firstName = first_name;
   }
   
   public String getLastName() {
      return lastName;
   }
   
   public void setLastName( String last_name ) {
      this.lastName = last_name;
   }
   
   public int getSalary() {
      return salary;
   }
   
   public void setSalary( int salary ) {
      this.salary = salary;
   }
}

ให้เราสร้างตาราง EMPLOYEE ต่อไปนี้เพื่อเก็บวัตถุพนักงาน -

create table EMPLOYEE (
   id INT NOT NULL auto_increment,
   first_name VARCHAR(20) default NULL,
   last_name  VARCHAR(20) default NULL,
   salary     INT  default NULL,
   PRIMARY KEY (id)
);

ต่อไปนี้จะเป็นไฟล์การแมปเพื่อแมปวัตถุพนักงานกับตารางพนักงาน -

<?xml version = "1.0" encoding = "utf-8"?>
<!DOCTYPE hibernate-mapping PUBLIC 
"-//Hibernate/Hibernate Mapping DTD//EN"
"http://www.hibernate.org/dtd/hibernate-mapping-3.0.dtd"> 

<hibernate-mapping>
   <class name = "Employee" table = "EMPLOYEE">
      
      <meta attribute = "class-description">
         This class contains the employee detail. 
      </meta>
      
      <id name = "id" type = "int" column = "id">
         <generator class="native"/>
      </id>
      
      <property name = "firstName" column = "first_name" type = "string"/>
      <property name = "lastName" column = "last_name" type = "string"/>
      <property name = "salary" column = "salary" type = "int"/>
      
   </class>
</hibernate-mapping>

สุดท้ายเราจะสร้างคลาสแอปพลิเคชันของเราด้วยเมธอด main () เพื่อเรียกใช้แอปพลิเคชันที่เราจะใช้ flush() และ clear() วิธีการที่มีอยู่ในวัตถุเซสชันเพื่อให้ Hibernate เขียนบันทึกเหล่านี้ลงในฐานข้อมูลแทนที่จะแคชไว้ในหน่วยความจำ

import java.util.*; 
 
import org.hibernate.HibernateException; 
import org.hibernate.Session; 
import org.hibernate.Transaction;
import org.hibernate.SessionFactory;
import org.hibernate.cfg.Configuration;

public class ManageEmployee {
   private static SessionFactory factory; 
   public static void main(String[] args) {
      
      try {
         factory = new Configuration().configure().buildSessionFactory();
      } catch (Throwable ex) { 
         System.err.println("Failed to create sessionFactory object." + ex);
         throw new ExceptionInInitializerError(ex); 
      }
      ManageEmployee ME = new ManageEmployee();

      /* Add employee records in batches */
      ME.addEmployees( );
   }
   
   /* Method to create employee records in batches */
   public void addEmployees( ){
      Session session = factory.openSession();
      Transaction tx = null;
      Integer employeeID = null;
      
      try {
         tx = session.beginTransaction();
         for ( int i=0; i<100000; i++ ) {
            String fname = "First Name " + i;
            String lname = "Last Name " + i;
            Integer salary = i;
            Employee employee = new Employee(fname, lname, salary);
            session.save(employee);
         	if( i % 50 == 0 ) {
               session.flush();
               session.clear();
            }
         }
         tx.commit();
      } catch (HibernateException e) {
         if (tx!=null) tx.rollback();
         e.printStackTrace(); 
      } finally {
         session.close(); 
      }
      return ;
   }
}

การรวบรวมและการดำเนินการ

นี่คือขั้นตอนในการรวบรวมและเรียกใช้แอปพลิเคชันที่กล่าวถึงข้างต้น ตรวจสอบให้แน่ใจว่าคุณได้ตั้งค่า PATH และ CLASSPATH อย่างเหมาะสมก่อนดำเนินการรวบรวมและดำเนินการ

• สร้างไฟล์กำหนดค่า hibernate.cfg.xml ตามที่อธิบายไว้ข้างต้น

• สร้างไฟล์การแมป Employee.hbm.xml ดังที่แสดงด้านบน

• สร้างไฟล์ซอร์ส Employee.java ตามที่แสดงด้านบนและคอมไพล์

• สร้างไฟล์ต้นฉบับ ManageEmployee.java ตามที่แสดงด้านบนและคอมไพล์

• ดำเนินการไบนารี ManageEmployee เพื่อรันโปรแกรมซึ่งจะสร้าง 100000 เรกคอร์ดในตาราง EMPLOYEE