Hibernate大数据量操作解决方案

 

Hibernate大数据量操作解决方案

阅读了Hibernate的Reference之后，可以采用批量处理的方法，当插入的数据超过10000时，就flush session并且clear。

下面是一个测试method。

 1 /** */ /**
 2      * 测试成批插入数据的事务处理，返回是否成功
 3      * 
 4      *  @param  objPO Object
 5      *  @return  boolean
 6       */
 7      public   boolean  insertBatch( final  Object objPO)  {
 8          boolean  isSuccess  =   false ;
 9         Transaction transaction  =   null ;
10         Session session  =  openSession();
11          try   {
12             transaction  =  session.beginTransaction();
13              for  ( int  i  =   0 ; i  <   100000 ; i ++ )  {
14                 session.save(objPO);
15                  if  (i  %   50   ==   0 )  {
16                      //  flush a batch of inserts and release memory
17                     session.flush();
18                     session.clear();
19                 }
20             }
21             transaction.commit();
22             logger.info( " transaction.wasCommitted: "
23                      +  transaction.wasCommitted());
24             isSuccess  =   true ;
25         }   catch  (HibernateException ex)  {
26              if  (transaction  !=   null )  {
27                  try   {
28                     transaction.rollback();
29                     logger.error( " transaction.wasRolledBack: "
30                              +  transaction.wasRolledBack());
31                 }   catch  (HibernateException ex1)  {
32                     logger.error(ex1.getMessage());
33                     ex1.printStackTrace();
34                 }
35             }
36             logger.error( " Insert Batch PO Error: "   +  ex.getMessage());
37             ex.printStackTrace();
38         }   finally   {
39              if  (transaction  !=   null )  {
40                 transaction  =   null ;
41             }
42             session.close();
43         }
44          return  isSuccess;
45     }
46

这只是简单的测试，实际项目中遇到的问题，要比这个复杂得多。
这时候，我们可以让Spring来控制Transaction，自己来控制Hibernate的Session，随时更新数据。
首先，利用HibernateDaoSupport类来自定义个方法打开Session；

1public Session openSession() {
2        
3        return getHibernateTemplate().getSessionFactory().openSession();
4        
5    }

然后，用打开的Session处理你的数据；

 1protected void doBusiness(Session session)  {
 2
 3        while (true) {
 4            //do your business with the opening session        
 5            someMethod(session);
 6            session.flush();
 7            session.clear();
 8            logger.info("good job!");            
 9        }       
10}

每做一次数据操作，就更新一次Session，这样可以保证每次数据操作都成功，否则就让Spring去控制它roll back吧。

最后，记得关闭Session。

1 Session session  =  openSession();
2 doBusiness(session);
3 session.close();  // 关闭session

至于处理速度呢，不会慢到哪儿去的，已经试验过了。; - )

还等什么，赶快试试吧！