datafileutil.java

来自「一个很好的开源项目管理系统源代码」· Java 代码 · 共 104 行

package net.java.workeffort.data;

import java.util.List;

import javax.sql.DataSource;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.dbunit.database.DatabaseDataSourceConnection;
import org.dbunit.database.IDatabaseConnection;
import org.dbunit.dataset.IDataSet;
import org.dbunit.dataset.xml.FlatXmlDataSet;
import org.dbunit.operation.DatabaseOperation;
import org.springframework.context.ApplicationContext;

/**
 * @author Antony Joseph
 */
public class DataFileUtil {
    public static String REQUIREMENT = "/net/java/workeffort/data/requirement.xml";
    public static String WORK = "/net/java/workeffort/data/work.xml";
    public static String WORK_TYPE = "/net/java/workeffort/data/workType.xml";
    public static String DELIVERABLE = "/net/java/workeffort/data/deliverable.xml";
    public static String FACILITY = "/net/java/workeffort/data/facility.xml";
    public static String FIXED_ASSET = "/net/java/workeffort/data/fixedAsset.xml";
    public static String PRODUCT = "/net/java/workeffort/data/product.xml";
    public static String SKILL = "/net/java/workeffort/data/skill.xml";
    public static String PART = "/net/java/workeffort/data/part.xml";
    public static String PARTY = "/net/java/workeffort/data/party.xml";
    public static String TIMESHEET = "/net/java/workeffort/data/timesheet.xml";
    public static String SECURITY = "/net/java/workeffort/data/security.xml";
    public static String TIMESHEET_AMOUNT_PROCESSING = "/net/java/workeffort/data/timesheetAmountProcessing.xml";

    public static String SEQUENCE = "/net/java/workeffort/data/sequence.xml";
    public static String LOGIN = "/net/java/workeffort/data/login.xml";

    public static String DATABASE_TABLES = "/net/java/workeffort/data/database-tables.xml";

    private static final Log logger = LogFactory.getLog(DataFileUtil.class);

    public void cleanAllDatabaseTables(ApplicationContext context)
            throws Exception {

        IDatabaseConnection connection = new DatabaseDataSourceConnection(
                (DataSource) context.getBean("dataSource"));

        IDataSet dataSet = null;
        try {
            dataSet = new FlatXmlDataSet(this.getClass().getResource(
                    DATABASE_TABLES));
            try {
                logger
                        .info("Cleaning database. Deleting all rows from all tables.");
                DatabaseOperation.DELETE_ALL.execute(connection, dataSet);
            }
            catch (Exception doe) {
                logger.error("1.Error during cleaning of all database tables.",
                        doe);
                throw doe;
            }
        }
        catch (Exception e) {
            logger.error("2.Error during cleaning tables. ", e);
            throw e;
        }
        finally {
            connection.close();
        }
    }

    public void loadDataFiles(List dataFiles, ApplicationContext context)
            throws Exception {
        if (dataFiles != null) {
            for (int i = 0; i < dataFiles.size(); i++) {
                String dataFile = (String) dataFiles.get(i);
                IDatabaseConnection connection = new DatabaseDataSourceConnection(
                        (DataSource) context.getBean("dataSource"));

                IDataSet dataSet = null;
                try {
                    dataSet = new FlatXmlDataSet(this.getClass().getResource(
                            dataFile));
                    try {
                        logger.info("Loading dataset() : " + dataFile);
                        DatabaseOperation.INSERT.execute(connection, dataSet);
                    }
                    catch (Exception doe) {
                        logger.error("1.Error during processing of dataSet."
                                + dataFile, doe);
                        throw doe;
                    }
                }
                catch (Exception e) {
                    logger.error("2.Error during processing  of dataSet. "
                            + dataFile, e);
                    throw e;
                }
                finally {
                    connection.close();
                }
            }
        }
    }
}