Java FlatXmlDataSet.write方法代码示例

import org.dbunit.dataset.xml.FlatXmlDataSet; //导入方法依赖的package包/类
/**
 * Tests export.
 */
public void testDeletion() throws Exception {
	File resultDir = new File(testDir, "result");
	if (!resultDir.exists()) {
		resultDir.mkdir();
	}
	File result = null;
	result = new File(resultDir, "delete.sql");
	File insertResult = new File(resultDir, "insert.sql");
	
	if (doDeletion(datamodelDir, new File(testDir, "extractionmodel.csv"), result, insertResult, scope)) {
		Session statementExecutor = new Session(connectionArguments.get(0), connectionArguments.get(1), connectionArguments.get(2), connectionArguments.get(3));
		SqlScriptExecutor.executeScript(result.getCanonicalPath(), statementExecutor);
		statementExecutor.shutDown();
		
		File actualState = new File(resultDir, "remaining-state.xml");
		IDataSet depDataset = getConnection().createDataSet(getDataSet().getTableNames());
		Writer out = new FileWriter(actualState);
	    FlatXmlDataSet.write(depDataset, out, Charset.defaultCharset().name());
	    out.close();

		assertEqualDatasets(expectedResult, actualState);
	}
}
 

import org.dbunit.dataset.xml.FlatXmlDataSet; //导入方法依赖的package包/类
/**
 * 把数据集写入到文件中
 * @param dataSet
 * @param destFilePath
 */
public static void writeToFileFromDataSet(IDataSet dataSet,String destFilePath) {
	File destFile = new File(destFilePath);
	try {
			String destFileName = destFile.getName();
			//转换为excel格式
			if(destFileName.endsWith(".xls")){
				XlsDataSet.write(dataSet, new FileOutputStream(destFile));
			}else if(destFileName.endsWith(".xml")){
				FlatXmlDataSet.write(dataSet, new FileOutputStream(destFile));
			}else{
				log.error("文件格式不是xls或者xml，不支持");
				return;
			}
			log.info("写入数据到文件 : "+destFile.getAbsolutePath());
	} catch (Exception e) {
			log.error("写入数据到文件失败");
			log.error(e.getMessage());
	}
}
 

import org.dbunit.dataset.xml.FlatXmlDataSet; //导入方法依赖的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.xml.FlatXmlDataSet; //导入方法依赖的package包/类
public static void main(String[] args) throws Exception
{
    Class driverClass = Class.forName("org.postgresql.Driver");
    Connection jdbcConnection = DriverManager.getConnection(
            "jdbc:postgresql://localhost:5432/FormBuilder?autoReconnect=true", "fbdev", "fbdev");
    IDatabaseConnection connection = new DatabaseConnection(jdbcConnection);

    // partial database export
    /*QueryDataSet partialDataSet = new QueryDataSet(connection);
    partialDataSet.addTable("FOO", "SELECT * FROM TABLE WHERE COL='VALUE'");
    partialDataSet.addTable("BAR");
    FlatXmlDataSet.write(partialDataSet, new FileOutputStream("partial.xml"));*/

    // full database export
    IDataSet fullDataSet = connection.createDataSet();
    FlatXmlDataSet.write(fullDataSet, new FileOutputStream("skips_on_table.xml"));
    
    // dependent tables database export: export table X and all tables that
    // have a PK which is a FK on X, in the right order for insertion
    /*String[] depTableNames = 
      TablesDependencyHelper.getAllDependentTables( connection, "X" );
    IDataSet depDataset = connection.createDataSet( depTableNames );
    FlatXmlDataSet.write(depDataSet, new FileOutputStream("dependents.xml"));*/          
    
}
 

import org.dbunit.dataset.xml.FlatXmlDataSet; //导入方法依赖的package包/类
@Override
public String generateString() throws IOException, SQLException, DatabaseUnitException {
  StringWriter writer = new StringWriter();
  try {
    IDataSet modifiedDataSet = modify(dataSet);
    FlatXmlDataSet.write(modifiedDataSet, writer);
    return writer.toString();
  } finally {
    writer.close();
  }
}
 

