QingLong/src/main/java/UnitTest/ImportExcel/ImportExcelData.java

package UnitTest.ImportExcel;

import com.dsideal.QingLong.Util.AsposeUtil;
import com.dsideal.QingLong.Util.ImportUtil;
import com.jfinal.kit.PropKit;
import com.jfinal.kit.StrKit;
import com.jfinal.plugin.activerecord.ActiveRecordPlugin;
import com.jfinal.plugin.activerecord.Record;
import com.jfinal.plugin.activerecord.CaseInsensitiveContainerFactory;
import com.jfinal.plugin.activerecord.dialect.PostgreSqlDialect;
import com.jfinal.plugin.hikaricp.HikariCpPlugin;
import org.apache.poi.ss.usermodel.CellType;
import org.apache.poi.xssf.usermodel.XSSFCell;
import org.apache.poi.xssf.usermodel.XSSFSheet;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
import org.apache.poi.ss.usermodel.Cell;

import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.util.List;
import java.util.Map;

public class ImportExcelData {
    public static String path = "D:\\dsWork\\QingLong\\src\\main\\resource\\Excel\\";

    //模拟上传文件的文件名称
    public static String upload_excel_filename = "b61a2af2-223f-4058-a675-b212e4dd9487" + ".xlsx";


    public static void main(String[] args) throws IOException {
        //加载License
        AsposeUtil.getLicense();

        //告之配置文件位置
        PropKit.use("application.properties");
        HikariCpPlugin hp = new HikariCpPlugin(PropKit.get("jdbcUrl"), PropKit.get("user"),
                PropKit.get("password").trim(), PropKit.get("driverClassName"));
        hp.start();
        // 配置ActiveRecord插件
        ActiveRecordPlugin arp = new ActiveRecordPlugin(hp);
        //配置默认小写
        arp.setContainerFactory(new CaseInsensitiveContainerFactory(true));
        arp.setDialect(new PostgreSqlDialect());
        arp.start();

        //模板文件
        String f1 = path + File.separator + upload_excel_filename;
        //用户上传的填充完的EXCEL文件
        String f2 = path + File.separator + "school.xlsx";
        //对比两个EXCEL文件 是不是格式一致，也就是是不是上传了正确的模板文件
        int sheetCnt = ImportUtil.getExcelSheetCount(upload_excel_filename);
        for (int i = 0; i < sheetCnt; i++) {
            List<Map.Entry<Integer, Integer>> chayi = ImportUtil.checkYiZhi(f1, f2, i);
            if (chayi.size() > 0) {
                System.out.println("第" + (i + 1) + "个Sheet表检查过程中发现与要求上传的模板不一致，请重新下载模板填写完成后再次上传！");
                return;
            }
        }
        System.out.println("OK,是我下发的模板！");


        //打开EXCEL，进行检查
        InputStream is = new FileInputStream(f2);
        XSSFWorkbook wb = new XSSFWorkbook(is);

        //遍历每个Sheet注册好的信息，对用户上传的数据表进行检查
        for (int i = 0; i < sheetCnt; i++) {//表
            //Sheet表
            XSSFSheet sheet = wb.getSheetAt(i);
            //数据起始行
            Record r = ImportUtil.getExcelSheetInfo(upload_excel_filename, i);
            int data_start_row = r.getInt("data_start_row");
            //数据有效行数
            int lastRowNum = sheet.getLastRowNum();

            //遍历每一列
            List<Record> list = ImportUtil.getExcelSheetColumnsInfo(upload_excel_filename, i);
            int colIdx = 0;
            for (Record record : list) {//列
                System.out.println(record);
                int excel_column_idx = record.getInt("excel_column_idx");//第几列
                String column_type = record.getStr("column_type");//类型
                boolean allow_blank = record.getBoolean("allow_blank");//是不是允许为空

                System.out.println("Sheet=" + i + ",excel_column_idx=" + excel_column_idx + ",allow_blank=" + allow_blank);

                if (!allow_blank) {
                    System.out.println("现在是第" + i + "个Sheet!");
                    // 非空项目未填写
                    for (int j = data_start_row; j <= lastRowNum; j++) {//行
                        if (StrKit.isBlank(ImportUtil.getCellValue(sheet.getRow(j).getCell(colIdx)).toString())) {
                            System.out.println("SheetIdx=" + i + ",Rows=" + j + ",Column=" + colIdx + "不允许为空！");
                            System.exit(0);
                        }
                    }
                }

                // 下拉列表不是从下拉列表中选择的

                // 检查数据类型是不是和规定的不兼容

                colIdx++;
            }
        }

        wb.close();
        // 全都通过，导入数据

    }
}