FTP测试工具类

因项目现场无法部署IDE工具联调FTP服务器，开发个简单的小工具，打成jar部署联调测。一下是该工具的源代码。

package cn.org.july.ftp;

import org.apache.commons.net.ftp.FTPClient;
import org.apache.commons.net.ftp.FTPFile;
import org.apache.commons.net.ftp.FTPReply;

import java.io.*;
import java.net.SocketException;
import java.util.Scanner;

public class FtpTest {

    private static String address = "";
    private static int port = 21;
    private static String userName;
    private static String pwd;
    private static FTPClient ftpClient;
    private static String userDir = System.getProperty("user.dir");


    public static void init() {
        Scanner scan = new Scanner(System.in);
        System.out.println("please enter the FTP service address: (请输入FTP服务地址：)");
        address = scan.nextLine();
        System.out.println("ftp address is ".concat(address));
        System.out.println("please enter the FTP service port:(请输入FTP服务端口：)");
        port = Integer.valueOf(scan.nextLine());
        System.out.println("ftp port is ".concat(port + ""));
        System.out.println("please enter the FTP service user: (请输入FTP服务用户：)");
        userName = scan.nextLine();
        System.out.println("ftp user is ".concat(userName));
        System.out.println("please enter the FTP service pwd: (请输入FTP服务用户密码：)");
        pwd = scan.nextLine();
        System.out.println("ftp pwd is ".concat(pwd));
    }

    public static void connect() {
        try {
            ftpClient = new FTPClient();
            ftpClient.setConnectTimeout(60000);
            ftpClient.connect(address, port);// 连接FTP服务器
            if (FTPReply.isPositiveCompletion(ftpClient.getReplyCode())) {
                if (ftpClient.login(userName, pwd)) {// 登陆FTP服务器
                    if (FTPReply.isPositiveCompletion(ftpClient.sendCommand(
                            "OPTS UTF8", "ON"))) {// 开启服务器对UTF-8的支持，如果服务器支持就用UTF-8编码，否则就使用本地编码（GBK）.
                        ftpClient.setControlEncoding("UTF-8");
                    } else {
                        ftpClient.setControlEncoding("GBK");
                    }
                    ftpClient.enterLocalPassiveMode();// 设置被动模式
                    ftpClient.setFileType(FTPClient.BINARY_FILE_TYPE);// 设置传输的模式，以二进制流的方式读取
                    ftpClient.enterLocalPassiveMode();
                    System.out.println("FTP服务连接成功！");
                } else {
                    System.out.println("FTP服务用户名或密码错误！");
                    disConnection();
                }
            } else {
                System.out.println("连接到FTP服务失败！");
                disConnection();
            }
        } catch (SocketException e) {
            e.printStackTrace();
            disConnection();
            System.out.println("FTP的IP地址可能错误，请正确配置。");
        } catch (IOException e) {
            e.printStackTrace();
            disConnection();
            System.out.println("FTP的端口错误,请正确配置。");
        }
    }


