/**
* Copyright 2011-2012 eBusiness Information, Groupe Excilys (www.excilys.com)
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package com.excilys.ebi.spring.dbunit;
import static com.excilys.ebi.spring.dbunit.utils.DbUnitUtils.getDatabaseConnection;
import static org.slf4j.LoggerFactory.getLogger;
import java.io.IOException;
import java.sql.BatchUpdateException;
import java.sql.Connection;
import java.sql.SQLException;
import org.dbunit.DatabaseUnitException;
import org.dbunit.database.DatabaseConnection;
import org.dbunit.dataset.IDataSet;
import org.dbunit.dataset.ReplacementDataSet;
import org.dbunit.operation.DatabaseOperation;
import org.slf4j.Logger;
import org.springframework.beans.factory.annotation.Required;
import org.springframework.jdbc.datasource.init.DatabasePopulator;
import org.springframework.util.StopWatch;
import com.excilys.ebi.spring.dbunit.config.DataSetConfiguration;
import com.excilys.ebi.spring.dbunit.config.Phase;
import com.excilys.ebi.spring.dbunit.dataset.DataSetDecorator;
public class DbUnitDatabasePopulator implements DatabasePopulator {
private static final Logger LOGGER = getLogger(DbUnitDatabasePopulator.class);
private DataSetConfiguration dataSetConfiguration;
private Phase phase;
@Override
public void populate(Connection connection) throws SQLException {
LOGGER.debug("populating");
StopWatch sw = new StopWatch("DbUnitDatabasePopulator");
DatabaseOperation operation = phase.getOperation(dataSetConfiguration);
try {
IDataSet dataSet = decorateDataSetIfNeeded(dataSetConfiguration.getDataSet(), dataSetConfiguration.getDecorators());
String schema = dataSetConfiguration.getSchema();
DatabaseConnection databaseConnection = getDatabaseConnection(connection, schema, dataSetConfiguration);
sw.start("populating");
operation.execute(databaseConnection, dataSet);
sw.stop();
LOGGER.debug(sw.prettyPrint());
} catch (BatchUpdateException e) {
LOGGER.error("BatchUpdateException while loading dataset", e);
LOGGER.error("Caused by : ", e.getNextException());
throw e;
} catch (DatabaseUnitException e) {
throw new DbUnitException(e);
} catch (IOException e) {
throw new DbUnitException(e);
}
}
private IDataSet decorateDataSetIfNeeded(IDataSet dataSet, Class<? extends DataSetDecorator>[] decorators) {
if (decorators == null || decorators.length == 0)
return dataSet;
ReplacementDataSet decoratedSet = new ReplacementDataSet(dataSet);
for (Class<? extends DataSetDecorator> decoratorClass : decorators) {
try {
DataSetDecorator decorator = decoratorClass.newInstance();
decoratedSet.addReplacementSubstring(decorator.getStringToReplace(), decorator.getStringReplacement());
} catch (InstantiationException e) {
LOGGER.error("Could not instantiate DataSetDecorator {}" + decoratorClass, e);
} catch (IllegalAccessException e) {
LOGGER.error("Could not instantiate DataSetDecorator {}" + decoratorClass, e);
}
}
return decoratedSet;
}
public DataSetConfiguration getDataSetConfiguration() {
return dataSetConfiguration;
}
public Phase getPhase() {
return phase;
}
@Required
public void setDataSetConfiguration(DataSetConfiguration dataSetConfiguration) {
this.dataSetConfiguration = dataSetConfiguration;
}
@Required
public void setPhase(Phase phase) {
this.phase = phase;
}
}