SpringBoot中使用EasyExcel并行导出多个excel文件并压缩zip后下载

在现代的应用开发中，使用Spring Boot来处理Excel文件是一种常见的需求。EasyExcel是一个非常强大的工具，它能够高效地读写Excel文件。在一些场景下，我们可能需要并行导出多个Excel文件，并最终将这些文件压缩为一个ZIP包供用户下载。本文将详细介绍如何在Spring Boot中实现这一功能，并提供必要的代码示例。

1. 引入依赖

首先，你需要在pom.xml中添加EasyExcel和ZIP相关的依赖。

<dependency>
    <groupId>com.alibaba</groupId>
    <artifactId>easyexcel</artifactId>
    <version>3.0.5</version>
</dependency>
<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-web</artifactId>
</dependency>

2. 创建数据模型

定义数据模型，以便后续生成Excel表格。下面我们创建一个简单的用户模型。

public class User {
    private String name;
    private int age;

    // getters and setters
}

3. 创建Excel导出服务

我们需要一个服务来处理Excel文件的生成。我们可以使用EasyExcel库来实现这个功能。

import com.alibaba.excel.EasyExcel;
import org.springframework.stereotype.Service;

import java.io.File;
import java.util.List;

@Service
public class ExcelExportService {

    public void exportUserData(String filename, List<User> users) {
        EasyExcel.write(filename, User.class).sheet("用户数据").doWrite(users);
    }
}

4. 并行导出多个Excel文件

为了并行生成多个Excel文件，我们可以使用Java的ExecutorService来管理线程池。

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import java.io.File;
import java.util.List;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;

@Service
public class MultiExcelExportService {

    @Autowired
    private ExcelExportService excelExportService;

    private final ExecutorService executorService = Executors.newFixedThreadPool(5);

    public void exportMultipleExcel(List<List<User>> userGroups, List<String> filenames) {
        for (int i = 0; i < userGroups.size(); i++) {
            final int index = i;
            executorService.submit(() -> {
                excelExportService.exportUserData(filenames.get(index), userGroups.get(index));
            });
        }
    }
}

5. 压缩ZIP文件

在导出完所有Excel文件之后，我们需要将其压缩为ZIP文件。

import java.io.*;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;

public class ZipUtils {

    public static void zipFiles(String zipFileName, String[] srcFiles) throws IOException {
        try (ZipOutputStream zos = new ZipOutputStream(new FileOutputStream(zipFileName))) {
            for (String srcFile : srcFiles) {
                File file = new File(srcFile);
                try (FileInputStream fis = new FileInputStream(file)) {
                    ZipEntry zipEntry = new ZipEntry(file.getName());
                    zos.putNextEntry(zipEntry);
                    byte[] buffer = new byte[1024];
                    int length;
                    while ((length = fis.read(buffer)) > 0) {
                        zos.write(buffer, 0, length);
                    }
                    zos.closeEntry();
                }
            }
        }
    }
}

6. 控制器层

最后，我们需要一个控制器来处理HTTP请求，触发导出和ZIP压缩。

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RestController;

import java.io.File;
import java.util.ArrayList;
import java.util.List;

@RestController
public class ExportController {

    @Autowired
    private MultiExcelExportService multiExcelExportService;

    @GetMapping("/export")
    public String export() {
        List<List<User>> userGroups = new ArrayList<>(); // 模拟用户数据
        // 假设填充userGroups数据

        List<String> filenames = new ArrayList<>();
        for (int i = 0; i < userGroups.size(); i++) {
            filenames.add("user_data_" + i + ".xlsx");
        }

        multiExcelExportService.exportMultipleExcel(userGroups, filenames);

        // 压缩为ZIP文件
        try {
            ZipUtils.zipFiles("exported_data.zip", filenames.toArray(new String[0]));
        } catch (IOException e) {
            e.printStackTrace();
            return "导出失败";
        }

        return "导出成功，文件已生成";
    }
}

总结

以上是一个使用Spring Boot和EasyExcel并行导出多个Excel文件并压缩成ZIP包的完整示例。在实际应用中，你可能需要进一步处理异常、调整线程池的大小和管理文件存储位置等问题。不过通过这一示例，你应该能够掌握基本的实现思路。这样的功能在数据分析、报表生成等场景中非常有用。