FTP工具类

package cn.tdfx.cms.clone.utils;


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.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Value;

import java.io.*;
import java.util.TimeZone;

public class FTPUtil {

    private FTPClient ftpClient;
    private String strIp;
    private int intPort;
    private String user;
    private String password;

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

    /**
     * 本地字符编码
     */
    private static String LOCAL_CHARSET = "UTF-8";

    /**
     * Ftp构造函数
     *
     * @param strIp
     * @param intPort
     * @param user
     * @param Password
     */
    public FTPUtil(String strIp, int intPort, String user, String Password) {
        this.strIp = strIp;
        this.intPort = intPort;
        this.user = user;
        this.password = Password;
        this.ftpClient = new FTPClient();
    }

    /**
     * @return 判断是否登入成功
     */
    public boolean ftpLogin() throws IOException {
        boolean isLogin = false;
        FTPClientConfig ftpClientConfig = new FTPClientConfig();

//        if (FTPReply.isPositiveCompletion(ftpClient.sendCommand(
//                "OPTS UTF8", "ON"))) {// 开启服务器对UTF-8的支持,如果服务器支持就用UTF-8编码,否则就使用本地编码(GBK).
//            LOCAL_CHARSET = "UTF-8";
//        }
        ftpClient.setControlEncoding(LOCAL_CHARSET);

        ftpClientConfig.setServerTimeZoneId(TimeZone.getDefault().getID());
        this.ftpClient.setControlEncoding(LOCAL_CHARSET);
        this.ftpClient.configure(ftpClientConfig);
        try {
            if (this.intPort > 0) {
                this.ftpClient.connect(this.strIp, this.intPort);
            } else {
                this.ftpClient.connect(this.strIp);
            }
            // FTP服务器连接回答
            int reply = this.ftpClient.getReplyCode();
            if (!FTPReply.isPositiveCompletion(reply)) {
                this.ftpClient.disconnect();
                logger.error("登录FTP服务失败!");
                return isLogin;
            }
            this.ftpClient.login(this.user, this.password);
            // 设置传输协议
            this.ftpClient.enterLocalPassiveMode();
            this.ftpClient.setFileType(FTPClient.BINARY_FILE_TYPE);
            logger.info("恭喜" + this.user + "成功登陆FTP服务器");
            isLogin = true;
        } catch (Exception e) {
            e.printStackTrace();
            logger.error(this.user + "登录FTP服务失败!" + e.getMessage());
        }
        this.ftpClient.setBufferSize(1024 * 2);
        this.ftpClient.setDataTimeout(30 * 1000);
        return isLogin;
    }

    /**
     * @退出关闭服务器链接
     */
    public void ftpLogOut() {
        if (null != this.ftpClient && this.ftpClient.isConnected()) {
            try {
                // 退出FTP服务器
                boolean result = this.ftpClient.logout();
                if (result) {
                    logger.info("成功退出服务器");
                }
            } catch (IOException e) {
                e.printStackTrace();
                logger.warn("退出FTP服务器异常!" + e.getMessage());
            } finally {
                try {
                    // 关闭FTP服务器的连接
                    this.ftpClient.disconnect();
                } catch (IOException e) {
                    e.printStackTrace();
                    logger.warn("关闭FTP服务器的连接异常!");
                }
            }
        }
    }

