hibernate 批量插入

一般如果要插入100万条数据，则会写如下代码：

[java] view plain copy

1. package org.xiazdong.test;  
2.   
3. import junit.framework.TestCase;  
4.   
5. import org.hibernate.Session;  
6. import org.hibernate.SessionFactory;  
7. import org.hibernate.Transaction;  
8. import org.hibernate.cfg.Configuration;  
9. import org.junit.Test;  
10. import org.xiazdong.Person;  
11.   
12. public class PersonTest extends TestCase{  
13.     @Test  
14.     public void testAdd(){  
15.         long begin = System.currentTimeMillis();  
16.         Configuration config = new Configuration();  
17.         config.configure();  
18.         SessionFactory factory = config.buildSessionFactory();  
19.         Session session = factory.openSession();   
20.         Transaction tx = session.beginTransaction();     
21.         for ( int i=0; i<1000000; i++ ) { //插入100万条数据  
22.             Person person = new Person("xiazdong-"+i,i+20);   
23.                 session.save(person);   
24.         }     
25.         tx.commit();   
26.         session.close();  
27.         long end = System.currentTimeMillis();  
28.         System.out.println((end-begin)/1000.0);  
29.     }  
30. }  

此程序正常来说是会报错：

 

这是因为Hibernate默认会把没有提交的数据全部缓存，但是缓存是不能存入100万条数据的，因此会报错，我们怎么样能够解决呢？

我们可以通过每100个insert语句时，就强制刷入数据库，实现代码如下：

[java] view plain copy

1. package org.xiazdong.test;  
2.   
3. import junit.framework.TestCase;  
4.   
5. import org.hibernate.Session;  
6. import org.hibernate.SessionFactory;  
7. import org.hibernate.Transaction;  
8. import org.hibernate.cfg.Configuration;  
9. import org.junit.Test;  
10. import org.xiazdong.Person;  
11.   
12. public class PersonTest extends TestCase{  
13.     @Test  
14.     public void testAdd(){  
15.         long begin = System.currentTimeMillis();  
16.         Configuration config = new Configuration();  
17.         config.configure();  
18.         SessionFactory factory = config.buildSessionFactory();  
19.         Session session = factory.openSession();   
20.         Transaction tx = session.beginTransaction();     
21.         for ( int i=0; i<1000000; i++ ) { //插入100万条数据  
22.             Person person = new Person("xiazdong-"+i,i+20);   
23.             session.save(person);   
24.             if(i%100==0){   //每一千条刷新并写入数据库  
25.                 session.flush();  
26.                 session.clear();  
27.             }  
28.         }  
29.         tx.commit();   
30.         session.close();  
31.         long end = System.currentTimeMillis();  
32.         System.out.println((end-begin)/1000.0);  
33.     }  
34. }  

经过测试，花费时间为：148秒；

注：如果数据库在程序执行过程中记录数为0，是因为数据库为read commited，事务提交之后才能够数据。

 

当然，同样可以使用Hibernate配置来实现同样功能：

在hibernate.cfg.xml中插入：

[html] view plain copy

1. <property name="hibernate.jdbc.batch_size">50</property>//每50条语句提交一次   
2. <property name="hiberante.cache.use_second_level_cache">false</property>//关闭二级缓存