Poi导入Excel校验表头

我们在日常的开发过程中,poi导入数据要根据注解的列名和顺序进行识别,那么首先要校验模版的正确与否,即表头是否正确,然后才能开始数据和业务层面的校验,最近也是被这个东西烦的不行,解决后记录一下,一个是方便自己以后查看,也给其他兄弟分享下,节省时间,话不多说,直接上代码:

//仅作事列,可根据具体代码改造
public Result dataImport(MultipartFile file, HttpServletResponse response, HttpServletRequest request, UserDTO userDTO) throws Exception {
    //1、校验文件是否符合基本标准
    if (ObjectUtils.isEmpty(file)) {
       return Result.failed("请选择上传文件");
    }
    if(!file.getOriginalFilename().endsWith(".xlsx") &&         
      !file.getOriginalFilename().endsWith(".xls")) {
       return Result.failed("请选择excel文件上传");
    }
    //2、定义正确表头有序集合
    List<String> list = new ArrayList<>();
    Collections.addAll(list, "姓名", "手机号", "性别");
    //3、开始校验
    boolean headerFlag = checkExcelHeaders(file, list);
    if (!headerFlag){
        return Result.failed("上传失败,模板错误,请核对后重新上传");
    }
}
        

/**
     * 检查 Excel 文件表头信息
     *
     * @param file
     * @param dataList
     * @return
     * @throws Exception
     */
    public boolean checkExcelHeaders(MultipartFile file, List<String> dataList) throws Exception {
        byte [] byteArr=file.getBytes();
        InputStream inputStream = new ByteArrayInputStream(byteArr);
        Workbook workbook = new XSSFWorkbook(inputStream);
        Sheet sheet = workbook.getSheetAt(0);
        System.out.println(sheet.getLastRowNum());
        //获取 excel 第一行数据(表头)
        Row row = sheet.getRow(0);
        //存放表头信息
        List<String> headerList = new ArrayList<>();
        //算下有多少列
        int colCount = sheet.getRow(0).getLastCellNum();
        System.out.println(colCount);
        for (int j = 0; j < colCount; j++) {
            Cell cell = row.getCell(j);
            String cellValue = cell.getStringCellValue().trim();
            headerList.add(cellValue);
        }
        return dataList.equals(headerList);
    }

通过有序集合的比较,就可以得出表头是否正确的结论啦,分享就到这里,有需要的兄弟快去试试吧!

更新如下:优化了传入Excel的格式,适配xls,以及对代码的细微调整以至于更加完善

/**
     * 检查 Excel 文件表头信息
     *
     * @param file
     * @param dataList
     * @return
     */
    public boolean checkExcelHeaders(MultipartFile file, List<String> dataList) throws IOException {
        InputStream inputStream = null;
        try {
            byte [] byteArr=file.getBytes();
            inputStream = new ByteArrayInputStream(byteArr);
            Workbook workbook = WorkbookFactory.create(inputStream);
            Sheet sheet = workbook.getSheetAt(0);
            System.out.println(sheet.getLastRowNum());
            //获取 excel 第一行数据(表头)
            Row row = sheet.getRow(0);
            //存放表头信息
            List<String> headerList = new ArrayList<>();
            //算下有多少列
            int colCount = sheet.getRow(0).getLastCellNum();
            for (int j = 0; j < colCount; j++) {
                Cell cell = row.getCell(j);
                String cellValue = cell.getStringCellValue().trim();
                headerList.add(cellValue);
            }
            return dataList.equals(headerList);
        }catch (Exception e){
            log.error("模版表头解析错误:", e);
            return false;
        }finally {
            if (inputStream != null){
                inputStream.close();
            }
        }
    }

评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

默慊$

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值