SpringBoot+FTP下载文件,单文件直接下载,多文件进行打包zip下载

直接调用downloadFiles方法即可

package com.demo.utils;

import org.apache.commons.io.IOUtils;
import org.apache.commons.net.ftp.FTPClient;
import org.apache.commons.net.ftp.FTPFile;
import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestParam;

import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.util.List;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;

/**
 * FTP文件下载器
 *
 * @author lhj
 * @date 2024/04/15
 */
public class FtpFileDownloader {


    /**
     * 下载FTP文件
     *
     * @param filePaths 文件路径
     * @return 响应实体<?>
     * @throws IOException ioexception
     */
    @PostMapping("/downloadFtpFiles")
    public ResponseEntity<?> downloadFtpFiles(@RequestParam("filePaths") List<String> filePaths) throws IOException {
        // 实现下载逻辑...
        return null;
    }

    /**
     * 下载文件。该方法支持单个文件和多个文件的下载。当下载多个文件时，它们会被打包成一个zip文件进行下载。
     *
     * @param filePaths 文件路径列表。可以包含一个或多个文件路径。
     * @param response  响应对象，用于设置下载响应信息和输出文件内容。
     * @throws IOException 当发生IO异常时抛出。
     */
    public void downloadFiles(List<String> filePaths, HttpServletResponse response) throws IOException {
        FTPClient ftpClient = new FTPClient();
        try {
            // 连接到FTP服务器
            ftpClient.login("your_username", "your_password");

            // 判断是否需要打包下载
            if (filePaths.size() > 1) {
                // 设置响应类型为zip文件
                response.setContentType("application/zip");
                // 设置附件下载头，并指定下载文件名
                response.setHeader("Content-Disposition", "attachment; filename=\"files.zip\"");

                // 创建Zip输出流
                ZipOutputStream zos = new ZipOutputStream(response.getOutputStream());
                for (String filePath : filePaths) {
                    // 获取FTP文件信息
                    FTPFile ftpFile = ftpClient.listFiles(filePath)[0];
                    if (ftpFile.isFile()) {
                        // 获取文件输入流
                        InputStream inputStream = ftpClient.retrieveFileStream(filePath);
                        if (inputStream == null) {
                            continue; // 文件不存在或读取失败，跳过当前文件
                        }

                        // 添加文件到Zip输出流
                        ZipEntry zipEntry = new ZipEntry(getFileNameFromPath(filePath));
                        zos.putNextEntry(zipEntry);

                        // 将文件内容写入Zip输出流
                        IOUtils.copy(inputStream, zos);
                        zos.closeEntry();
                        // 关闭文件输入流
                        inputStream.close();
                        // 必须加这行,不然第二次循环路径不对  调用此方法来尝试完成未完成的命令。它主要用于恢复连接和确保数据传输的完整性。
                        ftpClient.completePendingCommand();
                    }
                }
                // 结束Zip输出流并关闭
                zos.finish();
                zos.close();
            } else if (filePaths.size() == 1) {
                // 单个文件下载逻辑
                String singleFilePath = filePaths.get(0);
                FTPFile ftpFile = ftpClient.listFiles(singleFilePath)[0];
                if (ftpFile.isFile()) {
                    // 设置响应类型为文件类型
                    response.setContentType(getContentTypeFromFileName(singleFilePath));
                    // 设置附件下载头，并指定下载文件名
                    response.setHeader("Content-Disposition", "attachment; filename=\"" + getFileNameFromPath(singleFilePath) + "\"");
                    // 创建输出流
                    OutputStream outputStream = response.getOutputStream();
                    // 获取文件输入流
                    InputStream inputStream = ftpClient.retrieveFileStream(singleFilePath);
                    // 将文件内容写入输出流
                    IOUtils.copy(inputStream, outputStream);
                    // 关闭输入输出流
                    inputStream.close();
                    outputStream.flush();
                }
            }
        } catch (IOException e) {
            // 抛出IO异常
            throw e;
        } finally {
            // 断开FTP连接并登出
            ftpClient.logout();
            ftpClient.disconnect();
        }
    }


    /**
     * 从路径获取文件名
     *
     * @param path 路径
     * @return 字符串
     */
    private String getFileNameFromPath(String path) {
        return path.substring(path.lastIndexOf("/") + 1);
    }

    /**
     * 从文件名获取内容类型
     *
     * @param fileName 文件名称
     * @return 字符串
     */
    private String getContentTypeFromFileName(String fileName) {
        // 根据文件扩展名返回contentType，这里只是一个简单示例，实际项目中可能需要更完善的映射规则
        return fileName.endsWith(".pdf") ? "application/pdf" : "application/octet-stream";
    }
}