Java如何基于EasyExcel實現(xiàn)導(dǎo)入數(shù)據(jù)校驗并生成錯誤信息Excel
功能設(shè)計
由于項目中涉及到大量的文件導(dǎo)入功能,故考慮設(shè)計一個excel導(dǎo)入的通用框架,解決以下問題
- 導(dǎo)入的數(shù)據(jù)不可信任,可能出現(xiàn)空值校驗的許多判斷,如果將這些判斷加入業(yè)務(wù)代碼可能會造成大量代碼的堆積,如下情況:
if(name==null){ throw new RuntimeException("名稱不能為空"); } if(age==null){ throw new RuntimeException("年齡不能為空"); } if(sex==null){ throw new RuntimeException("性別不能為空"); } if(order.size()>10){ throw new RuntimeException("訂單號長度不能大于10"); }
EasyExcel幫我處理導(dǎo)入文件時,只是簡單的根據(jù)列名把內(nèi)容set到字段上,如果字段類型不符是會
直接報錯
的!而我們需要將數(shù)據(jù)的錯誤內(nèi)容提交給用戶,所以如下的報錯是不可取的針對文件中的問題,需要清晰地呈現(xiàn)給用戶,每一行具體出現(xiàn)了哪種類型的錯誤,例如如下:
基于EasyExcel封裝,由于項目中本身使用的EasyExcel,考慮到不改動項目的技術(shù)組成,還是基于EasyExcel開發(fā)。
設(shè)計思路
EasyExcel做的工作其實很簡單,就是把文件中的內(nèi)容映射到我們實體類的字段上,我們要做的就是在映射前和映射后做校驗
代碼解析
我先把完整代碼貼上,下面再詳細(xì)分析
注解類
/** * 導(dǎo)入校驗注解 * * @author wangmeng * @since 2024/5/25 */ @Retention(RetentionPolicy.RUNTIME) @Target(ElementType.FIELD) public @interface ExcelCheck { /** * 是否可以為空,默認(rèn)是 */ boolean canEmpty() default true; /** * 是否可以重復(fù),默認(rèn)是 */ boolean canRepeat() default true; /** * 長度校驗,只對String生效 */ int length() default -1; }
錯誤信息實體類
/** * excel導(dǎo)入錯誤信息 * * @author wangmeng * @since 2024/5/25 */ @Data @AllArgsConstructor @NoArgsConstructor @Accessors(chain = true) public class ExcelErrorMessage { /** * 行號 */ private Integer rowNum; /** * 列名 */ private String colHeaderName; /** * 錯誤信息 */ private String message; }
導(dǎo)入通用的listener
/** * excel導(dǎo)入共通監(jiān)聽類 * * @author wangmeng * @since 2024/5/25 */ @Slf4j public class CheckableImportListener<T> extends AnalysisEventListener<T> { /** * check注解對象 */ protected List<Object[]> filedList; /** * excel數(shù)據(jù) */ protected final List<T> list = new ArrayList<>(); /** * 錯誤信息集合 */ @Getter private final List<ExcelErrorMessage> errorList = new ArrayList<>(); private Boolean isEmpty = false; public CheckableImportListener() { super(); } @Override public void onException(Exception exception, AnalysisContext context) throws Exception { log.error("解析單元格失敗,", exception); if (exception instanceof ExcelDataConvertException) { ExcelDataConvertException excelDataConvertException = (ExcelDataConvertException) exception; log.error("第{}行,第{}列解析異常,數(shù)據(jù)為:{}", excelDataConvertException.getRowIndex(), excelDataConvertException.getColumnIndex(), excelDataConvertException.getCellData()); } } @Override public void invoke(T data, AnalysisContext context) { if (CollectionUtils.isEmpty(list)) { Class<?> clazz = data.getClass(); //含check注解的字段 filedList = Arrays.stream(clazz.getDeclaredFields()) .filter(o -> null != o.getAnnotation(ExcelCheck.class)) .map(o -> new Object[]{o, o.getAnnotation(ExcelCheck.class), o.getAnnotation(ExcelProperty.class)}).collect(Collectors.toList()); } log.info("data:{}", JSON.toJSONString(data)); list.add(data); if (CollectionUtils.isNotEmpty(filedList)) { checkEmpty(data); //存在空值則不進(jìn)行其他校驗 if (isEmpty) { return; } // 校驗長度 checkLength(data); } } @Override public void doAfterAllAnalysed(AnalysisContext context) { if (isEmpty) { return; } errorList.sort(Comparator.comparing(ExcelErrorMessage::getRowNum)); } /** * 檢驗非空 * * @param data */ public void checkEmpty(T data) { for (Object[] os : filedList) { Field filed = (Field) os[0]; filed.setAccessible(true); ExcelCheck excelCheck = (ExcelCheck) os[1]; ExcelProperty excelProperty = (ExcelProperty) os[2]; try { //校驗非空 if (!excelCheck.canEmpty()) { if (filed.get(data) == null || (filed.getType() == String.class && StringUtils.isEmpty((String) filed.get(data)))) { errorList.add(new ExcelErrorMessage() .setRowNum(list.size() + 1) .setColHeaderName(excelProperty.value()[0]) .setMessage(excelProperty.value()[0] + "字段不能為空!")); isEmpty = true; } } } catch (IllegalAccessException e) { log.error("校驗excel信息失敗,", e); e.printStackTrace(); } } } /** * 校驗長度 * * @param data */ public void checkLength(T data) { for (Object[] os : filedList) { Field filed = (Field) os[0]; filed.setAccessible(true); ExcelCheck excelCheck = (ExcelCheck) os[1]; ExcelProperty excelProperty = (ExcelProperty) os[2]; try { //校驗非空 if (excelCheck.length() > 0 && filed.getType() == String.class) { String value = (String) filed.get(data); if (value.length() > excelCheck.length()) { errorList.add(new ExcelErrorMessage() .setRowNum(list.size() + 1) .setColHeaderName(excelProperty.value()[0]) .setMessage(excelProperty.value()[0] + "字段長度大于" + excelCheck.length() + "!")); } } } catch (IllegalAccessException e) { log.error("校驗字段長度失敗,", e); throw new RuntimeException(e); } } } /** * 檢驗重復(fù) */ public void checkRepeat() { List<Object[]> repeatAnnotation = filedList.stream().filter(o -> { ExcelCheck excelCheck = (ExcelCheck) o[1]; return !excelCheck.canRepeat(); }).collect(Collectors.toList()); for (Object[] objects : repeatAnnotation) { ExcelProperty property = (ExcelProperty) objects[2]; //使用iterate方式構(gòu)建流以獲取行號 Stream.iterate(0, i -> i + 1).limit(list.size()).collect(Collectors.groupingBy(i -> { Field field = (Field) objects[0]; String result = ""; try { field.setAccessible(true); result = JSON.toJSONString(field.get(list.get(i))); } catch (IllegalAccessException e) { e.printStackTrace(); } return result; }, LinkedHashMap::new, Collectors.mapping(i -> i + 2, Collectors.toList()))) .forEach((k, v) -> { if (v.size() > 1) { for (int i = 0; i < v.size(); i++) { if (i == 0) { continue; } errorList.add(new ExcelErrorMessage() .setRowNum(v.get(i)) .setColHeaderName(property.value()[0]) .setMessage(property.value()[0] + "字段重復(fù)!")); } } }); } } public void addError(Integer index, String errorMessage) { ExcelErrorMessage excelErrorMessage = new ExcelErrorMessage().setRowNum(index).setMessage(errorMessage); errorList.add(excelErrorMessage); } }
導(dǎo)入處理器類
/** * excel導(dǎo)入處理器,在easyExcel基礎(chǔ)封裝,增加通用讀取、校驗功能 * * @author wangmeng * @since 2024/6/7 */ @Setter @Getter @Accessors(chain = true) @Slf4j public class ExcelImportProcessor { /** * 默認(rèn)校驗類型listener */ private CheckableImportListener<?> listener = new CheckableImportListener<>(); private Consumer<ExcelReaderBuilder> readerBuilderConsumer; /** * 默認(rèn)第一個sheet */ private Integer sheetNo = 0; /** * 錯誤列名 */ private final static String ERROR_COLUMN_NAME = "錯誤信息"; public ExcelImportProcessor() { } public ExcelImportProcessor(CheckableImportListener<?> listener) { this.listener = listener; } public <R> List<R> importData(MultipartFile file, Class<R> clazz) { // 校驗文件 validateExcel(file); List<R> dataList = null; try (InputStream inputStream = file.getInputStream()) { ExcelReaderBuilder readerBuilder = EasyExcel.read(inputStream, clazz, listener); if (readerBuilderConsumer != null) { readerBuilderConsumer.accept(readerBuilder); } dataList = readerBuilder.sheet(sheetNo).doReadSync(); } catch (ExcelAnalysisException e) { ExcelDataConvertException exception = (ExcelDataConvertException) e.getCause(); List<ExcelErrorMessage> errorList = listener.getErrorList(); String headerName = exception.getExcelContentProperty().getField().getAnnotation(ExcelProperty.class).value()[0]; errorList.add(new ExcelErrorMessage().setRowNum(exception.getRowIndex() + 1) .setColHeaderName(headerName) .setMessage("'" + headerName + "'類型轉(zhuǎn)換失敗,請輸入正確格式")); } catch (IOException ioe) { log.info("導(dǎo)入失敗,異常,", ioe); throw new RuntimeException("導(dǎo)入失敗!"); } if (CollectionUtils.isEmpty(dataList)) { throw new RuntimeException("解析數(shù)據(jù)為空!"); } return dataList; } public List<ExcelErrorMessage> getErrorList() { return listener.getErrorList(); } /** * 手動添加錯誤 * * @param index data的下標(biāo)(從0開始) * @param errorMessage 錯誤信息 */ public void addError(Integer index, String errorMessage) { // 下標(biāo)從0開始+1,標(biāo)題占一行+1,總計+2 Integer row = index + 2; listener.addError(row, errorMessage); } /** * 生成錯誤信息excel,在原excel文件追加錯誤列 * * @param filePath 源文件路徑 */ public Boolean generateErrorSheet(String filePath) { List<ExcelErrorMessage> errorList = listener.getErrorList(); if (CollectionUtils.isEmpty(errorList)) { return false; } Map<Integer, String> errorMap = errorList.stream().collect(Collectors.groupingBy(ExcelErrorMessage::getRowNum, Collectors.mapping(ExcelErrorMessage::getMessage, Collectors.joining(";")))); Workbook workbook = null; // 打開原excel文件 try ( FileInputStream inputStream = new FileInputStream(filePath)) { workbook = new XSSFWorkbook(inputStream); Sheet sheet = workbook.getSheetAt(sheetNo); // 添加錯誤列 Row headerRow = sheet.getRow(0); short lastCellNum = headerRow.getLastCellNum(); // 檢查是否已經(jīng)存在錯誤列 Cell lastValidCell = headerRow.getCell(lastCellNum - 1); if (lastValidCell != null) { if (!ERROR_COLUMN_NAME.equals(lastValidCell.getStringCellValue())) { Cell errorHeaderCell = headerRow.createCell(lastCellNum); errorHeaderCell.setCellValue(ERROR_COLUMN_NAME); errorMap.forEach((rowNum, msg) -> { Row row = sheet.getRow(rowNum - 1); if (row != null) { Cell errorCell = row.createCell(lastCellNum); errorCell.setCellValue(msg); } }); } else { int lastRowNum = sheet.getLastRowNum(); for (int rowNum = 1; rowNum <= lastRowNum; rowNum++) { Row row = sheet.getRow(rowNum); String setErrorMsg = errorMap.get(rowNum + 1); // 如果沒有需要設(shè)置的錯誤信息,要把舊的錯誤信息清除 Cell errorCell = row.getCell(lastCellNum - 1); if (setErrorMsg == null) { if (errorCell != null) { errorCell.setCellValue((String) null); } } else { if (errorCell == null) { errorCell = row.createCell(lastCellNum - 1); } errorCell.setCellValue(setErrorMsg); } } } } } catch (IOException e) { log.error("生成錯誤信息失敗,", e); throw new RuntimeException("生成錯誤信息失敗"); } try (FileOutputStream outputStream = new FileOutputStream(filePath)) { // 寫回去 workbook.write(outputStream); workbook.close(); } catch (IOException e) { log.error("生成錯誤信息失敗,", e); throw new RuntimeException("生成錯誤信息失敗"); } return true; } public static boolean isExcel2007(String filePath) { return filePath.matches("^.+\\.(?i)(xlsx)$"); } /** * 驗證EXCEL文件 * * @param file * @return */ public static void validateExcel(MultipartFile file) { if (file == null) { throw new RuntimeException("文件為空!"); } String fileName = file.getOriginalFilename(); if (fileName != null && !isExcel2007(fileName)) { throw new RuntimeException("導(dǎo)入文件必須是xlsx格式!"); } if (StringUtils.isEmpty(fileName) || file.getSize() == 0) { throw new RuntimeException("文件內(nèi)容不能為空"); } } }
捕獲類型轉(zhuǎn)換異常
導(dǎo)入的第一步就是處理字段類型錯誤,因為如果出現(xiàn)類型轉(zhuǎn)換錯誤,會直接導(dǎo)致程序異常,這里通過try,catch捕獲ExcelAnalysisException異常來獲取出現(xiàn)錯誤的列和行。
這里通過exception對象獲取到了field,再獲取字段上的ExcelProperty注解。
在AnalysisEventListener中實現(xiàn)校驗邏輯
在listener中的invoke
方法中為每一行數(shù)據(jù)做校驗,這里主要使用了反射
獲取到Error后,根據(jù)錯誤信息生成Excel
這里是拿導(dǎo)入的原本Excel文件,在最后追加一列錯誤信息列,并將錯誤信息與行對應(yīng),代碼如下
總結(jié)
到此這篇關(guān)于Java如何基于EasyExcel實現(xiàn)導(dǎo)入數(shù)據(jù)校驗并生成錯誤信息Excel的文章就介紹到這了,更多相關(guān)EasyExcel導(dǎo)入數(shù)據(jù)校驗生成錯誤信息內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家!
相關(guān)文章
java獲取鍵盤輸入的數(shù)字,并進(jìn)行排序的方法
今天小編就為大家分享一篇java獲取鍵盤輸入的數(shù)字,并進(jìn)行排序的方法,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧2018-07-07Linux環(huán)境卸載Centos7自帶的OpenJDK和安裝JDK1.8圖文教程
CentOS系統(tǒng)是開發(fā)者常用的Linux操作系統(tǒng),安裝它時會默認(rèn)安裝自帶的舊版本的OpenJDK,但在開發(fā)者平時開發(fā)Java項目時還是需要完整的JDK,這篇文章主要給大家介紹了關(guān)于Linux環(huán)境卸載Centos7自帶的OpenJDK和安裝JDK1.8的相關(guān)資料,需要的朋友可以參考下2024-07-07java打印表格 將ResultSet中的數(shù)據(jù)打印成表格問題
這篇文章主要介紹了java打印表格 將ResultSet中的數(shù)據(jù)打印成表格問題。具有很好的參考價值,希望對大家有所幫助。如有錯誤或未考慮完全的地方,望不吝賜教2022-12-12淺談maven的jar包和war包區(qū)別 以及打包方法
下面小編就為大家分享一篇淺談maven的jar包和war包區(qū)別 以及打包方法,具有很好的參考價值,希望對大家有所幫助2017-11-11