Springboot集成Hutool导出CSV的方法

1. 引入依赖

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

2. CSV工具类

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. 接口

@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);
    }
}

到此这篇关于Springboot集成Hutool导出CSV的方法的文章就介绍到这了,更多相关Springboot Hutool导出CSV内容请搜索QQ沐编程以前的文章或继续浏览下面的相关文章希望大家以后多多支持QQ沐编程！