alibaba的easyexcel导入导出

前言

Excel表数据的导入和导出Excel表数据

一、导入

1、模板类

import com.alibaba.excel.annotation.ExcelProperty;
import com.alibaba.excel.metadata.BaseRowModel;
import lombok.Data;

/**
 *  无资产工单数据
 */
@Data
public class CrmWorkOrderNoDataImport extends BaseRowModel {

    @ExcelProperty(value = "工单号", index = 0)
    private String crmWorkOrderNo;

    @ExcelProperty(value = "大区", index = 1)
    private String office;

    @ExcelProperty(value = "是否是客户工单", index = 2)
    private String isCustomer;

    @ExcelProperty(value = "优先级", index = 3)
    private String priority;

    @ExcelProperty(value = "工程师", index = 4)
    private String engineer;

    @ExcelProperty(value = "巡检类型", index = 5)
    private String inspectionType;

    @ExcelProperty(value = "工单类型", index = 6)
    private String workOrderType;

    @ExcelProperty(value = "服务主管", index = 7)
    private String serviceSupervisor;

    @ExcelProperty(value = "工单状态", index = 8)
    private String status;

    @ExcelProperty(value = "提交日期", index = 9)
    private String submitDate;

    @ExcelProperty(value = "派工时间", index = 10)
    private String dispatchTime;

    @ExcelProperty(value = "客户要求处理时间", index = 11)
    private String customerTopicTime;

    @ExcelProperty(value = "遗留问题", index = 12)
    private String remainingProblem;

    @ExcelProperty(value = "客户名称", index = 13)
    private String siteName;

    @ExcelProperty(value = "上级单位", index = 14)
    private String customer;

    @ExcelProperty(value = "站点id", index = 15)
    private String siteId;

    @ExcelProperty(value = "错误信息", index = 16)
    private String faultReason;
}

2、导入的工具类方法

import com.alibaba.excel.EasyExcelFactory;
import com.alibaba.excel.metadata.Sheet;
import com.hqhop.easyExcel.model.*;
import com.hqhop.utils.StringUtils;

import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.InputStream;
import java.util.LinkedList;
import java.util.List;
import java.util.stream.Collectors;

/**
 *  数据导入工具类
 */
public class CrmWorkOrderExcelUtils {

    // 无资产工单数据Excel读取工具
    public static List<CrmWorkOrderNoDataImport> readCrmWorkOrderNoData(String fileName) {
        //创建输入流
        InputStream inputStream = null;
        try {
            inputStream = new FileInputStream(fileName);
        } catch (FileNotFoundException e) {
            e.printStackTrace();
        }
        Sheet sheet = new Sheet(1, 1, CrmWorkOrderNoDataImport.class);
        List<Object> typeList = EasyExcelFactory.read(inputStream, sheet);
        List<CrmWorkOrderNoDataImport> types = new LinkedList<>();
        for (Object student : typeList) {
            types.add((CrmWorkOrderNoDataImport) student);
        }
        List<CrmWorkOrderNoDataImport> collect = types.stream().filter(crmWorkOrderDataImport -> StringUtils.isNotBlank(crmWorkOrderDataImport.getCrmWorkOrderNo())).collect(Collectors.toList());
        return collect;
    }
}

3、导入测试方法(导出有问题的数据到Excel表中)

// 无资产的工单导入
    @Test
    @Transactional
    public void readCrmWorkOrderNoData() throws ParseException {
        List<CrmWorkOrderNoDataImport> crmWorkOrderDataImports = CrmWorkOrderExcelUtils.readCrmWorkOrderNoData("C:\\Users\\JanRain\\Desktop\\系统信息\\导出无问题的无资产工单数据.xlsx");
        // 根据工单号分组
        Map<String, List<CrmWorkOrderNoDataImport>> collect = crmWorkOrderDataImports.stream().collect(Collectors.groupingBy(crmWorkOrderDataImport -> crmWorkOrderDataImport.getCrmWorkOrderNo()));
        List<CrmWorkOrderNoDataImport> correctList = new ArrayList(); // 无误工单数据
        List<CrmWorkOrderNoDataImport> faultList = new ArrayList(); // 有无工单数据
        for (String key : collect.keySet()) {
            StringBuffer msg = new StringBuffer();
            List<CrmWorkOrderNoDataImport> crmWorkOrderDataImports1 = collect.get(key);
            for (CrmWorkOrderNoDataImport crmWorkOrderNoDataImport : crmWorkOrderDataImports1) {
                // 查询站点有无
                List<CrmSite> crmSiteList = crmSiteRepository.findDataOne(crmWorkOrderNoDataImport.getSiteName());
                if(crmSiteList.size() == 1){
                    crmWorkOrderNoDataImport.setSiteId(crmSiteList.get(0).getId().toString()); // 正确的站点id
                    correctList.add(crmWorkOrderNoDataImport);
                }else if(crmSiteList.size() != 1) {
//                    msg = msg.append(",站点有误");
                    crmWorkOrderNoDataImport.setFaultReason("站点有误");
                    faultList.add(crmWorkOrderNoDataImport);
                }
                // 查询Excel表中上级单位为字符 0 的是否存在
                if(crmWorkOrderNoDataImport.getCustomer().equals("0")){
                    List<CompanyInfo> customer = companyInfoRepository.findData(crmWorkOrderNoDataImport.getSiteName());
                    if(customer.size() == 1){
                        crmWorkOrderNoDataImport.setSiteId(customer.get(0).getCompanyKey().toString()); // 正确的客户id
                        correctList.add(crmWorkOrderNoDataImport);
                    }else if(customer.size() != 1){
//                        msg = msg.append(",客户有误");
                        crmWorkOrderNoDataImport.setFaultReason("客户有误");
                        faultList.add(crmWorkOrderNoDataImport);
                    }
                }
            }
        }
        // 导出有问题的无资产工单数据
        exportFaultCrmWorkOrderNoData(faultList);
        // 导出无问题的无资产工单数据
        exportCorrectCrmWorkOrderNoData(correctList);
		// -----------------  下面可以对正确数据做持久化操作 ---------------
		// -------
	}        

