dbpersistenceservicefactory.java

jBpm是一个灵活可扩展的工作流管理系统。作为jBpm运行时server输入的业务流程使用简单强大的语言表达并打包在流程档案中

/*
 * JBoss, Home of Professional Open Source
 * Copyright 2005, JBoss Inc., and individual contributors as indicated
 * by the @authors tag. See the copyright.txt in the distribution for a
 * full listing of individual contributors.
 *
 * This is free software; you can redistribute it and/or modify it
 * under the terms of the GNU Lesser General Public License as
 * published by the Free Software Foundation; either version 2.1 of
 * the License, or (at your option) any later version.
 *
 * This software is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this software; if not, write to the Free
 * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
 * 02110-1301 USA, or see the FSF site: http://www.fsf.org.
 */
package org.jbpm.persistence.db;

import javax.sql.DataSource;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.hibernate.SessionFactory;
import org.hibernate.cfg.Configuration;
import org.hibernate.tool.hbm2ddl.SchemaExport;
import org.jbpm.JbpmConfiguration;
import org.jbpm.db.hibernate.HibernateHelper;
import org.jbpm.svc.Service;
import org.jbpm.svc.ServiceFactory;
import org.jbpm.util.JndiUtil;

public class DbPersistenceServiceFactory implements ServiceFactory {

  private static final long serialVersionUID = 1L;

  Configuration configuration = null;

  String sessionFactoryJndiName = null;
  SessionFactory sessionFactory = null;
  
  String dataSourceJndiName = null;
  DataSource dataSource = null;

  boolean isTransactionEnabled = true;

  SchemaExport schemaExport = null;

  public Service openService() {
    log.debug("creating persistence service");
    return new DbPersistenceService(this);
  }

  public synchronized Configuration getConfiguration() {
    if (configuration==null) {
      String hibernateCfgXmlResource = null;
      if (JbpmConfiguration.Configs.hasObject("resource.hibernate.cfg.xml")) {
        hibernateCfgXmlResource = JbpmConfiguration.Configs.getString("resource.hibernate.cfg.xml");
      }
      String hibernatePropertiesResource = null;
      if (JbpmConfiguration.Configs.hasObject("resource.hibernate.properties")) {
        hibernatePropertiesResource = JbpmConfiguration.Configs.getString("resource.hibernate.properties");
      }
      configuration = HibernateHelper.createConfiguration(hibernateCfgXmlResource, hibernatePropertiesResource);
    }
    return configuration;
  }

  public synchronized SchemaExport getSchemaExport() {
    if (schemaExport==null) {
      log.debug("creating schema export");
      schemaExport = new SchemaExport(getConfiguration());
    }
    return schemaExport;
  }

  public synchronized SessionFactory getSessionFactory() {
    if (sessionFactory==null) {

      if (sessionFactoryJndiName!=null) {
        log.debug("looking up hibernate session factory in jndi '"+sessionFactoryJndiName+"'");
        sessionFactory = (SessionFactory) JndiUtil.lookup(sessionFactoryJndiName, SessionFactory.class);
        
      } else {
        log.debug("building hibernate session factory");
        sessionFactory = getConfiguration().buildSessionFactory();
      }
    }
    return sessionFactory;
  }
  
  public DataSource getDataSource() {
    if ( (dataSource==null)
         && (dataSourceJndiName!=null)
       ) {
      log.debug("looking up datasource from jndi location '"+dataSourceJndiName+"'");
      dataSource = (DataSource) JndiUtil.lookup(dataSourceJndiName, DataSource.class);
    }
    return dataSource;
  }
  
  public void createSchema() {
    getSchemaExport().create(getScript(), true);
    HibernateHelper.clearHibernateCache(getSessionFactory());
  }
  
  public void dropSchema() {
    HibernateHelper.clearHibernateCache(getSessionFactory());
    getSchemaExport().drop(getScript(), true);
  }

  public void close() {
    if (sessionFactory!=null) {
      log.debug("closing hibernate session factory");
      sessionFactory.close();
    }
  }

  boolean getScript() {
    boolean script = false;
    String showSql = getConfiguration().getProperty("hibernate.show_sql");
    if ("true".equalsIgnoreCase(showSql)) {
      script = true;
    }
    return script;
  }

  public String getDataSourceJndiName() {
    return dataSourceJndiName;
  }
  public void setDataSourceJndiName(String dataSourceJndiName) {
    this.dataSourceJndiName = dataSourceJndiName;
  }
  public String getSessionFactoryJndiName() {
    return sessionFactoryJndiName;
  }
  public void setSessionFactoryJndiName(String sessionFactoryJndiName) {
    this.sessionFactoryJndiName = sessionFactoryJndiName;
  }
  public void setConfiguration(Configuration configuration) {
    this.configuration = configuration;
  }
  public void setDataSource(DataSource dataSource) {
    this.dataSource = dataSource;
  }
  public void setSchemaExport(SchemaExport schemaExport) {
    this.schemaExport = schemaExport;
  }
  public void setSessionFactory(SessionFactory sessionFactory) {
    this.sessionFactory = sessionFactory;
  }
  public boolean isTransactionEnabled() {
    return isTransactionEnabled;
  }
  public void setTransactionEnabled(boolean isTransactionEnabled) {
    this.isTransactionEnabled = isTransactionEnabled;
  }
  
  private static Log log = LogFactory.getLog(DbPersistenceServiceFactory.class);
}