从ftp获取文件并生成压缩包

依赖

<dependency>
            <groupId>commons-net</groupId>
            <artifactId>commons-net</artifactId>
            <version>3.4</version>
        </dependency>

        <dependency>
            <groupId>org.apache.commons</groupId>
            <artifactId>commons-lang3</artifactId>
            <version>3.3.2</version>
        </dependency>

        <dependency>
            <groupId>org.apache.ant</groupId>
            <artifactId>ant</artifactId>
            <version>1.10.5</version>
        </dependency>

可选择多个路径进行压缩

依赖

<dependency>
  <groupId>org.apache.ant</groupId>
  <artifactId>ant</artifactId>
  <version>1.10.5</version>
</dependency>

可选择多个路径进行压缩

依赖

<dependency>
  <groupId>org.apache.ant</groupId>
  <artifactId>ant</artifactId>
  <version>1.10.5</version>
</dependency>

可选择多个路径进行压缩

package per.qiao.utils.ftp;


import org.apache.commons.net.ftp.FTPClient;
import org.apache.commons.net.ftp.FTPClientConfig;
import org.apache.commons.net.ftp.FTPFile;
import org.apache.commons.net.ftp.FTPReply;
import org.apache.tools.zip.ZipEntry;
import org.apache.tools.zip.ZipOutputStream;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.io.File;
import java.io.IOException;
import java.io.InputStream;

/**
 * Create by IntelliJ Idea 2018.2
 *
 * @author: qyp
 * Date: 2019-07-19 17:10
 */
public class FtpFileZipUtls {

    private final static Logger logger = LoggerFactory.getLogger(FtpFileZipUtls.class);

    /**
     * 加载一个非空文件夹
     */
    private static void loadFile(FTPClient ftp, String dirPath, ZipOutputStream zos) throws IOException {

        FTPFile[] ftpFiles = ftp.listFiles();
        // length小于0用于区分非空文件夹与空文件夹和文件的区别
        // 退到上层再进入
        if (ftpFiles.length <= 0 && ftp.changeToParentDirectory()) {
            logger.info("退回到: " + ftp.printWorkingDirectory());
            dirPath = dirPath.replaceAll("\\\\", "/");
            boolean b = ftp.changeWorkingDirectory(getFileName(dirPath.substring(dirPath.lastIndexOf("/") + 1)));
            logger.info("进入: " + ftp.printWorkingDirectory());
            // 是一个空文件夹
            if (b && (ftpFiles == null || ftpFiles.length <= 0)) {
                zos.putNextEntry(new ZipEntry(getFileName(dirPath) + File.separator));
                return;
            }
        }

        for (FTPFile f : ftpFiles) {
            if (f.isFile()) {
                InputStream in = ftp.retrieveFileStream(f.getName());
                zos.putNextEntry(new ZipEntry(getFileName(dirPath) + File.separator + f.getName()));
                addToZOS(in, zos);
                ftp.completePendingCommand();
            } else {
                if (ftp.changeWorkingDirectory(f.getName())) {
                    loadFile(ftp, dirPath + File.separator + f.getName(), zos);
                    // ftp指针移动到上一层
                    ftp.changeToParentDirectory();
                }

            }
        }
    }

    /**
     * 将文件夹中的 .替换为^^
     * @param sb
     * @return
     */
    private static void replaceFileName(StringBuilder sb, String onepath) {
        sb.replace(sb.indexOf(onepath), sb.indexOf(onepath) + onepath.length(), onepath.replaceAll("\\.", "^^"));
    }
    /**
     * 将文件夹路径中的 ^^ 替换成 .
     * @param sb
     * @return
     */
    private static String getFileName(String sb) {
        StringBuilder result = new StringBuilder(sb);
        while (result.indexOf("^^") > -1) {
            result.replace(result.indexOf("^^"), result.indexOf("^^") + 2, ".");
        }
        return result.toString();
    }

    /**
     * 切换目录 返回切换的层级数
     * @param sb
     * @param ftp
     * @return 切换的层级数
     * @throws IOException
     */
    private static int exchageDir(StringBuilder sb, FTPClient ftp) throws IOException {
        // 计入进入的文件夹的次数,方便回退
        int level = 0;
        String[] pathes = sb.toString().split("/");
        for (int i = 0, len = pathes.length; i < len; i++) {
            String onepath = pathes[i];
            if (onepath == null || "".equals(onepath.trim())) {
                continue;
            }
            //文件排除
            boolean flagDir = ftp.changeWorkingDirectory(onepath);
            if (flagDir) {
                level++;
                if (onepath.contains(".")) {
                    //把文件夹中的.用^^代替
                    replaceFileName(sb, onepath);
                }
                logger.info("成功连接ftp目录:" + ftp.printWorkingDirectory());
            } else {
                logger.warn("连接ftp目录失败:" + ftp.printWorkingDirectory());
            }
        }

        return level;
    }