二、导出

	说明：导出上面导入的Excel表中有问题的数据

1、模板类

也就是上面导入的同一模板类

2、导出工具类方法

// 导出无问题的无资产工单数据
    public void exportCorrectCrmWorkOrderNoData(List<CrmWorkOrderNoDataImport> data){
        String filePath = "C:\\Users\\JanRain\\Desktop\\导出无问题的无资产工单数据.xlsx";
        com.hqhop.easyexcel.ExcelUtil.writeWithTemplate(filePath,data);
    }

import com.alibaba.excel.EasyExcelFactory;
import com.alibaba.excel.ExcelWriter;
import com.alibaba.excel.context.AnalysisContext;
import com.alibaba.excel.event.AnalysisEventListener;
import com.alibaba.excel.metadata.BaseRowModel;
import com.alibaba.excel.metadata.Sheet;
import lombok.Data;
import lombok.Getter;
import lombok.Setter;
import lombok.extern.slf4j.Slf4j;
import org.springframework.util.CollectionUtils;
import org.springframework.util.StringUtils;

import java.io.*;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;

/**
 * @description:
 * @author: chenmingjian
 * @date: 19-3-18 16:16
 */
@Slf4j
public class ExcelUtil {

    private static Sheet initSheet;

    static {
        initSheet = new Sheet(1, 0);
        initSheet.setSheetName("sheet");
        //设置自适应宽度
        initSheet.setAutoWidth(Boolean.TRUE);
    }

    /**
     * 读取少于1000行数据
     * @param filePath 文件绝对路径
     * @return
     */
    public static List<Object> readLessThan1000Row(String filePath){
        return readLessThan1000RowBySheet(filePath,null);
    }

    /**
     * 读小于1000行数据, 带样式
     * filePath 文件绝对路径
     * initSheet ：
     *      sheetNo: sheet页码，默认为1
     *      headLineMun: 从第几行开始读取数据，默认为0, 表示从第一行开始读取
     *      clazz: 返回数据List<Object> 中Object的类名
     */
    public static List<Object> readLessThan1000RowBySheet(String filePath, Sheet sheet){
        if(!StringUtils.hasText(filePath)){
            return null;
        }

        sheet = sheet != null ? sheet : initSheet;

        InputStream fileStream = null;
        try {
            fileStream = new FileInputStream(filePath);
            return EasyExcelFactory.read(fileStream, sheet);
        } catch (FileNotFoundException e) {
            log.info("找不到文件或文件路径错误, 文件：{}", filePath);
        }finally {
            try {
                if(fileStream != null){
                    fileStream.close();
                }
            } catch (IOException e) {
                log.info("excel文件读取失败, 失败原因：{}", e);
            }
        }
        return null;
    }

    /**
     * 读大于1000行数据
     * @param filePath 文件觉得路径
     * @return
     */
    public static List<Object> readMoreThan1000Row(String filePath){
        return readMoreThan1000RowBySheet(filePath,null);
    }

    /**
     * 读大于1000行数据, 带样式
     * @param filePath 文件觉得路径
     * @return
     */
    public static List<Object> readMoreThan1000RowBySheet(String filePath, Sheet sheet){
        if(!StringUtils.hasText(filePath)){
            return null;
        }

        sheet = sheet != null ? sheet : initSheet;

        InputStream fileStream = null;
        try {
            fileStream = new FileInputStream(filePath);
            ExcelListener excelListener = new ExcelListener();
            EasyExcelFactory.readBySax(fileStream, sheet, excelListener);
            return excelListener.getDatas();
        } catch (FileNotFoundException e) {
            log.error("找不到文件或文件路径错误, 文件：{}", filePath);
        }finally {
            try {
                if(fileStream != null){
                    fileStream.close();
                }
            } catch (IOException e) {
                log.error("excel文件读取失败, 失败原因：{}", e);
            }
        }
        return null;
    }

    /**
     * 生成excle
     * @param filePath  绝对路径, 如：/home/chenmingjian/Downloads/aaa.xlsx
     * @param data 数据源
     * @param head 表头
     */
    public static void writeBySimple(String filePath, List<List<Object>> data, List<String> head){
        writeSimpleBySheet(filePath,data,head,null);
    }

