Hibernate - 批量处理
考虑一种情况，你需要使用 Hibernate 将大量记录上传到数据库中。以下是使用 Hibernate 实现此目的的代码片段 −
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 会将所有持久化对象缓存到 session 级别的缓存中，最终你的应用程序会在大约第 50,000 行时抛出 OutOfMemoryException。如果你使用 Hibernate 的 batch processing，就可以解决这个问题。
要使用批量处理功能，首先将 hibernate.jdbc.batch_size 设置为 20 或 50（取决于对象大小）。这将告诉 Hibernate 容器每 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 ) { // 与 JDBC batch size 相同
//刷新一批插入操作并释放内存：
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>
<!-- 假设 students 是数据库名称 -->
<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>
<!-- XML 映射文件列表 -->
<mapping resource = "Employee.hbm.xml"/>
</session-factory>
</hibernate-configuration>
考虑以下 POJO Employee class −
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 表来存储 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) );
以下是映射文件，用于将 Employee 对象映射到 EMPLOYEE 表 −
<?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">
此类包含员工详细信息。
</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() 方法的应用程序类来运行应用程序，在其中我们将使用 Session 对象提供的 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();
/* 批量添加员工记录 */
ME.addEmployees( );
}
/* 批量创建员工记录的方法 */
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 可执行文件运行程序，该程序将在 EMPLOYEE 表中创建 100000 条记录。