dbunit 用来做数据库数据的导入导出

比较有用的文档地址：

http://www.dbunit.org/faq.html#streaming

http://www.dbunit.org/properties.html#qualifiedtablenames

http://www.dbunit.org/components.html#streamingdataset

工具类

java 代码

package dbunit;

import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.IOException;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.SQLException;
import java.util.HashSet;
import java.util.Iterator;
import java.util.Map;
import java.util.Set;

import org.dbunit.DatabaseUnitException;
import org.dbunit.database.DatabaseConfig;
import org.dbunit.database.DatabaseConnection;
import org.dbunit.database.ForwardOnlyResultSetTableFactory;
import org.dbunit.database.IDatabaseConnection;
import org.dbunit.database.QueryDataSet;
import org.dbunit.dataset.CachedDataSet;
import org.dbunit.dataset.DataSetException;
import org.dbunit.dataset.IDataSet;
import org.dbunit.dataset.stream.IDataSetProducer;
import org.dbunit.dataset.stream.StreamingDataSet;
import org.dbunit.dataset.xml.FlatXmlDataSet;
import org.dbunit.dataset.xml.FlatXmlProducer;
import org.dbunit.dataset.xml.XmlDataSet;
import org.dbunit.dataset.xml.XmlProducer;
import org.dbunit.operation.DatabaseOperation;
import org.xml.sax.InputSource;

public abstract class DBXMLHelper {

    // partial database export
    public static void exportPartial(IDatabaseConnection connection,
            String fileName, Set tableNames, Map sqls, boolean streamed,
            boolean flat) throws SQLException, DataSetException,
            FileNotFoundException, IOException {
        connection.getConfig().setFeature(
                DatabaseConfig.FEATURE_QUALIFIED_TABLE_NAMES, true);
        if (streamed)
            connection.getConfig().setProperty(
                    DatabaseConfig.PROPERTY_RESULTSET_TABLE_FACTORY,
                    new ForwardOnlyResultSetTableFactory());

        QueryDataSet partialDataSet = new QueryDataSet(connection);
        // all data
        if (null != tableNames)
            for (Iterator it = tableNames.iterator(); it.hasNext();) {
                partialDataSet.addTable((String) it.next());
            }
        // sql query
        if (null != sqls)
            for (Iterator it = sqls.keySet().iterator(); it.hasNext();) {
                String key = (String) it.next();
                partialDataSet.addTable(key, (String) sqls.get(key));
            }

        if (flat)
            FlatXmlDataSet
                    .write(partialDataSet, new FileOutputStream(fileName));
        else
            XmlDataSet.write(partialDataSet, new FileOutputStream(fileName));
    }

    // full database export
    public static void exportDatabase(IDatabaseConnection connection,
            String fileName, boolean streamed, boolean flat)
            throws SQLException, DatabaseUnitException, FileNotFoundException,
            IOException {
        if (streamed)
            connection.getConfig().setProperty(
                    DatabaseConfig.PROPERTY_RESULTSET_TABLE_FACTORY,
                    new ForwardOnlyResultSetTableFactory());

        IDataSet fullDataSet = connection.createDataSet();
        if (flat)
            FlatXmlDataSet.write(fullDataSet, new FileOutputStream(fileName));
        else
            XmlDataSet.write(fullDataSet, new FileOutputStream(fileName));
    }

    public static void refreshData(IDatabaseConnection connection,
            String input, boolean streamed, boolean flat)
            throws DatabaseUnitException, SQLException {
        // 把Documenth转换为DBUnit的DataSet并解决中文编码问题
        // is.setEncoding("GBK");
        IDataSetProducer producer;
        IDataSet dataSet;
        if (flat)
            producer = new FlatXmlProducer(new InputSource(input));
        else
            producer = new XmlProducer(new InputSource(input));
        if (streamed)
            dataSet = new StreamingDataSet(producer);
        else
            dataSet = new CachedDataSet(producer);

        // 执行插入操作
        DatabaseOperation.REFRESH.execute(connection, dataSet);
    }

    public static void cleanInsertData(IDatabaseConnection connection,
            String input, boolean flat) throws DatabaseUnitException,
            SQLException {
        IDataSetProducer producer;
        IDataSet dataSet;
        if (flat)
            producer = new FlatXmlProducer(new InputSource(input));
        else
            producer = new XmlProducer(new InputSource(input));

        dataSet = new CachedDataSet(producer);

        // 执行插入操作
        DatabaseOperation.CLEAN_INSERT.execute(connection, dataSet);
    }

}

分享到：

XP设备管理器里出现了两个主要IDE通道 | Oracle-Schema - zt

• 2007-10-19 19:46
• 浏览 6649
• 评论(7)
• 查看更多

评论

7 楼 yumcn.com 2010-05-28  

huangxx 写道

找到了关于外键约束的另外的解决办法使用CsvDataSetWriter来替代FlatXmlDataSet就可以避免外键约束的问题。

jim19770812 写道

找到了关于外键约束的另外的解决办法使用CsvDataSetWriter来替代FlatXmlDataSet就可以避免外键约束的问题。

这么老的文章也能找到？

huangxx 写道

找到了关于外键约束的另外的解决办法使用CsvDataSetWriter来替代FlatXmlDataSet就可以避免外键约束的问题。

jim19770812 写道

找到了关于外键约束的另外的解决办法使用CsvDataSetWriter来替代FlatXmlDataSet就可以避免外键约束的问题。

这么老的文章也能找到？

什么用CsvDataSetWriter，用依赖方式导出表数据即可。
TablesDependencyHelper.getAllDependentTables(connection, "表名");

6 楼 huangxx 2009-12-18  

找到了关于外键约束的另外的解决办法使用CsvDataSetWriter来替代FlatXmlDataSet就可以避免外键约束的问题。

jim19770812 写道

找到了关于外键约束的另外的解决办法使用CsvDataSetWriter来替代FlatXmlDataSet就可以避免外键约束的问题。

这么老的文章也能找到？

5 楼 jim19770812 2009-12-18  

找到了关于外键约束的另外的解决办法使用CsvDataSetWriter来替代FlatXmlDataSet就可以避免外键约束的问题。

4 楼 jim19770812 2009-12-17  

可以试试先禁用外键和约束，然后导入，最后再恢复外键和约束

3 楼 huangxx 2008-04-03  

呵呵，去看看它的文档吧。

2 楼 lying2 2008-01-04  

在导出数据后，重新导入数据，却报出外键的异常，查看导出的xml文件发现，dbunit导出时，表的顺序是按英文字母顺序，倒入时也是，但是外键关联的表，可能在正在加载的表的后面，所以出现了问题，请教你有没有好方法？

1 楼 lying2 2008-01-04