    /***
     * 上传Ftp文件
     * @param localFile 当地文件
     * @param romotUpLoadePath 上传服务器路径 - 应该以/结束
     * */
    public boolean uploadFile(File localFile, String romotUpLoadePath) {
        BufferedInputStream inStream = null;
        boolean success = false;
        try {
            // 改变工作路径

            boolean b = this.ftpClient.changeWorkingDirectory(romotUpLoadePath);
            inStream = new BufferedInputStream(new FileInputStream(localFile));
            logger.info(localFile.getName() + "开始上传.....");
            success = this.ftpClient.storeFile(localFile.getName(), inStream);
            if (success == true) {
                logger.info(localFile.getName() + "上传成功");
                return success;
            }
        } catch (FileNotFoundException e) {
            e.printStackTrace();
            logger.error(localFile + "未找到");
        } catch (IOException e) {
            e.printStackTrace();
        } finally {
            if (inStream != null) {
                try {
                    inStream.close();
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
        }
        return success;
    }

    /***
     * 下载文件
     * @param remoteFileName   待下载文件名称
     * @param localDires 下载到当地那个路径下
     * @param remoteDownLoadPath remoteFileName所在的路径
     * */

    public boolean downloadFile(String remoteFileName, String localDires,
                                String remoteDownLoadPath) {
        String strFilePath = localDires + remoteFileName;
        BufferedOutputStream outStream = null;
        boolean success = false;
        try {
            boolean b = this.ftpClient.changeWorkingDirectory(remoteDownLoadPath);
            outStream = new BufferedOutputStream(new FileOutputStream(
                    strFilePath));
            logger.info(remoteFileName + "开始下载....");
            success = this.ftpClient.retrieveFile(remoteFileName, outStream);
            if (success == true) {
                logger.info(remoteFileName + "成功下载到" + strFilePath);
                return success;
            }
        } catch (Exception e) {
            e.printStackTrace();
            logger.error(remoteFileName + "下载失败");
        } finally {
            if (null != outStream) {
                try {
                    outStream.flush();
                    outStream.close();
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
        }
        if (success == false) {
            logger.error(remoteFileName + "下载失败!!!");
        }
        return success;
    }

    /***
     * @上传文件夹
     * @param localDirectory    当地文件夹
     * @param targetPath    Ftp 服务器路径 以目录"/"结束
     * */
    public boolean uploadDirectory(String localDirectory, String targetPath, Boolean isRoot) {
        try {
            if (StringUtils.isNotEmpty(targetPath) && isRoot) {
                String mkdirsStr = targetPath;
                if (targetPath.contains("tdfx/upload")) {
                    mkdirsStr = targetPath.split("a/b")[1];
                }
                String[] pah = mkdirsStr.split("/");
                // 先创建目录,在转到当前目录,再保存文件
                // 分层创建目录
                for (String pa : pah) {
                    if (StringUtils.isNotEmpty(pa)) {
                        // 每创建一层文件夹
                        this.ftpClient.makeDirectory(pa);
                        // 切到到对应目录
                        this.ftpClient.changeWorkingDirectory(pa);
                        logger.info("创建文件夹 : " + pa);
                        logger.info("切换目录 : " + pa);
                    }
                }
                uploadDirectory(localDirectory, targetPath, false);
            }
            File file = new File(localDirectory);
            if (!file.exists()) {
                logger.info(localDirectory + "源文件不存在");
                return false;
            }

            boolean b = this.ftpClient.changeWorkingDirectory(targetPath);
            String fileName = file.getName();
            boolean makeDirFlag = this.ftpClient.makeDirectory(fileName);
            targetPath = targetPath + "/" + fileName;
            logger.info("localDirectory : " + localDirectory);
            logger.info("remoteDirectoryPath : " + targetPath);
            logger.info("changeWorkingDirectory : " + b);
            logger.info("file.getName() : " + file.getName());
            logger.info("makeDirFlag : " + makeDirFlag);
//            if (!makeDirFlag) {
//                logger.info(fileName + "目录已存在");
//                return false;
//            }
            if (makeDirFlag) {
                File[] allFile = file.listFiles();
                for (int currentFile = 0; currentFile < allFile.length; currentFile++) {
                    if (!allFile[currentFile].isDirectory()) {
                        String srcName = allFile[currentFile].getPath().toString();
                        uploadFile(new File(srcName), targetPath);
                    }
                }
                for (int currentFile = 0; currentFile < allFile.length; currentFile++) {
                    if (allFile[currentFile].isDirectory()) {
                        // 递归
                        uploadDirectory(allFile[currentFile].getPath().toString(),
                                targetPath, false);
                    }
                }
            }
        } catch (Exception e) {
            e.printStackTrace();
            logger.info(targetPath + "上传模板失败");
            return false;
        }
        return true;
    }

    /***
     * @下载文件夹
     * @param localDirectoryPath 本地地址
     * @param remoteDirectory 远程文件夹
     * */
    public boolean downLoadDirectory(String localDirectoryPath, String remoteDirectory) {
        try {
            String fileName = new File(remoteDirectory).getName();
            localDirectoryPath = localDirectoryPath + fileName + "//";
            new File(localDirectoryPath).mkdirs();
            FTPFile[] allFile = this.ftpClient.listFiles(remoteDirectory);
            for (int currentFile = 0; currentFile < allFile.length; currentFile++) {
                if (!allFile[currentFile].isDirectory()) {
                    downloadFile(allFile[currentFile].getName(), localDirectoryPath, remoteDirectory);
                }
            }
            for (int currentFile = 0; currentFile < allFile.length; currentFile++) {
                if (allFile[currentFile].isDirectory()) {
                    String strremoteDirectoryPath = remoteDirectory + "/" + allFile[currentFile].getName();
                    downLoadDirectory(localDirectoryPath, strremoteDirectoryPath);
                }
            }
        } catch (IOException e) {
            e.printStackTrace();
            logger.info("下载文件夹失败");
            return false;
        }
        return true;
    }

    public static void main(String[] args) throws IOException {
        FTPUtil ftp = new FTPUtil("192.168.1.1", 21, "abc", "123456");
        ftp.ftpLogin();
        //上传文件夹
        //如果是admin/那么传的就是所有文件,如果只是/那么就是传文件夹
        boolean uploadFlag = ftp.uploadDirectory("F:\\a\\Templets", "/a/b/lfzzb", true);
        System.out.println("uploadFlag : " + uploadFlag);
        //下载文件夹
//        ftp.downLoadDirectory("F:\\a\\", "/a/b/c/Templets/");
        ftp.ftpLogOut();
    }
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值