hibernatetesthelper.java

来自「一个java工作流引擎」· Java 代码 · 共 48 行

package org.jbpm.persistence.hibernate;

import java.util.*;

import org.jbpm.*;
import net.sf.hibernate.*;

/**
 * creates hibernate objects based on the configurations found in 
 * the file core/src/test/config/jbpm.properties.  The
 * created session factory is cached.
 */
public class HibernateTestHelper {
  private static HibernateJdbcSessionFactory hibernateSessionFactory = null;
  private Session session = null;
  private Transaction transaction = null;
	private JbpmConfiguration jbpmConfiguration = null;
  
  public HibernateTestHelper() {
  }
  
  public HibernateTestHelper( JbpmConfiguration jbpmConfiguration ) {
    this.jbpmConfiguration = jbpmConfiguration;
  }
  
  public void startTransaction() throws HibernateException {
    if ( hibernateSessionFactory == null ) {
      if ( jbpmConfiguration == null ) {
        Properties hibernateProperties = TestHelper.getProperties( "jbpm.properties" );
        jbpmConfiguration = new JbpmConfiguration( hibernateProperties );
      }
      hibernateSessionFactory = new HibernateJdbcSessionFactory( jbpmConfiguration );
    }
    session = hibernateSessionFactory.getSessionFactory().openSession();
		transaction = session.beginTransaction();
  }
  
  public void commitTransaction() throws HibernateException {
    session.flush();
    transaction.commit();
    session.close();
  }
  
  public Session getSession() {
    return session; 
  }
}