import org.dbunit.dataset.xml.FlatXmlDataSet; //导入方法依赖的package包/类
@Override
public void writeDataSetString(File file) throws IOException, SQLException, DatabaseUnitException {
  OutputStream outputStream = new FileOutputStream(file);
  try {
    IDataSet modifiedDataSet = modify(dataSet);
    FlatXmlDataSet.write(modifiedDataSet, outputStream);
  } finally {
    outputStream.close();
  }
}
 

import org.dbunit.dataset.xml.FlatXmlDataSet; //导入方法依赖的package包/类
@After
@Transactional
public void dumpDatabase() throws SQLException, DatabaseUnitException, FileNotFoundException, IOException {
    // full database export
    Connection jdbcConnection = dataSource.getConnection();
    IDatabaseConnection connection = new DatabaseConnection(jdbcConnection);
    IDataSet fullDataSet = connection.createDataSet();
    FlatXmlDataSet.write(fullDataSet, new FileOutputStream("./target/QueryKeyRepositoryTest-dbunit.xml"));
}
 

import org.dbunit.dataset.xml.FlatXmlDataSet; //导入方法依赖的package包/类
@After
@Transactional
public void dumpDatabase() throws SQLException, DatabaseUnitException, FileNotFoundException, IOException {
    // full database export
    Connection jdbcConnection = dataSource.getConnection();
    IDatabaseConnection connection = new DatabaseConnection(jdbcConnection);
    IDataSet fullDataSet = connection.createDataSet();
    FlatXmlDataSet.write(fullDataSet, new FileOutputStream(
        "./target/TermFrequencyResultsDataManagerTest-dbunit.xml"));
}
 

import org.dbunit.dataset.xml.FlatXmlDataSet; //导入方法依赖的package包/类
public void writeQueryTablesToFile(Map<String, String> queries, File output) throws Exception {
    QueryDataSet dataSet = new QueryDataSet(this.dbUnitConnection);
    for (String table : queries.keySet()) {
        dataSet.addTable(table, queries.get(table));
    }
    if (output.getName().endsWith(".xml")) {
        FlatXmlDataSet.write(dataSet, new FileOutputStream(output));
    } else if (output.getName().endsWith(".xls")) {
        XlsDataSet.write(dataSet, new FileOutputStream(output));
    } else {
        new CsvDataSetWriter(output).write(dataSet);
    }
}
 

import org.dbunit.dataset.xml.FlatXmlDataSet; //导入方法依赖的package包/类
public static void doWork(Connection conn, TreeMap<String, String> tableQueryMap) throws Exception
{
  IDatabaseConnection connection = new DatabaseConnection(conn);

  // partial database export
  QueryDataSet partialDataSet = new QueryDataSet(connection);
  for (String table : tableQueryMap.keySet()) {
    partialDataSet.addTable(table, tableQueryMap.get(table));
  }

  FlatXmlDataSet.write(partialDataSet, System.out);
}
 

import org.dbunit.dataset.xml.FlatXmlDataSet; //导入方法依赖的package包/类
public static void main(String[] args) throws Exception {
    Config config = new Config();
    Class.forName(config.getDbDriver());
    Connection jdbcConnection = DriverManager.getConnection(config.getDbUrl(), config.getDbUser(), config.getDbPass());
    IDatabaseConnection connection = new DatabaseConnection(jdbcConnection, "public");

    // dtd
    FlatDtdDataSet.write(connection.createDataSet(), new FileOutputStream("src/test/java/it/pronetics/acl/conf/dbunit/database-schema.dtd"));

    // export
    IDataSet fullDataSet = connection.createDataSet();
    FlatXmlDataSet.write(fullDataSet, new FileOutputStream("src/test/java/opennms/opennms/acl/conf/dbunit/full.xml"));
}
 

import org.dbunit.dataset.xml.FlatXmlDataSet; //导入方法依赖的package包/类
protected void writeExpectedFile(IDatabaseConnection dbunitConnection,
		File outputFile, String tableName) throws IOException,
		DataSetException {

	QueryDataSet partialDataSet = new QueryDataSet(dbunitConnection);
	partialDataSet.addTable(tableName);

	FlatXmlDataSet.write(partialDataSet, new FileOutputStream(outputFile));
}
 