    /**
     * 关闭FTP服务链接
     *
     * @throws IOException
     */
    public static void disConnection() {
        try {
            if (ftpClient.isConnected()) {
                ftpClient.disconnect();
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
    }


    /**
     * 获取文件夹下的所有文件信息
     *
     * @param path 文件路径
     */
    public static void getFTPDirectoryFiles(String path) {
        FTPFile[] files = null;
        try {
            ftpClient.changeWorkingDirectory(path);
            files = ftpClient.listFiles();
            for (FTPFile file : files) {
                System.out.println(file.getName());
            }
        } catch (Exception e) {
            e.printStackTrace();
            System.out.println("FTP读取数据异常！");
        } finally {
            //关闭连接
            disConnection();
        }
    }

    /**
     * 获取文件夹下的所有文件信息
     *
     * @param path 文件路径
     */
    public static void getFTPFile(String path, String fileName) {
        InputStream in = null;
        try {
            ftpClient.changeWorkingDirectory(path);
            FTPFile[] files = ftpClient.listFiles();
            if (files.length > 0) {
                in = ftpClient.retrieveFileStream(new String(fileName.getBytes(), "ISO-8859-1"));
            }
            writeToLocal("ftpFile".concat(File.separator).concat(System.currentTimeMillis() + "").concat(fileName), in);
        } catch (Exception e) {
            e.printStackTrace();
            System.out.println("FTP读取数据异常！");
        } finally {
            //关闭连接
            disConnection();
        }
    }

    /**
     * 将InputStream写入本地文件
     *
     * @param destination 写入本地目录
     * @param input       输入流
     * @throws IOException IOException
     */
    public static void writeToLocal(String destination, InputStream input)
            throws IOException {
        int index;
        byte[] bytes = new byte[1024];
        String downPath = userDir.concat(File.separator).concat(destination);
        File file = new File(downPath);
        if (!file.exists()) {
            if (!file.getParentFile().exists()) {
                file.getParentFile().mkdir();
            }
            file.createNewFile();
        }
        FileOutputStream downloadFile = new FileOutputStream(downPath);
        while ((index = input.read(bytes)) != -1) {
            downloadFile.write(bytes, 0, index);
            downloadFile.flush();
        }
        input.close();
        downloadFile.close();
        System.out.println("文件下载路径：" + downPath);
    }


    public static void operation() throws FileNotFoundException {
        System.out.println("1:下载操作。2：上传操作。3：遍历目录下文件");
        Scanner scan = new Scanner(System.in);
        String exit = scan.nextLine();
        if ("1".equals(exit)) {
            System.out.println("ftp download path (下载路径：)");
            String path = scan.nextLine();
            System.out.println("ftp download path is ".concat(path));
            System.out.println("ftp download fileName(下载文件名称：) ");
            String fileName = scan.nextLine();
            System.out.println("ftp download fileName is ".concat(fileName));
            getFTPFile(path, fileName);
        } else if ("2".equals(exit)) {
            System.out.println("please enter the local file path（请输入本地文件路径E:/test.pdf）");
            String localPath = scan.nextLine();
            System.out.println("local file path is ".concat(localPath));
            System.out.println("please enter the ftp file path (请输入ftp文件路径)");
            String ftpPath = scan.nextLine();
            System.out.println("ftp path is ".concat(ftpPath));
            System.out.println("please enter ftp file name (请输入ftp文件名称)");
            String fileName = scan.nextLine();
            System.out.println("ftp file name is ".concat(fileName));
            boolean b = uploadFile(ftpPath, fileName, new FileInputStream(new File(localPath)));
            if (b) {
                System.out.println("file upload success");
            } else {
                System.out.println("file upload fail");
            }
        } else if ("3".equals(exit)) {
            System.out.println("please enter ftp path:(请输入FTP文件路径)");
            String path = scan.nextLine();
            getFTPDirectoryFiles(path);
        } else {
            System.out.println("操作类型错误。");
        }
    }


    /**
     * Description: 向FTP服务器上传文件
     *
     * @param ftpPath  FTP服务器中文件所在路径 格式： ftptest/aa
     * @param fileName ftp文件名称
     * @param input    文件流
     * @return 成功返回true，否则返回false
     */
    public static boolean uploadFile(String ftpPath,
                                     String fileName, InputStream input) {
        boolean success = false;
        try {
            int reply;
            reply = ftpClient.getReplyCode();
            if (!FTPReply.isPositiveCompletion(reply)) {
                ftpClient.disconnect();
                return success;
            }
            ftpClient.setControlEncoding("UTF-8"); // 中文支持
            ftpClient.setFileType(FTPClient.BINARY_FILE_TYPE);
            ftpClient.enterLocalPassiveMode();
            if (!ftpClient.changeWorkingDirectory(ftpPath)) {
                System.err.println("不存在");
                boolean makeDirectory = ftpClient.makeDirectory(ftpPath);
                boolean changeWorkingDirectory = ftpClient.changeWorkingDirectory(ftpPath);
                System.err.println(ftpPath + "创建：" + makeDirectory + ";切换：" + changeWorkingDirectory);
            } else {
                ftpClient.changeWorkingDirectory(ftpPath);
            }
            ftpClient.storeFile(fileName, input);
            input.close();
            ftpClient.logout();
            success = true;
        } catch (IOException e) {
            e.printStackTrace();
        } finally {
            if (ftpClient.isConnected()) {
                try {
                    ftpClient.disconnect();
                } catch (IOException ioe) {
                }
            }
        }
        return success;
    }

    public static void main(String[] args) {
        init();
        connect();
        try {
            operation();
        } catch (FileNotFoundException e) {
            e.printStackTrace();
        }
    }
}