001package com.unitils.boot.xls;
002
003import org.unitils.core.UnitilsException;
004import org.unitils.dbunit.datasetfactory.DataSetFactory;
005import org.unitils.dbunit.util.MultiSchemaDataSet;
006
007import java.io.File;
008import java.util.Arrays;
009import java.util.Properties;
010
011/**
012 * @Author: yangjianzhou
013 * @Description:
014 * @Date:Created in 2018-07-08
015 */
016public class MultiSchemaXlsDataSetFactory implements DataSetFactory {
017
018    protected String defaultSchemaName;
019
020    public void init(Properties configuration, String defaultSchemaName) {
021        this.defaultSchemaName = defaultSchemaName;
022    }
023
024    public MultiSchemaDataSet createDataSet(File... dataSetFiles) {
025        try {
026            MultiSchemaXlsDataSetReader xlsDataSetReader = new MultiSchemaXlsDataSetReader(
027                defaultSchemaName);
028            return xlsDataSetReader.readDataSetXls(dataSetFiles);
029        } catch (Exception e) {
030            throw new UnitilsException("创建数据集失败: "
031                + Arrays.toString(dataSetFiles), e);
032        }
033    }
034
035    public String getDataSetFileExtension() {
036        return "xls";
037    }
038
039}