从MultipartFile处理ZIP文件：无需本地路径的临时提取策略

import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.multipart.MultipartFile;

import java.io.*;
import java.nio.file.Files;
import java.util.Objects;
import java.util.zip.ZipEntry;
import java.util.zip.ZipInputStream;

@RestController
public class ZipFileUploadController {

    private static final int BUFFER_SIZE = 1024; // 定义缓冲区大小

    /**
     * 处理上传的ZIP文件，提取其内容到临时目录并进行处理。
     *
     * @param multipartFile 上传的ZIP文件
     * @return 响应实体
     * @throws IOException 如果文件操作失败
     */
    @PostMapping("/upload-zip")
    public ResponseEntity<?> uploadZipFile(@RequestParam("file") MultipartFile multipartFile) throws IOException {
        // 1. 创建一个临时目录来存放解压后的文件
        File unzippedFolder = createTempZipFile(multipartFile);

        try {
            // 2. 遍历临时目录中的文件并进行业务处理
            // 注意：listFiles() 可能返回 null，需要进行检查
            File[] extractedFiles = unzippedFolder.listFiles();
            if (extractedFiles != null) {
                for (File file : extractedFiles) {
                    if (file.isFile()) { // 确保处理的是文件而不是子目录
                        processExtractedFile(file);
                    }
                }
            } else {
                System.out.println("No files extracted or directory is empty.");
            }
            return ResponseEntity.ok("ZIP file processed successfully.");
        } finally {
            // 3. 确保临时目录在应用关闭时被删除
            // 对于Web请求，通常依赖OS的临时文件清理机制，但显式标记 deleteOnExit 更保险
            // 注意：deleteOnExit() 只能删除空目录，如果目录非空，需要递归删除
            // Files.walkFileTree 提供了更健壮的删除方式，但此处简化为仅删除根目录
            // 实际生产环境，可以考虑在请求结束后异步清理，或依赖操作系统
            // 简单起见，此处不进行递归删除，仅依赖OS清理
            // 或者使用 try-with-resources 配合 Files.deleteIfExists 进行更精细的控制
            // 但 Files.createTempDirectory 创建的目录通常由OS管理其生命周期
            // unzippedFolder.deleteOnExit(); // 仅对空目录有效
            // 更安全的删除（如果需要立即删除）
            // deleteDirectory(unzippedFolder.toPath());
        }
    }

    /**
     * 将上传的ZIP文件解压到临时目录中。
     *
     * @param file 上传的MultipartFile
     * @return 包含解压后文件的临时目录
     * @throws IOException 如果解压过程中发生错误
     */
    private File createTempZipFile(MultipartFile file) throws IOException {
        // 创建一个带有 "data" 前缀的临时目录
        File tempDir = Files.createTempDirectory("data").toFile();
        System.out.println("Created temporary directory: " + tempDir.getAbsolutePath());

        byte[] buffer = new byte[BUFFER_SIZE];

        // 使用 try-with-resources 确保 ZipInputStream 和 FileOutputStream 正确关闭
        try (ZipInputStream zis = new ZipInputStream(file.getInputStream())) {
            ZipEntry zipEntry;

            // 遍历ZIP文件中的每个条目
            while ((zipEntry = zis.getNextEntry()) != null) {
                // 忽略目录条目
                if (!zipEntry.isDirectory()) {
                    // 构建目标文件路径，确保不包含路径遍历漏洞
                    File newFile = new File(tempDir, zipEntry.getName());

                    // 确保父目录存在
                    // Files.createDirectories(newFile.getParentFile().toPath()); // 如果zipEntry.getName()包含路径
                    // 对于简单的zipEntry.getName()，通常不需要此步，因为文件直接在tempDir下

                    // 写入文件内容
                    try (FileOutputStream fos = new FileOutputStream(newFile)) {
                        int len;
                        while ((len = zis.read(buffer)) > 0) {
                            fos.write(buffer, 0, len);
                        }
                    }
                    System.out.println("Extracted file: " + newFile.getAbsolutePath());
                }
                zis.closeEntry(); // 关闭当前条目
            }
        }
        return tempDir;
    }

    /**
     * 处理从ZIP文件中提取出的单个文件。
     *
     * @param file 提取出的文件
     * @throws IOException 如果文件读取失败
     */
    private void processExtractedFile(File file) throws IOException {
        System.out.println("Processing file: " + file.getName());
        try (BufferedReader br = new BufferedReader(new FileReader(file))) {
            String line;
            while ((line = br.readLine()) != null) {
                // 在这里执行你的业务逻辑，例如：
                // - 解析每一行数据
                // - 将数据写入数据库
                // - 进行数据验证等
                System.out.println("  Content: " + line);
            }
        }
        // 处理完成后，可以选择删除此单个临时文件
        // file.delete(); // 如果需要立即删除，否则等待OS清理
    }

    // 辅助方法：递归删除目录（如果需要立即删除临时目录）
    /*
    private void deleteDirectory(Path path) throws IOException {
        Files.walk(path)
             .sorted(Comparator.reverseOrder())
             .map(Path::toFile)
             .forEach(File::delete);
    }
    */
}