    /**
     * 处理单个文件和空文件夹的情况  处理了返回true 未处理返回false
     * 注意:这里是指数组中给的路径直接是文件或者空文件夹的情况
     * @param ftp
     * @param path 需要压缩的文件(文件夹)路径(相对根)
     * @param zos
     * @return
     * @throws IOException
     */
    public static boolean delFile(FTPClient ftp, StringBuilder path, ZipOutputStream zos) throws IOException {

        int times = exchageDir(path, ftp);

        String fileRelativePaht = path.toString();
        String[] split = fileRelativePaht.split("/");
        // 是否处理过
        boolean isDel = false;
        //有父级目录
        if (split != null && split.length > 0) {

            String lastSegmeng = split[split.length - 1];

            //path直接是文件
            if (split[split.length - 1].contains(".")) {
                zos.putNextEntry(new ZipEntry(fileRelativePaht));
                addToZOS(ftp.retrieveFileStream(lastSegmeng), zos);
                ftp.completePendingCommand();
                isDel = true;
            } else if (ftp.listFiles().length <= 0) {
                // 空文件夹
                zos.putNextEntry(new ZipEntry(fileRelativePaht + File.separator));
                isDel = true;
            }
            // 如果被处理过(空文件夹或者直接给的文件的路径),退回到根路径(以免污染下次循环的ftp指针)
            if (isDel) {
                for (int i = 0; i < times; i++) {
                    ftp.changeToParentDirectory();
                }
            }
        }
        return isDel;
    }

    /**
     * 添加文件到压缩流
     * @param in  输入流(一般就直接是从FTP中获取的)
     * @param zos 压缩输出流
     * @throws IOException
     */
    public static void addToZOS(InputStream in, ZipOutputStream zos) throws IOException {

        byte[] buf = new byte[1024];
        //BufferedOutputStream bzos = new BufferedOutputStream(zos);
        try {
            for (int len; (len = (in.read(buf))) != -1; ) {
                zos.write(buf, 0 , len);
            }
        } catch (IOException e) {
            System.out.println("流转换异常");
            throw e;
        } finally {
            if (in != null) {
                try {
                    in.close();
                } catch (IOException e) {
                    System.out.println("关流异常");
                    e.printStackTrace();
                }
            }
        }
    }


    public static FTPClient getFtpClient(String path) throws Exception {

        String server = "127.0.0.1";
        int port = 21;
        String username = "test";
        String password = "test";
//        path = "/FTPStation/";
        FTPClient ftpClient = connectServer(server, port, username, password, path);
        return ftpClient;
    }

    /**
     *
     * @param server
     * @param port
     * @param username
     * @param password
     * @param path 连接的节点(相对根路径的文件夹)
     * @return
     */
    public static FTPClient connectServer(String server, int port, String username, String password, String path) throws IOException {
        path = path == null ? "" : path;
        FTPClient ftp = new FTPClient();

        //下面四行代码必须要,而且不能改变编码格式,否则不能正确下载中文文件
        // 如果使用serv-u发布ftp站点,则需要勾掉“高级选项”中的“对所有已收发的路径和文件名使用UTF-8编码”
        ftp.setControlEncoding("GBK");
        FTPClientConfig conf = new FTPClientConfig(FTPClientConfig.SYST_NT);
        conf.setServerLanguageCode("zh");
        ftp.configure(conf);

        // 判断ftp是否存在
        ftp.connect(server, port);
        ftp.setDataTimeout(2 * 60 * 1000);
        if (!FTPReply.isPositiveCompletion(ftp.getReplyCode())) {
            ftp.disconnect();
            System.out.println(server + "拒绝连接");
        }
        //登陆ftp
        boolean login = ftp.login(username, password);
        if (logger.isDebugEnabled()) {
            if (login) {
                logger.debug("登陆FTP成功! ip: " + server);
            } else {
                logger.debug("登陆FTP失败! ip: " + server);
            }
        }

        //根据输入的路径,切换工作目录。这样ftp端的路径就可以使用相对路径了
        exchageDir(new StringBuilder(path), ftp);

        return ftp;
    }

    public static void main(String[] args) throws Exception {

        String[] paths = {"/CAD/你好.txt", "/CAD/1.第一层"};
        String savePath = "e:/temp/zz.zip";
        //连接到的节点
        String rootDir = "/CAD";
        FTPClient ftp = getFtpClient(rootDir);
        zipFileByPaths(ftp, savePath, paths);
    }

    /**
     * 多FTP路径压缩
     * @param ftp
     * @param savePath
     * @param filePaths
     * @throws Exception
     */
    public static void zipFileByPaths(FTPClient ftp, String savePath, String[] filePaths) throws Exception {

        try (ZipOutputStream zos = ZipUtils.getOutPutStream(savePath)) {
            for (String path : filePaths) {
                StringBuilder sb = new StringBuilder(path);
                //delFile时进入的文件夹的路径(当前需要遍历的文件的路径)
                String before = ftp.printWorkingDirectory();
                if (delFile(ftp, sb, zos)) {
                    continue;
                }
                loadFile(ftp, sb.toString(), zos);
                // 文件加载完毕后,需要将ftp的指针退到开始状态
                String after = ftp.printWorkingDirectory();
                backPath(before, after, ftp);
            }
            zos.flush();
        } catch (Exception e) {
            logger.error("多路径文件压缩失败");
            e.printStackTrace();
            throw e;
        }
    }

    /**
     * ftp 指针回退
     * @param end
     * @param start
     * @param ftp
     * @throws IOException
     */
    private static void backPath(String end, String start, FTPClient ftp) throws IOException {
        long startTime = System.currentTimeMillis();
        do {
            if (!end.equals(start) && ftp.changeToParentDirectory()) {
                start = ftp.printWorkingDirectory();
            }
            // 防止意外出现空循环 30s等待时间
            if (System.currentTimeMillis() - startTime > 30 * 1000) {
                break;
            }
        } while (!end.equals(start));
    }
}


posted @ 2019-07-20 02:15  复合式→展开式  阅读(2100)  评论(1编辑  收藏  举报