/*
* 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.test;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import org.hibernate.Hibernate;
import org.hibernate.SQLQuery;
import org.hibernate.Session;
import org.hibernate.SessionFactory;
import org.hibernate.cfg.Configuration;
import org.hibernate.dialect.Dialect;
import org.hibernate.dialect.MySQLDialect;
import org.hibernate.engine.SessionFactoryImplementor;
import org.hibernate.mapping.ForeignKey;
import org.hibernate.mapping.Table;
import org.jbpm.api.ProcessEngine;
import org.jbpm.internal.log.Log;
/**
* @author Tom Baeyens
*/
public class Db {
private static final Log log = Log.getLog(Db.class.getName());
static Map<ProcessEngine, String[]> cleanSqlCache = new HashMap<ProcessEngine, String[]>();
static Map<ProcessEngine, String[]> tableNamesCache = new HashMap<ProcessEngine, String[]>();
public static void clean(ProcessEngine processEngine) {
SessionFactory sessionFactory = processEngine.get(SessionFactory.class);
// when running this with a remote ejb invocation configuration, there is no
// session factory and no cleanup needs to be done
if (sessionFactory==null) {
return;
}
String[] cleanSql = cleanSqlCache.get(processEngine);
if (cleanSql == null) {
Configuration configuration = processEngine.get(Configuration.class);
SessionFactoryImplementor sessionFactoryImplementor = (SessionFactoryImplementor) sessionFactory;
Dialect dialect = sessionFactoryImplementor.getDialect();
// loop over all foreign key constraints
List<String> dropForeignKeysSql = new ArrayList<String>();
List<String> createForeignKeysSql = new ArrayList<String>();
Iterator<Table> iter = configuration.getTableMappings();
//if no session-factory is build, the configuration is not fully initialized.
//Hence, the ForeignKey's won't have a referenced table. This is calculated on
//second pass.
configuration.buildMappings();
while (iter.hasNext()) {
Table table = (Table) iter.next();
if (table.isPhysicalTable()) {
String catalog = table.getCatalog();
String schema = table.getSchema();
Iterator<ForeignKey> subIter = table.getForeignKeyIterator();
while (subIter.hasNext()) {
ForeignKey fk = (ForeignKey) subIter.next();
if (fk.isPhysicalConstraint()) {
// collect the drop foreign key constraint sql
dropForeignKeysSql.add(fk.sqlDropString(dialect, catalog, schema));
// MySQLDialect creates an index for each foreign key.
// see
// http://opensource.atlassian.com/projects/hibernate/browse/HHH-2155
// This index should be dropped or an error will be thrown during
// the creation phase
if (dialect instanceof MySQLDialect) {
dropForeignKeysSql.add("alter table " + table.getName() + " drop key " + fk.getName());
}
// and collect the create foreign key constraint sql
createForeignKeysSql.add(fk.sqlCreateString(dialect, sessionFactoryImplementor, catalog, schema));
}
}
}
}
List<String> deleteSql = new ArrayList<String>();
iter = configuration.getTableMappings();
while (iter.hasNext()) {
Table table = (Table) iter.next();
if (table.isPhysicalTable()) {
deleteSql.add("delete from " + table.getName());
}
}
// glue
// - drop foreign key constraints
// - delete contents of all tables
// - create foreign key constraints
// together to form the clean script
List<String> cleanSqlList = new ArrayList<String>();
cleanSqlList.addAll(dropForeignKeysSql);
cleanSqlList.addAll(deleteSql);
cleanSqlList.addAll(createForeignKeysSql);
cleanSql = (String[]) cleanSqlList.toArray(new String[cleanSqlList.size()]);
cleanSqlCache.put(processEngine, cleanSql);
}
Session session = sessionFactory.openSession();
try {
for (String query : cleanSql) {
// log.trace(query);
session.createSQLQuery(query).executeUpdate();
}
} finally {
session.close();
}
}
public static String verifyClean(ProcessEngine processEngine) {
SessionFactory sessionFactory = processEngine.get(SessionFactory.class);
// when running this with a remote ejb invocation configuration, there is no
// session factory and no cleanup needs to be done
if (sessionFactory==null) {
return null;
}
String[] tableNames = tableNamesCache.get(processEngine);
if (tableNames == null) {
Configuration configuration = processEngine.get(Configuration.class);
// loop over all foreign key constraints
List<String> tableNamesList = new ArrayList<String>();
Iterator iter = configuration.getTableMappings();
while (iter.hasNext()) {
Table table = (Table) iter.next();
if (table.isPhysicalTable()) {
tableNamesList.add(table.getName());
}
}
tableNames = tableNamesList.toArray(new String[tableNamesList.size()]);
tableNamesCache.put(processEngine, tableNames);
}
String recordsLeftMsg = "";
Session session = sessionFactory.openSession();
try {
for (String tableName : tableNames) {
String countSql = "select count(*) as RECORD_COUNT_ from "+tableName;
SQLQuery sqlQuery = session.createSQLQuery(countSql);
sqlQuery.addScalar("RECORD_COUNT_", Hibernate.LONG);
Long recordCount = (Long) sqlQuery.uniqueResult();
if (recordCount>0L) {
recordsLeftMsg += tableName+":"+recordCount+", ";
}
}
} finally {
session.close();
}
if (recordsLeftMsg.length()>0) {
clean(processEngine);
}
return recordsLeftMsg;
}
}