Java47导出数据到Excel

Posted 2022-07-19 huashengweilong

前言：

一般后台的报表查询页面，都会有导出数据到Excel表的功能

正文：

JS：

window.open("后台接口地址");

Controller层：

@ApiOperation(value = "导出列表")
@RequestMapping(value = "excel/exportExcel", method = RequestMethod.GET)
public void exportExcel(HttpServletResponse response, QueryVo query) 
    this.theService.exportCsvPointList(response, query);

Service层：

@Override
public void exportCsvPointList(HttpServletResponse response, QueryVo query) 
    String title = "积分记录";
    String fileName = title + ".csv";

    //根据查询条件query，查询出对应的数据
    List<Entity> list;

    LinkedHashMap<String, String> headMap = new LinkedHashMap<>();
    headMap.put("phone", "手机号"); //phone为Entity中的字段名，手机号为excel表的表头名称
    headMap.put("name", "姓名");

    ExportCsvUtil.exportCSV(title, fileName, list, headMap, response);

工具类：

ExportCsvUtil

package com.bf.base.utils;

import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;

import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletResponse;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.OutputStreamWriter;
import java.net.URLEncoder;
import java.util.ArrayList;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map.Entry;

/**
 * 下载CSV文件
 */
public class ExportCsvUtil 
    /**
     * @param response 返回体
     * @param list 数据集合
     * @param title 主标题
     * @param map 子标题与对象属性的映射
     * @param fileName 文件名
     */

    public static <T> void exportCSV (String title, , String fileName, List<T> list, LinkedHashMap<String,String> map, HttpServletResponse response) 
        JSONArray array = new JSONArray();
        for (T obj : list) 
            array.add(obj);
        
        
        // 表格头
        List<String> headList = new ArrayList<>();
        for (Entry<String, String> entry : map.entrySet()) 
            headList.add(entry.getValue());
        
        
        // 数据
        BufferedWriter csvWtriter = null;
        try 
            List<List<String>> dataList = new ArrayList<List<String>>();
            for (Object obj : array) 
                List<String> rowList = new ArrayList<>();
                JSONObject jo = (JSONObject) JSONObject.toJSON(obj);
                for (Entry<String, String> entry : map.entrySet()) 
                    String value = entry.getValue();
                    String key = entry.getKey();
                    String joValue = jo.get(key) == null ? "" : jo.get(key).toString();
                    rowList.add(joValue);
                
                dataList.add(rowList);
            

            // 文件下载，使用如下代码
             response.setContentType("application/csv;charset=gb18030");
             response.setHeader("Content-disposition", "attachment;filename=" + URLEncoder.encode(fileName, "UTF-8"));
             ServletOutputStream out = response.getOutputStream();
             csvWtriter = new BufferedWriter(new OutputStreamWriter(out, "gb18030"), 1024);

            /*int num = headList.size() / 2;
            StringBuffer buffer = new StringBuffer();
            for (int i = 0; i < num; i++) 
                buffer.append(" ,");
            
            csvWtriter.write(buffer.toString() + title + buffer.toString());
            csvWtriter.newLine();*/

            // 写入文件头部
            writeRow(headList, csvWtriter);

            // 写入文件内容
            for (List<String> row : dataList) 
                writeRow(row, csvWtriter);
            
            csvWtriter.flush();
            csvWtriter.close();
         catch (Exception e) 
            e.printStackTrace();
         finally 
            try 
                csvWtriter.close();
             catch (IOException e) 
                e.printStackTrace();
            
        
    

    /**
     * 写一行数据
     * @param row 数据列表
     * @param csvWriter
     * @throws IOException
     */
    private static void writeRow(List<String> row, BufferedWriter csvWriter) throws IOException 
        for (String data : row) 
            StringBuffer sb = new StringBuffer();
            String rowStr = sb.append("\"").append(data).append("\",").toString();
            csvWriter.write(rowStr);
        
        csvWriter.newLine();