Java项目中Excel百万级别的导入导出

在Excel文件导入过程中实际上有两个问题 (采用apche poi的方式)
* xls的文件限制了不能一次性导出65536条数据
* 实际过程中Excel文件的解析有可能会造成内存溢出的问题
* mybatis(mysql)批量导入数据性能过差

内存泄漏:是指申请的内存空间无法正常被收回,造成“无用又不可收回”的状况内存溢出:指程序在申请内存空间时,已经没有足够的内存空间被使用。(内存泄漏通常是内存溢出的主要原因)针对上述的问题,需要对项目进一步地优化:
1. 文件上传到服务器的请求方式使用ftp而不是http
2. 尽可能减少遍历的次数
3. 使用StringBuffer而不是String (并且StringBuffer相对于StringBuilder线程安全的)
4. 使用FileWriter写入文件时要要一次写入,减少IO读写次数
5. 采用多线程的方式去批量地提交事务 (补充 提供List.subList(开始index,结束index)的方式截取集合,或者采用流式表达式的方式进行切分)https://blog.csdn.net/FKJGFK/article/details/104004216
6. mysql数据库不能同时支持30万数据的 一次性事务提交,需要分批次地去进行事务的提交,否则数据将堆积到mysql数据库中,导致数据库性能过低。(需要采用编程式事务的方式分批次多次提交)

方式一:优化内存

针对内存溢出的情况我们做以下调整:
1.提高了编译器的jvm的空间 -Xms128m -Xmx1700m (在项目的启动脚本中 nohup -Xms128m -Xmx1700m -jar 包名 & )
2.创建Workbook 借助xlsx-streamer(StreamingReader) 来创建一个缓冲区域批量地读取文件 (不会将整个文件实例化到对象当中)
依赖


   com.monitorjbl
   xlsx-streamer
   2.1.0

实现方式

public List<AddressBookPojo> importExcelxlsx( String tmpFile ) {
    List<AddressBookPojo> list = new ArrayList<AddressBookPojo>();
    FileInputStream in=null;
    try {
        in = new FileInputStream(tmpFile);
        Workbook wk = StreamingReader.builder()
                .rowCacheSize(1000)  //缓存到内存中的行数,默认是10
                .bufferSize(8192)  //读取资源时,缓存到内存的字节大小,默认是1024
                .open(in);  //打开资源,必须,可以是InputStream或者是File,注意:只能打开XLSX格式的文件
        Sheet sheet = wk.getSheetAt(0);
        int i =0;
        for (Row row : sheet) {
            i++;//从第2行开始取数据,默认第一行是表头.
            if(i==1)
                continue;
            AddressBookPojo abp = new AddressBookPojo();
            //遍历所有的列
            int j=0;
            for (Cell cell : row) {
                switch (j) {
                    case 0 :abp.setName(cell.getStringCellValue());break;
                    case 1 :abp.setFristDepartName(cell.getStringCellValue());break;
                    /**这里省略其他的属性**/
                    default: break;
                }
                j++;
            }
            list.add(abp);
        }
    } catch (IOException e) {
        e.printStackTrace();
    }finally {
        IOUtils.closeQuietly(in);
    }
    return list;
}

方式二:EasyExcel

EasyExcel是阿里的引用了apache poi,所以不能再引用apache poi的依赖。
使用lombok可以简化Excel对POJO的绑定关系。
依赖



   com.alibaba
   easyexcel
   1.1.2-beat1





   org.projectlombok
   lombok
   1.18.2
   true

工具类

package com.xiyuan.startingpoint.utils;
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.apache.poi.ss.usermodel.Workbook;
import org.springframework.util.CollectionUtils;
import org.springframework.util.StringUtils;

import javax.servlet.http.HttpServletResponse;
import java.io.*;
import java.net.URLEncoder;
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);
    }

    public static void downLoadExcel(String fileName, HttpServletResponse response, Workbook workbook) {
     try {
            response.setCharacterEncoding("UTF-8");
            response.setContentType("application/octet-stream;charset=utf-8");
            response.setHeader("Content-Disposition", "attachment;filename=" + URLEncoder.encode(fileName, "UTF-8"));
            workbook.write(response.getOutputStream());
        } catch (IOException e) {
            // throw new NormalException(e.getMessage());
        }
    }

    /**
     * 读取少于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的类名
     */
    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);
            }
        }


        /**
         * 解析完所有数据后会调用该方法
         */
        @Override
        public void doAfterAllAnalysed(AnalysisContext context) {
            //解析结束销毁不用的资源
        }

    }

    /************************匿名内部类结束,可以提取出去***************************/


}


 
  