import org.dbunit.dataset.xml.FlatXmlDataSet; //导入方法依赖的package包/类
/**
 * Performs the extraction. If no tables or queries are specified, data from
 * entire database will be extracted.
 * Otherwise, a partial extraction will be performed.
 * 
 * @throws Exception
 */
public void extract() throws Exception {
    Connection conn = null;
    try {
        conn = dataSource.getConnection();
        logger.info("Beginning extraction from '" + conn.toString() + "'.");
        IDatabaseConnection connection = new DatabaseConnection(conn, schema);
        configConnection((DatabaseConnection) connection);
        if (tableList != null || queryList != null) {
            // partial database export
            QueryDataSet partialDataSet = new QueryDataSet(connection);
            addTables(partialDataSet);
            addQueries(partialDataSet);
            FlatXmlDataSet.write(partialDataSet, new FileOutputStream(dataSetName));
        } else {
            // full database export
            IDataSet fullDataSet = connection.createDataSet();
            FlatXmlDataSet.write(fullDataSet, new FileOutputStream(dataSetName));

            // dependent tables database export: export table X and all
            // tables that
            // have a PK which is a FK on X, in the right order for
            // insertion
            String[] depTableNames = TablesDependencyHelper.getAllDependentTables(connection, "research");
            IDataSet depDataset = connection.createDataSet(depTableNames);

            FlatXmlWriter datasetWriter = new FlatXmlWriter(new FileOutputStream("target/dependents.xml"));
            datasetWriter.write(depDataset);

        }
    } finally {
        if (conn != null) {
            conn.close();
        }
    }
    logger.info("Completed extraction to '" + dataSetName + "'.");
}
 

import org.dbunit.dataset.xml.FlatXmlDataSet; //导入方法依赖的package包/类
/**
 * Cambiar los datos de la conexión de acuerdo a la base de datos que se quiera usar como muestra
 * @param args
 * @throws Exception
 */
public static void main(String[] args) throws Exception
{
    String url  = JOptionPane.showInputDialog("Escriba la URL de MySQL incluyendo driver");
    String user = JOptionPane.showInputDialog("Escriba el nombre de usuario de la BD");
    String pass = JOptionPane.showInputDialog("Escriba la clave de la BD");
    String dir  = "test/src/omoikane/repository/sampleData.xml";

    Class driverClass = Class.forName("com.mysql.jdbc.Driver");
    Connection jdbcConnection = DriverManager.getConnection(
            url, user, pass);
    IDatabaseConnection connection = new DatabaseConnection(jdbcConnection);

    // Datos seleccionados considerando la base de datos de Omoikane 1.3.3 -> 1.4.0 alpha
    QueryDataSet partialDataSet = new QueryDataSet(connection);
    partialDataSet.addTable("articulos"        , "SELECT * FROM articulos         ORDER BY id_articulo DESC LIMIT 200");
    partialDataSet.addTable("base_para_precios", "SELECT * FROM base_para_precios ORDER BY id_articulo DESC LIMIT 200");
    partialDataSet.addTable("ventas"           , "SELECT * FROM ventas            ORDER BY id_venta DESC LIMIT 50");
    partialDataSet.addTable("ventas_detalles"  , "SELECT vd.* FROM ventas_detalles vd JOIN (SELECT id_venta FROM ventas ORDER BY id_venta DESC LIMIT 50) v ON (vd.id_venta = v.id_venta)");
    partialDataSet.addTable("cajas"            , "SELECT * FROM cajas");
    FlatXmlDataSet.write(partialDataSet, new FileOutputStream(dir));

    FlatDtdDataSet.write(connection.createDataSet(), new FileOutputStream("test/src/omoikane/repository/sampleBDSchema.dtd"));

    System.out.println("{}");
    JOptionPane.showMessageDialog(null, "Proceso completado. BD importada en "+dir);

}
 

import org.dbunit.dataset.xml.FlatXmlDataSet; //导入方法依赖的package包/类
private void writeBackupFile(IDataSet ds) throws IOException, DataSetException {
	tempFile = File.createTempFile("back","xml");
	FlatXmlDataSet.write(ds, new FileWriter(tempFile));
}