    /**
     * 生成excle
     * @param filePath 绝对路径, 如：/home/chenmingjian/Downloads/aaa.xlsx
     * @param data 数据源
     * @param sheet excle页面样式
     * @param head 表头
     */
    public static void writeSimpleBySheet(String filePath, List<List<Object>> data, List<String> head, Sheet sheet){
        sheet = (sheet != null) ? sheet : initSheet;

        if(head != null){
            List<List<String>> list = new ArrayList<>();
            head.forEach(h -> list.add(Collections.singletonList(h)));
            sheet.setHead(list);
        }

        OutputStream outputStream = null;
        ExcelWriter writer = null;
        try {
            outputStream = new FileOutputStream(filePath);
            writer = EasyExcelFactory.getWriter(outputStream);
            writer.write1(data,sheet);
        } catch (FileNotFoundException e) {
            log.error("找不到文件或文件路径错误, 文件：{}", filePath);
        }finally {
            try {
                if(writer != null){
                    writer.finish();
                }

                if(outputStream != null){
                    outputStream.close();
                }

            } catch (IOException e) {
                log.error("excel文件导出失败, 失败原因：{}", e);
            }
        }

    }

    /**
     * 生成excle
     * @param filePath 绝对路径, 如：/home/chenmingjian/Downloads/aaa.xlsx
     * @param data 数据源
     */
    public static void writeWithTemplate(String filePath, List<? extends BaseRowModel> data){
        writeWithTemplateAndSheet(filePath,data,null);
    }

    /**
     * 生成excle
     * @param filePath 绝对路径, 如：/home/chenmingjian/Downloads/aaa.xlsx
     * @param data 数据源
     * @param sheet excle页面样式
     */
    public static void writeWithTemplateAndSheet(String filePath, List<? extends BaseRowModel> data, Sheet sheet){
        if(CollectionUtils.isEmpty(data)){
            return;
        }

        sheet = (sheet != null) ? sheet : initSheet;
        sheet.setClazz(data.get(0).getClass());

        OutputStream outputStream = null;
        ExcelWriter writer = null;
        try {
            outputStream = new FileOutputStream(filePath);
            writer = EasyExcelFactory.getWriter(outputStream);
            writer.write(data,sheet);
        } catch (FileNotFoundException e) {
            log.error("找不到文件或文件路径错误, 文件：{}", filePath);
        }finally {
            try {
                if(writer != null){
                    writer.finish();
                }

                if(outputStream != null){
                    outputStream.close();
                }
            } catch (IOException e) {
                log.error("excel文件导出失败, 失败原因：{}", e);
            }
        }

    }

    /**
     * 生成多Sheet的excle
     * @param filePath 绝对路径, 如：/home/chenmingjian/Downloads/aaa.xlsx
     * @param multipleSheelPropetys
     */
    public static void writeWithMultipleSheel(String filePath,List<MultipleSheelPropety> multipleSheelPropetys){
        if(CollectionUtils.isEmpty(multipleSheelPropetys)){
            return;
        }

        OutputStream outputStream = null;
        ExcelWriter writer = null;
        try {
            outputStream = new FileOutputStream(filePath);
            writer = EasyExcelFactory.getWriter(outputStream);
            for (MultipleSheelPropety multipleSheelPropety : multipleSheelPropetys) {
                Sheet sheet = multipleSheelPropety.getSheet() != null ? multipleSheelPropety.getSheet() : initSheet;
                if(!CollectionUtils.isEmpty(multipleSheelPropety.getData())){
                    sheet.setClazz(multipleSheelPropety.getData().get(0).getClass());
                }
                writer.write(multipleSheelPropety.getData(), sheet);
            }

        } catch (FileNotFoundException e) {
            log.error("找不到文件或文件路径错误, 文件：{}", filePath);
        }finally {
            try {
                if(writer != null){
                    writer.finish();
                }

                if(outputStream != null){
                    outputStream.close();
                }
            } catch (IOException e) {
                log.error("excel文件导出失败, 失败原因：{}", e);
            }
        }

    }


    /*********************匿名内部类开始，可以提取出去******************************/

    @Data
    public static class MultipleSheelPropety{

        private List<? extends BaseRowModel> data;

        private Sheet sheet;
    }

    /**
     * 解析监听器，
     * 每解析一行会回调invoke()方法。
     * 整个excel解析结束会执行doAfterAllAnalysed()方法
     *
     * @author: chenmingjian
     * @date: 19-4-3 14:11
     */
    @Getter
    @Setter
    public static class ExcelListener extends AnalysisEventListener {

        private List<Object> datas = new ArrayList<>();

        /**
         * 逐行解析
         * object : 当前行的数据
         */
        @Override
        public void invoke(Object object, AnalysisContext context) {
            //当前行
            // context.getCurrentRowNum()
            if (object != null) {
                datas.add(object);
            }
        }
    }
}

三 、结束

说明：同一张表中生成正确和错误数据各一张表