关于利用ExcelUtil.importExcel工具类导入文件前的效验工作

傷城~ 2023-10-10 14:59 6阅读 0赞

@Override
        @Transactional(rollbackFor = Exception.class)
        public void importMeaning(MultipartFile file) throws Exception {
            LoginUser loginUser = tokenService.getLoginUser();
            if(loginUser == null){
                throw new Exception("获取当前登录用户失败，请刷新或者退出重新登录！");
            }
            Date now = new Date();
            String userId = String.valueOf(loginUser.getUserid());
            //判断文件是否为空
            if(file.isEmpty()){
                throw new Exception("上传文件为空，请选择文件进行上传！");
            }
            if (file.getOriginalFilename() == null) throw new Exception("文件名不能为空！");
            //获取文件后缀
            String fileExtension = file.getOriginalFilename().substring(file.getOriginalFilename().lastIndexOf(".")+1);
            //判断文件类型
            if(!fileExtension.equals(CommonConstant.EXCEL_TYPE_XLS) &&!fileExtension.equals(CommonConstant.EXCEL_TYPE_XLSX)){
                throw new Exception("导入的文件类型有误，请上传(xls/xlsx)文件！");
            }
            //限制上传的文件大小
            if(file.getSize() > CommonConstant.MAX_FILE_SIZE){
                throw new Exception("单个文件大小超过最大限10M");
            }
            //获取文件输入流
            InputStream inputStreamData = null;
            inputStreamData = file.getInputStream();
            //读取数据sheet文件信息
            ExcelUtil<ICQvRepFormatMeaning> excelUtil = new ExcelUtil<ICQvRepFormatMeaning>(ICQvRepFormatMeaning.class);
            //默认读取第一页数据
            List<ICQvRepFormatMeaning> meaningVOS = excelUtil.importExcel(inputStreamData);
            //进行数据校验
            int size = meaningVOS.size();
            if(size > 0) {
                for (int i = 0; i < size; i++) {
                    ICQvRepFormatMeaning icQvRepFormatMeaning = meaningVOS.get(i);
                    //校验字符不能为空
                    if (StringUtils.isBlank(icQvRepFormatMeaning.getMeaningCharacter())) throw new Exception("第" + (i + 2) + "行的字符不能为空！");
                 
                    //入库
                    try {
                        saveCommonFun(icQvRepFormatMeaning, userId, now);
                    }catch (Exception e){
                        e.printStackTrace();
                        throw new Exception("第" + (i + 2) + "行数据保存失败：" + ExceptionUtils.getMessage(e));
                    }
                }
            }
    
        }

这里不提供excelUtil工具类的代码，可自行去hoot上下载也该excel工具包，即可。

这里主要提供的代码为在导入文件数据之前对文件的效验工作

步骤为：

1.判断文件是否为空

2.获取文件后缀是否为xls/xlsx

3.限制上传大小

4.获取字节流后通过excel的importExcel获取数据后在存储进数据库时判断某些字段是否为空。