csv文件导出

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.util.StringUtils;
import javax.servlet.http.HttpServletResponse;
import java.io.*;
import java.nio.charset.StandardCharsets;
import java.text.SimpleDateFormat;
import java.util.*;

/**
 * @Description 封装csv 格式的调用方法
 * @Author ROCIA
 * @Date 2020/10/26
 */
public abstract class ExportCsvServiceImpl {
    private static final Logger logger = LoggerFactory.getLogger(ExportCsvServiceImpl.class);

    /**
     * 一次性查询后再一次性写入
     * @param fileName 文件名称： 无需加时间戳
     * @param columnName 列标题
     * @param theadColumn 列标题属性
     * @param dataList 数据 LinkedHashMap<String,Object> 小于20w
     * @param response 响应
     */
    public void exportCSV(String fileName ,String[] columnName,String[] theadColumn,  List<LinkedHashMap<String,Object>> dataList,  HttpServletResponse response) {
        setCsvHeader(fileName,response);
          BufferedWriter bw = null;
        try {
            java.io.OutputStream fos = response.getOutputStream();
            OutputStreamWriter writer = new OutputStreamWriter(fos, StandardCharsets.UTF_8);
            bw = new BufferedWriter(writer);
            StringBuilder str =  new StringBuilder();
            bw.write(getTableColumnName(columnName,fileName));
            String data = buildDataForCSV(theadColumn, dataList);
            bw.write(str.append(data).toString());
            bw.newLine();
        }catch (Exception e){
            logger.error(e.getMessage());
        }finally {
            bw.flush();
            bw.close();
            logger.info("导出结束");
        }
    }

    /**
     * 初始化 csv文件头
     * @param fileName 文件名称
     * @param columnName 列标题
     * @param response 响应
     */
    public BufferedWriter getBufferedWriter(String fileName, String[] columnName, HttpServletResponse response) throws IOException {
        setCsvHeader(fileName,response);
        OutputStream fos = response.getOutputStream();
        OutputStreamWriter writer = new OutputStreamWriter(fos, StandardCharsets.UTF_8);
        BufferedWriter bw = new BufferedWriter(writer);
        bw.write(getTableColumnName(columnName,fileName));
        return bw;
    }

    /**
     * 对文件内容的追加
     * @param dataList 数据
     * @param str 追加的StringBuilder
     * @param THEAD 表头
     * @param bw 写入
     * @throws IOException
     */
    public void appendStr(List<LinkedHashMap<String, Object>> dataList,StringBuilder str,String[] THEAD,BufferedWriter bw) throws IOException {
        str.setLength(0);
        for (Iterator c = dataList.iterator(); c.hasNext(); str.append("\n")){
            Map map = (Map)c.next();
            //分批写入
            for (int k = 0; k < THEAD.length; k++){
                Object obj = StringUtils.isEmpty(map.get(THEAD[k]))?"":map.get(THEAD[k]);
                if (k != THEAD.length - 1) {
                    str.append("\"" + "\t").append(obj).append("\",");
                }else {
                    str.append("\"" + "\t").append(obj).append("\"");
                }
            }
        }
        if (str.length() > 0){
            bw.write(str.toString());
        }
    }

    /**
     * 设置csv响应头部
     * @param fileName 文件名
     * @param response 响应
     */
    public void setCsvHeader(String fileName, HttpServletResponse response) {
        try {
            // 设置文件后缀
            SimpleDateFormat sdf = new SimpleDateFormat("yyyyMMddHHmmss");
            String filename = gbEncoding(fileName) + sdf.format(new Date()) + ".csv";
            //需要将 Content-Disposition 进行暴露出来，前端才能获取到文件名称以及扩展名
            response.setHeader("Access-Control-Expose-Headers","Content-Disposition"); 
            response.setHeader("Pragma", "public");
            response.setHeader("Cache-Control", "must-revalidate, post-check=0, pre-check=0");
            response.setHeader("Content-Disposition", "attachment;filename="+filename);
            response.setContentType("application/x-download;charset=UTF-8");
        }catch (Exception e){
            e.printStackTrace();
        }
    }
    
    /*
     * 中文转unicode编码
     */
    public static String gbEncoding(final String gbString) {
        char[] utfBytes = gbString.toCharArray();
        String unicodeBytes = "";
        for (int i = 0; i < utfBytes.length; i++) {
            String hexB = Integer.toHexString(utfBytes[i]);
            if (hexB.length() <= 2) {
                hexB = "00" + hexB;
            }
            unicodeBytes = unicodeBytes + "\\u" + hexB;
        }
        return unicodeBytes;
    }

    /**
     * 处理表头标题部分
     * @param columnName 列名称
     * @param fileName 文件名称
     */
    public String getTableColumnName(String columnName[],String fileName){
        StringBuilder sb = new StringBuilder();
        sb.append(fileName).append("\n");
        for (int i = 0; i < columnName.length; i++)
            if (i != columnName.length - 1) {
                sb.append((new StringBuilder()).append("\"").append(columnName[i]).append("\",").toString());
            } else {
                sb.append((new StringBuilder()).append("\"").append(columnName[i]).append("\"").toString());
            }
        sb.append("\n");
        return sb.toString();
    }

    /**
     * 数据转csv格式
     * @param theadColumn 列数据
     * @param dataList 查询出的数据
     */
    private  String buildDataForCSV(String[] theadColumn, List dataList) {
        StringBuilder strb = new StringBuilder();
        int j =0;
        for (Iterator i$ = dataList.iterator(); i$.hasNext(); strb.append("\n"),j++)
        {
            Map map = (Map)i$.next();
            for (int i = 0; i < theadColumn.length; i++){
                Object obj = StringUtils.isEmpty(map.get(theadColumn[i]))?"":map.get(theadColumn[i]);
                if (i != theadColumn.length - 1) {
                    strb.append((new StringBuilder()).append("\"").append("\t"+obj).append("\",").toString());
                }else {
                    strb.append((new StringBuilder()).append("\"").append("\t"+obj).append("\"").toString());
                }
            }
        }
        return strb.toString();
    }
}

调用示例：

		//查询 导出记录 是否存在
        if (count > 0) {
            Integer per = ExportConst.PER_NUM_2W;
            
            //分批次写入
            BufferedWriter bw = getBufferedWriter("会员卡余额",ExportConst.TITLE_BALANCE,response);
            try{
                int maxPageNum = count/per +((count % per) == 0?0:2);
                StringBuilder str = new StringBuilder();
                for (int i = 0; i < maxPageNum; i++) {
                    ex.setPageNum(i * per);
                    ex.setOffset(per);
                    List<LinkedHashMap<String, Object>> dataList = this.finBalanceService.exportFinBalanceExtList(ex,dataEncryptList);
                    //分批次追加数据
                    appendStr(dataList,str,ExportConst.THEAD_BALANCE,bw);
                }
            } catch (Exception e) {
                responseModel.setMessage("导出余额EXCEL成功");
                responseModel.setErrorCode(0);
                return responseModel;
            }finally {
                bw.flush();
                bw.close();
                logger.info("导出结束");
            }
            
            //一次性导出
            //CsvUtil.exportBigCSV("会员卡余额",ExportConst.TITLE_BALANCE,ExportConst.THEAD_BALANCE,dataList,response);
        }else {
            responseModel.setMessage("没有可以导出的余额数据");
            responseModel.setErrorCode(0);
            return responseModel;
        }
        responseModel.setMessage("导出余额EXCEL成功");
        return responseModel;
    }

学习使用。