Springboot集成Hutool導出CSV的方法

更新時間：2026年01月05日 11:04:41 作者：彭于晏Yan

文章介紹了如何引入依賴、使用CSV工具類以及實現(xiàn)接口的相關內容,本文通過實例代碼給大家介紹的非常詳細,感興趣的朋友跟隨小編一起看看吧

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);
            }
            // 確保所有數(shù)據(jù)寫入輸出流
            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("地區(qū)結果.zip", StandardCharsets.UTF_8);
        response.setHeader("Content-Disposition", "attachment;filename=\"" + zipFileName + "\"");
        // 創(chuàng)建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();
                // 創(chuàng)建ZIP條目
                ZipEntry zipEntry = new ZipEntry(fileName + ".csv");
                zipOut.putNextEntry(zipEntry);
                // 創(chuàng)建CSV寫入器
                OutputStreamWriter writer = new OutputStreamWriter(zipOut, StandardCharsets.UTF_8);
                writer.write("\uFEFF");
                CsvWriter csvWriter = CsvUtil.getWriter(writer);
                // 寫入數(shù)據(jù)
                for (T data : dataList) {
                    String[] row = convertToRow(data);
                    csvWriter.write(row);
                }
                csvWriter.flush(); // 確保所有數(shù)據(jù)寫入輸出流
                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("數(shù)據(jù)轉換異常:{}",e.getMessage());
        }
        return row;
    }
}

3. 接口

@Slf4j
@RestController
public class RegionExportController {
    @SneakyThrows
    @PostMapping("/api/v1/regionExport")
    public void exportVehicleResult(HttpServletResponse response) {
        // 模擬數(shù)據(jù)
        List<RegionImportDTO> list = RegionImportDTO.exportData();
        CsvUtils.export(response, "區(qū)域結果表", list);
    }
    @SneakyThrows
    @PostMapping("/api/v1/regionExportZip")
    public void exportZipVehicleResult(HttpServletResponse response) {
        // 模擬數(shù)據(jù)
        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內容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: