import org.dbunit.dataset.FilteredDataSet; //导入依赖的package包/类
@Scheduled(cron = "0 0 3 * * *")
public void run() throws SQLException, DatabaseUnitException, IOException {
    if (enable) {
        log.info("begin backup database");

        String bak = root + "/" + new SimpleDateFormat("yyyyMMddHHmmss").format(new Date());

        ITableFilter filter = new DatabaseSequenceFilter(conn);
        IDataSet dataSet = new FilteredDataSet(filter, conn.createDataSet());

        //IDataSet dataSet = conn.createDataSet();
        FlatXmlDataSet.write(dataSet, new FileOutputStream(bak + ".xml"));
        FlatDtdDataSet.write(dataSet, new FileOutputStream(bak + ".dtd"));
        log.info("end backup database");
    }


}
 

import org.dbunit.dataset.FilteredDataSet; //导入依赖的package包/类
/**
 * Helper method to dump the current database.
 *
 * @param fileName the file where to store.
 * @param tableNames the name of the tables.
 * @throws Exception
 */
public void dumpResult(String fileName, String... tableNames) {
    final FuWeStaSampleBuilderDataSetWriter writer =
            new FuWeStaSampleBuilderDataSetWriter(
                    "de.ppi.samples.fuwesta.dbunit.dataset", fileName);
    try {
        if (tableNames != null && tableNames.length > 0) {
            writer.write(new FilteredDataSet(tableNames, databaseTester
                    .getConnection().createDataSet()));
        } else {
            writer.write(deleteDataSet);
        }
    } catch (Exception e) {
        throw new IllegalStateException("Error during dump result", e);
    }

}
 

import org.dbunit.dataset.FilteredDataSet; //导入依赖的package包/类
/**
 * Generates the actual content of the DTD file as an in-memory string.
 *
 * @return the DTD content, not null
 */
protected String generateDtdContent() {
    Connection conn = null;
    try {
        conn = sqlHandler.getDataSource().getConnection();
        IDatabaseConnection dbUnitDatabaseConnection = new DatabaseConnection(conn, defaultDbSupport.getSchemaName());

        StringWriter stringWriter = new StringWriter();

        FlatDtdWriter datasetWriter = new FlatDtdWriter(stringWriter);
        datasetWriter.setContentModel(FlatDtdWriter.CHOICE);

        // create a dataset for the database content
        // filter out all system table names
        Set<String> tableNames = defaultDbSupport.getTableNames();
        IDataSet actualDataSet = dbUnitDatabaseConnection.createDataSet();
        IDataSet filteredActualDataSet = new FilteredDataSet(new IncludeTableFilter(tableNames.toArray(new String[0])), actualDataSet);

        datasetWriter.write(filteredActualDataSet);
        return stringWriter.toString();

    } catch (Exception e) {
        throw new UnitilsException("Error generating content for DTD file.", e);
    } finally {
        DbUtils.closeQuietly(conn);
    }
}
 

import org.dbunit.dataset.FilteredDataSet; //导入依赖的package包/类
protected IDataSet getFilteredDataset(IDataSet dataset)
        throws Exception
{
    IDatabaseConnection conn = null;

    try
    {
        conn = getConnection();

        long start = System.currentTimeMillis();
        if (filter == null)
        {
            filter = new SequenceTableFilter(sortedTableNames); //new DatabaseSequenceFilter(conn);
        }
        long end = System.currentTimeMillis();

        //System.err.println("TOOK:" + (end - start));

        IDataSet newds = new FilteredDataSet(filter, dataset);

        return newds;
    }
    catch (Exception e)
    {
        throw e;
    }
}
 

import org.dbunit.dataset.FilteredDataSet; //导入依赖的package包/类
public DependencyAwareDataset(IDataSet dataSet,
		IDatabaseConnection databaseConnection) throws DataSetException,
		SQLException {
	DatabaseSequenceFilter databaseSequenceFilter = new DatabaseSequenceFilter(
			databaseConnection);
	this.dataSet = new FilteredDataSet(databaseSequenceFilter, dataSet);

}
 

import org.dbunit.dataset.FilteredDataSet; //导入依赖的package包/类
public IDataSet createDataSet() throws GoalExecutionException {
	IDatabaseConnection connection = richDriverConfig.createConnection();
	try {
		String[] dependentTables = exportDataSet.getDependentTables();
		IDataSet dataSet = null;
		if (dependentTables == null) {
			dataSet = connection.createDataSet();
		} else {
			String[] depTableNames = TablesDependencyHelper
					.getAllDependentTables(connection, dependentTables);
			dataSet = connection.createDataSet(depTableNames);
		}

		String[] tableNames = dataSet.getTableNames();
		List<String> filteredTableNameList = new ArrayList<String>(
				Arrays.asList(tableNames));

		String[] excludeTables = exportDataSet.getExcludeTables();
		filteredTableNameList.removeAll(Arrays.asList(excludeTables));

		String[] filteredTableNames = (String[]) filteredTableNameList
				.toArray(new String[filteredTableNameList.size()]);

		FilteredDataSet filteredDataSet = new FilteredDataSet(
				filteredTableNames, dataSet);
		return filteredDataSet;
	} catch (Exception e) {
		try {
			connection.close();
		} catch (SQLException e1) {
		}
		throw new GoalExecutionException("Unable to create data set", e);
	}
}
 

import org.dbunit.dataset.FilteredDataSet; //导入依赖的package包/类
private IDataSet excludeTables(final IDataSet dataSet, final String... tablesToExclude) {
    return new FilteredDataSet(new ExcludeTableFilter(tablesToExclude), dataSet);
}