Table of Contents

public class TestImportExcelVo {

    /**
     * 表头
     */
    public final static String[] HEADER = {"name","age","gender"};


    /**
     * 姓名
     */
    private String name;

    /**
     * 年纪
     */
    private String age;

    /**
     * 性别
     */
    private String gender;
}

可以看到唯一不同的就是多了一个常量HEADER，它的规则就是导入模板字段顺序

3-2、新增一个ImportExcel接口的实现类，实现里面的 check 方法（这个接口下面讲解）

@Service
public class TestImportExcelImpl implements ImportExcel<TestImportExcelVo> {

    @Override
    public String[] check(ImportItem importItem, List<TestImportExcelVo> successResult, List<ImportItem> errorResult) {
        
        return new String[]{"1",""};
    }
}

3-3、调用

@RestController
public class ImportExcelController {
    @Autowired
    public TestImportExcelImpl testImportExcelImpl;

    @PostMapping("/importExcel")
    public Object importExcel(MultipartFile file) {

       return testImportExcelImpl.importCheck(file, TestImportExcelVo.HEADER, TestImportExcelVo.class);
    }
}

四、代码思路讲解

其实就是看接口 ImportExcel 里面的两个方法。

4-1、准备

这里使用了两个bean如下（去掉了get/set）

/**
 * 导出错误结果集实体bean
 *
 * 因为错误结果集需要多一个原因字段
 *
 * @author 小道仙
 * @date 2021年8月4日
 */
public class ImportItem<T> {

    private String errorReason;

    public T t;
}

/**
 * 导入Excel返回结果实体
 *
 * @author 小道仙
 * @date 2021-08-04
 */
public class ImportResultVo<T> {

    /**
     * 总行数
     */
    private Integer totalRows;

    /**
     * 错误行数
     */
    private Integer errorRows;

    /**
     * 成功行数
     */
    private Integer successRows;

    /**
     * 失败结果集
     */
    private List<T> errorResults;

    /**
     * 成功结果集
     */
    private List<T> successResults;

    public ImportResultVo() {
    }

    public ImportResultVo(Integer totalRows,Integer errorRows,Integer successRows,List<T> errorResults,List<T> successResults) {
        this.totalRows = totalRows;
        this.errorRows = errorRows;
        this.successRows = successRows;
        this.errorResults =errorResults;
        this.successResults = successResults;
    }
}

4-2、importCheck方法

这是一个默认方法完成一些通用的功能，如：excel解析、实体bean封装等

default ImportResultVo importCheck(MultipartFile file, String[] header,Class clazz){
    // 从excel里面解析出来数据
    List<Map<String, Object>> maps;
    try {
        maps = ExcelUtils.importExcel(file, header);
    }catch (Exception e){
        logger.error("解析失败，数据模板不对",e);
        throw new RuntimeException("解析失败，数据模板不对");
    }
    if (CollectionUtils.isEmpty(maps)){
        return new ImportResultVo(0,0,0,null,null);
    }

    // 对数据进行逐步的验证
    int totalRows = 0;
    int successRows = 0;
    int errorRows = 0;
    List<T> successResult = new ArrayList<>(50);
    List<ImportItem> errorResult = new ArrayList<>(50);
    for (Map map : maps) {
        ImportItem<T> importItem = new ImportItem();
        totalRows ++;
        try {
            Object tmpT = JsonUtils.objectToPojo(map, clazz);
            importItem.setT((T)tmpT);
        }catch (Exception e) {
            errorRows ++;
            logger.error("bean转换失败", e);
            continue;
        }
        String[] check = check(importItem, successResult, errorResult);
        if ("1".equals(check[0])) {
            successRows ++;
            successResult.add(importItem.getT());
        }else if ("2".equals(check[0])) {
            errorRows ++;
            importItem.setErrorReason(check[1]);
            errorResult.add(importItem);
        }else if ("3".equals(check[0])){
            successRows --;
            errorRows ++;
            importItem.setErrorReason(check[1]);
            errorResult.add(importItem);
        }
    }
    return new ImportResultVo(totalRows,errorRows,successRows,errorResult,successResult);
}

4-3、check 方法

这是一个接口方法，需要自己去实现具体的校验规则

/**
 *
 * @param t 当前需要校验的实体bean
 * @param successResult 已校验的正确结果集
 * @param errorResult 已校验错误的结果集
 * @return 第一个参数是校验结果，第二个是失败的原因
 *      1  成功
 *      2  正常失败（比如字段为空，字段格式不对，和数据库冲突等）
 *      3  异常失败（和之前已经成功的数据冲突，比如唯一性字段已经存在了）
 *      注：【异常失败】的时候，你需要在check方法里面去删除掉之前成功的数据,并把之前的数据插入到 errorResult 里面
 */
String[] check(ImportItem t, List<T> successResult, List<ImportItem> errorResult);

4-4、其它

这里的excel导出工具类之前已经说过了，这里就简单的复制一下吧。

/**
 * Excel 导入
 * @param file 文件
 * @param keys 数据顺序
 */
public static  List<Map<String, Object>>  importExcel(MultipartFile file, String[] keys) throws Exception{
    Workbook wb = null;
    String fileName = file.getOriginalFilename();
    if (fileName.endsWith("xls")) {
        POIFSFileSystem pois = new POIFSFileSystem(file.getInputStream());
        wb = new HSSFWorkbook(pois);
    } else if (fileName.endsWith("xlsx")) {
        wb = new XSSFWorkbook(file.getInputStream());
    }
    Sheet sheet = wb.getSheetAt(0);
    int rowCount = sheet.getPhysicalNumberOfRows();
    if (sheet.getRow( 1).getPhysicalNumberOfCells() != keys.length){
        throw new RuntimeException("导入的Excel和模板的列不匹配");
    }
    List<Map<String,Object>> result = new ArrayList<>();
    for (int i = 0; i < rowCount - 1; i++) {
        Row row = sheet.getRow(i + 1);
        Map<String,Object> tmp = new HashMap<>();
        for (int j = 0;j < keys.length; j++){
            Cell cell = row.getCell(j);
            if(cell == null) {
                tmp.put(keys[j], null);
            }else {
                // 把类型转行Spring
                cell.setCellType(CellType.STRING);
                tmp.put(keys[j], cell.getStringCellValue());
            }
        }
        result.add(tmp);
    }
    return result;
}

视频看讲解地址：https://www.bilibili.com/video/BV1kf4y1i761

关注微信公众号回复：xdxFrameSimple 获取源码。