dbunit mysql bug no_dbunit经典的NoSuchColumnException解决之道

package com.ridge.test.unitils.ext;

import org.dbunit.database.DatabaseConfig;

import org.dbunit.database.DefaultMetadataHandler;

import org.dbunit.dataset.DataSetException;

import org.dbunit.dataset.IDataSet;

import org.dbunit.dataset.datatype.IDataTypeFactory;

import org.dbunit.dataset.filter.ITableFilterSimple;

import org.dbunit.ext.db2.Db2DataTypeFactory;

import org.dbunit.ext.db2.Db2MetadataHandler;

import org.dbunit.ext.mysql.MySqlDataTypeFactory;

import org.dbunit.ext.mysql.MySqlMetadataHandler;

import org.unitils.core.UnitilsException;

import org.unitils.core.dbsupport.DbSupport;

import org.unitils.core.dbsupport.DefaultSQLHandler;

import org.unitils.core.dbsupport.SQLHandler;

import org.unitils.dbunit.DbUnitModule;

import org.unitils.dbunit.util.DbUnitDatabaseConnection;

import javax.sql.DataSource;

import static org.dbunit.database.DatabaseConfig.FEATURE_BATCHED_STATEMENTS;

import static org.dbunit.database.DatabaseConfig.PROPERTY_DATATYPE_FACTORY;

import static org.dbunit.database.DatabaseConfig.PROPERTY_ESCAPE_PATTERN;

import static org.unitils.core.dbsupport.DbSupportFactory.getDbSupport;

import static org.unitils.core.util.ConfigUtils.getInstanceOf;

/**

* @author : chenxh(quickselect@163.com)

* @date: 13-10-9

*/

public class MyDbunitModule extends DbUnitModule {

protected DbUnitDatabaseConnection createDbUnitConnection(String schemaName) {

// A DbSupport instance is fetched in order to get the schema name in correct case

DataSource dataSource = getDatabaseModule().getDataSourceAndActivateTransactionIfNeeded();

SQLHandler sqlHandler = new DefaultSQLHandler(dataSource);

DbSupport dbSupport = getDbSupport(configuration, sqlHandler, schemaName);

// Create connection

DbUnitDatabaseConnection connection = new DbUnitDatabaseConnection(dataSource, dbSupport.getSchemaName());

DatabaseConfig config = connection.getConfig();

// Make sure that dbunit's correct IDataTypeFactory, that handles dbms specific data type issues, is used

IDataTypeFactory dataTypeFactory = getInstanceOf(IDataTypeFactory.class, configuration, dbSupport.getDatabaseDialect());

config.setProperty(PROPERTY_DATATYPE_FACTORY, dataTypeFactory);

// Make sure that table and column names are escaped using the dbms-specific identifier quote string

if (dbSupport.getIdentifierQuoteString() != null)

config.setProperty(PROPERTY_ESCAPE_PATTERN, dbSupport.getIdentifierQuoteString() + '?' + dbSupport.getIdentifierQuoteString());

// Make sure that batched statements are used to insert the data into the database

config.setProperty(FEATURE_BATCHED_STATEMENTS, "true");

// Make sure that Oracle's recycled tables (BIN$) are ignored (value is used to ensure dbunit-2.2 compliancy)

config.setProperty("http://www.dbunit.org/features/skipOracleRecycleBinTables", "true");

//注意这儿：根据不同的数据库(unitils的database.dialect配置参数)为dbunit

//指定使用不同的IMetadataHandler实现(其它数据库都可以用默认的，还有一个Netezza也是特别的，这里忽略了)

if("db2".equalsIgnoreCase(configuration.getProperty("database.dialect"))){

config.setProperty(DatabaseConfig.PROPERTY_DATATYPE_FACTORY,

new Db2DataTypeFactory());

//由于dbunit自身提供的Db2MetadataHandler有BUG，所以这里使用自己写的

//MyDb2MetadataHandler,源码在后面了。

config.setProperty(DatabaseConfig.PROPERTY_METADATA_HANDLER,

new MyDb2MetadataHandler());

}else if("mysql".equalsIgnoreCase(configuration.getProperty("database.dialect"))){

config.setProperty(DatabaseConfig.PROPERTY_DATATYPE_FACTORY,

new MySqlDataTypeFactory());

config.setProperty(DatabaseConfig.PROPERTY_METADATA_HANDLER,

new MySqlMetadataHandler());

}

return connection;

}

}