Springboot集成Hutool导出CSV

1. 引入依赖

xml 复制代码

 <dependency>
     <groupId>cn.hutool</groupId>
     <artifactId>hutool-all</artifactId>
     <version>5.8.11</version>
 </dependency>

2. CSV工具类

java 复制代码

import cn.hutool.core.text.csv.CsvUtil;
import cn.hutool.core.text.csv.CsvWriter;
import lombok.extern.slf4j.Slf4j;

import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.io.OutputStream;
import java.io.OutputStreamWriter;
import java.lang.reflect.Field;
import java.net.URLEncoder;
import java.nio.charset.StandardCharsets;
import java.util.List;
import java.util.Map;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;

@Slf4j
public class CsvUtils {

    public static <T> void export(HttpServletResponse response, String fileName, List<T> dataList) throws IOException {
        // 设置响应内容类型
        response.setContentType("text/csv");
        response.setCharacterEncoding(StandardCharsets.UTF_8.name());
        fileName = URLEncoder.encode(fileName, StandardCharsets.UTF_8);
        response.setHeader("Content-Disposition", "attachment;filename=\"" + fileName + ".csv\"");
        try (OutputStream out = response.getOutputStream();
             OutputStreamWriter writer = new OutputStreamWriter(out, StandardCharsets.UTF_8)) {
            // 写入UTF-8 BOM，帮助Notepad识别
            writer.write("\uFEFF");
            // 使用HuTool的CsvWriter
            CsvWriter csvWriter = CsvUtil.getWriter(writer);
            for (T data : dataList) {
                String[] row = convertToRow(data);
                csvWriter.write(row);
            }
            // 确保所有数据写入输出流
            csvWriter.flush();
        }
    }

    public static <T> void exportZip(HttpServletResponse response, Map<Integer, List<T>> dataLists) throws IOException {
        // 设置响应内容类型
        response.setContentType("application/zip");
        response.setCharacterEncoding(StandardCharsets.UTF_8.name());
        String zipFileName = URLEncoder.encode("地区结果.zip", StandardCharsets.UTF_8);
        response.setHeader("Content-Disposition", "attachment;filename=\"" + zipFileName + "\"");

        // 创建ZIP输出流
        try (OutputStream out = response.getOutputStream();
             ZipOutputStream zipOut = new ZipOutputStream(out, StandardCharsets.UTF_8)) {

            for (Map.Entry<Integer, List<T>> listEntry : dataLists.entrySet()) {
                Integer fileName = listEntry.getKey();
                List<T> dataList = listEntry.getValue();
                // 创建ZIP条目
                ZipEntry zipEntry = new ZipEntry(fileName + ".csv");
                zipOut.putNextEntry(zipEntry);

                // 创建CSV写入器
                OutputStreamWriter writer = new OutputStreamWriter(zipOut, StandardCharsets.UTF_8);
                writer.write("\uFEFF");

                CsvWriter csvWriter = CsvUtil.getWriter(writer);

                // 写入数据
                for (T data : dataList) {
                    String[] row = convertToRow(data);
                    csvWriter.write(row);
                }
                csvWriter.flush(); // 确保所有数据写入输出流
                zipOut.closeEntry(); // 关闭当前条目
            }

        }
    }

    private static <T> String[] convertToRow(T data) {
        Field[] fields = data.getClass().getDeclaredFields();
        String[] row = new String[fields.length];

        try {
            for (int i = 0; i < fields.length; i++) {
                // 设置字段可访问
                fields[i].setAccessible(true);
                // 获取字段值
                Object value = fields[i].get(data);
                // 转换为字符串，处理null
                row[i] = value != null ? value.toString() : "";
            }
        } catch (IllegalAccessException e) {
            log.error("数据转换异常:{}",e.getMessage());
        }
        return row;
    }
}

3. 接口

java 复制代码

@Slf4j
@RestController
public class RegionExportController {

    @SneakyThrows
    @PostMapping("/api/v1/regionExport")
    public void exportVehicleResult(HttpServletResponse response) {
        // 模拟数据
        List<RegionImportDTO> list = RegionImportDTO.exportData();
        CsvUtils.export(response, "区域结果表", list);
    }


    @SneakyThrows
    @PostMapping("/api/v1/regionExportZip")
    public void exportZipVehicleResult(HttpServletResponse response) {
        // 模拟数据
        List<RegionImportDTO> list = RegionImportDTO.exportData();
        Map<Integer, List<RegionImportDTO>> listMap = list.stream().collect(Collectors.groupingBy(RegionImportDTO::getType));
        CsvUtils.exportZip(response, listMap);
    }
}