POJO

/**
* 用户Pojo
*/
@Data
@TableName("tb_user")
@EqualsAndHashCode(callSuper = true)
@JsonIgnoreProperties(ignoreUnknown = true)
public class UserInfoPojo extends CommentPojo{

    private static final long serialVersionUID = 1L;
    //解决mybatis-plus和easyExcel的冲突问题
    @JsonIgnoreProperties(ignoreUnknown = true)
    @TableField(exist = false)
    private Map<Integer,CellStyle> cellStyleMap = new HashMap<Integer, CellStyle>();


    @TableId(type = IdType.UUID)
    private String id; //主键


    @TableField("user_name")
    @ExcelProperty(value = "用户名(必填)", index = 0)
    private String userName; //用户名
    @ExcelProperty(value = "密码(必填)", index = 1)
    private String password;//密码
}

Excel导入

/**
* 导入Excel
*
* @param file
* @param username
* @return
*/
@Override
public ResponseResult importExcel(MultipartFile file, String username) throws Exception {
    String tempfilePath = null;
    File tempfile = null;
    FileWriter fw = null;
    try {
        //第一步:创建临时文件
        String fileType = file.getOriginalFilename().substring(file.getOriginalFilename().lastIndexOf(".") + 1);
        tempfilePath = tempPath + CommonUtil.getUUID() + "." + fileType;
        tempfile = new File(tempfilePath);
        file.transferTo(tempfile);
        List<Object> objects = ExcelUtil.readMoreThan1000Row(tempfilePath);
        //第二步:进行Excel文件校验
        ResponseResult responseResult = checkExcel(objects,username);
        if (responseResult.getSuccess()) {//List读取成功
            List<UserInfoPojo> list = (List<UserInfoPojo>) responseResult.getValue();
            insertBatch(list);
            return ResponseResult.success(ConstantsUtil.OPERATE_SUCCESS);
        } else {//读取失败
            String errortempFile = tempPath +"/"+username+ "/user/" + CommonUtil.getUUID() + ".txt";
            File file1 = new File(errortempFile);
            fw = new FileWriter(errortempFile);
            fw.write(responseResult.getMessage());
            responseResult.setValue(errortempFile);
            return responseResult;
        }
    } catch (IOException e) {
        logger.error("用户模块导入失败" + e.getMessage(), e);
        return ResponseResult.error(ConstantsUtil.OPERATE_ERROR);
    } finally {
        if(fw !=null)
        fw.close();


        FileUtils.deleteQuietly(tempfile);
    }
}

private ResponseResult checkExcel(List<Object> objects,String username) throws Exception {
    List<UserInfoPojo> userInfoPojos = new ArrayList<>();
    StringBuffer bf = new StringBuffer();
    if (objects != null && objects.size() > 1) {
        int i = 2;
        for (int j = 1; j < objects.size(); j++) {
            List<String> list = (List<String>) objects.get(j);
            if(list.size()<2){
                bf.append("第" + i + "行中的存在大量必填项未填写,请检查" + newLine);
                continue;
            }
            UserInfoPojo userInfoPojo = new UserInfoPojo();
            userInfoPojo.setId(CommonUtil.getUUID());
            userInfoPojo.setUpdater(username);
            userInfoPojo.setUpdateTime(new Date());
            if (StringUtils.isBlank(list.get(0))) {
                bf.append("第" + i + "行中的用户名为必填项不得为空" + newLine);
            } else {
                userInfoPojo.setUserName(list.get(0));
            }
            if (StringUtils.isBlank(list.get(1))) {
                bf.append("第" + i + "行中的密码为必填项不得为空" + newLine);
            } else {
                userInfoPojo.setPassword(list.get(1));
            }
            userInfoPojos.add(userInfoPojo);
            i++;
        }
    }
    if (bf.length() > 0) {//导入失败
        return ResponseResult.error(bf.toString());
    } else {//导入成功
        return ResponseResult.success(userInfoPojos, ConstantsUtil.OPERATE_SUCCESS);
    }
}

Excel导出

/**
* Excel文件导入
*
* @param list
* @return
*/
@Override
public String exportExcel(List<UserInfoPojo> list) throws Exception {
    String tempfile = tempPath + CommonUtil.getUUID() + ".xlsx";
    ExcelUtil.writeWithTemplate(tempfile, list);
    return tempfile;
}

你可能感兴趣的:(java,excel)