SFTP多文件上傳,刪除

　　公司項目中需要把項目的相關文件上傳到服務器的tomcat中,需要在項目中進行以下幾步操作:

　　1.添加項目信息,包括名稱,描述,服務器ip,sftp的用戶名,密碼,端口號等,存在配置,部署,刪除等操作

　　2.配置:顯示出文件信息,包括文件路徑,目標路徑,類型(上傳,刪除),狀態(是否部署),

　　3.點擊部署時進行自動的部署,可以是文件上傳,也可以是相關文件的刪除

 

結合網上有關sftp完成的sftp工具類,只使用了多文件上傳和文件刪除功能,其他沒有進行測試,多文件上傳需要本地路徑和遠程路徑參數,文件夾刪除需要遠程路徑參數

package MyUtils;

import com.jcraft.jsch.*;
import org.apache.log4j.Logger;

import java.io.*;
import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;
import java.util.Vector;

/**
 * sftp工具類
 *

 */
public class SFTPUtils {
    private static Logger log = Logger.getLogger(SFTPUtils.class.getName());
    private static ChannelSftp sftp = null;
    private static Session sshSession = null;
    private static Integer i = 0;


    /**
     * 通過SFTP連接服務器
     */
    public static void connect(String ip, String username, String password, Integer port) throws Exception {

        JSch jsch = new JSch();
        try {
            if (port <= 0) {
                // 連接服務器，采用默認端口
                sshSession = jsch.getSession(username, ip);
            } else {
                // 采用指定的端口連接服務器
                sshSession = jsch.getSession(username, ip, port);
            }

            // 如果服務器連接不上，則拋出異常
            if (sshSession == null) {
                throw new Exception("服務器異常");
            }

            // 設置登陸主機的密碼
            sshSession.setPassword(password);// 設置密碼
            // 設置第一次登陸的時候提示，可選值：(ask | yes | no)
            sshSession.setConfig("StrictHostKeyChecking", "no");
            // 設置登陸超時時間
            sshSession.connect(300000);
            Channel channel = sshSession.openChannel("sftp");
            channel.connect();

            sftp = (ChannelSftp) channel;

        } catch (Exception e) {
            e.printStackTrace();
            throw e;
        }
    }

    /**
     * 關閉連接
     */
    public static void disconnect() {
        if (sftp != null) {
            if (sftp.isConnected()) {
                sftp.disconnect();
                if (log.isInfoEnabled()) {
                    log.info("已關閉sftp");
                }
            }
        }
        if (sshSession != null) {
            if (sshSession.isConnected()) {
                sshSession.disconnect();
                if (log.isInfoEnabled()) {
                    log.info("已關閉sshSession");
                }
            }
        }
    }

    /**
     * 批量下載文件
     *
     * @param remotePath：遠程下載目錄(以路徑符號結束)
     *
     * @param localPath：本地保存目錄(以路徑符號結束,D:\Duansha\sftp\)
     *
     * @param fileFormat：下載文件格式(以特定字符開頭,為空不做檢驗)
     *
     * @param fileEndFormat：下載文件格式(文件格式)
     * @param del：下載后是否刪除sftp文件
     * @return
     */
    public List<String> batchDownLoadFile(String remotePath, String localPath,
                                          String fileFormat, String fileEndFormat, boolean del) throws SftpException {
        List<String> filenames = new ArrayList<String>();

        Vector v = sftp.ls(remotePath);
        if (v.size() > 0) {
            System.out.println("本次處理文件個數不為零,開始下載...fileSize=" + v.size());
            Iterator it = v.iterator();
            while (it.hasNext()) {
                ChannelSftp.LsEntry entry = (ChannelSftp.LsEntry) it.next();
                String filename = entry.getFilename();
                SftpATTRS attrs = entry.getAttrs();
                if (!attrs.isDir()) {
                    boolean flag = false;
                    String localFileName = localPath + filename;
                    fileFormat = fileFormat == null ? "" : fileFormat
                            .trim();
                    fileEndFormat = fileEndFormat == null ? ""
                            : fileEndFormat.trim();
                    // 三種情況
                    if (fileFormat.length() > 0 && fileEndFormat.length() > 0) {
                        if (filename.startsWith(fileFormat) && filename.endsWith(fileEndFormat)) {
                            flag = downloadFile(remotePath, filename, localPath, filename);
                            if (flag) {
                                filenames.add(localFileName);
                                if (flag && del) {
                                    deleteSFTP(remotePath);
                                }
                            }
                        }
                    } else if (fileFormat.length() > 0 && "".equals(fileEndFormat)) {
                        if (filename.startsWith(fileFormat)) {
                            flag = downloadFile(remotePath, filename, localPath, filename);
                            if (flag) {
                                filenames.add(localFileName);
                                if (flag && del) {
                                    deleteSFTP(remotePath);
                                }
                            }
                        }
                    } else if (fileEndFormat.length() > 0 && "".equals(fileFormat)) {
                        if (filename.endsWith(fileEndFormat)) {
                            flag = downloadFile(remotePath, filename, localPath, filename);
                            if (flag) {
                                filenames.add(localFileName);
                                if (flag && del) {
                                    deleteSFTP(remotePath);
                                }
                            }
                        }
                    } else {
                        flag = downloadFile(remotePath, filename, localPath, filename);
                        if (flag) {
                            filenames.add(localFileName);
                            if (flag && del) {
                                deleteSFTP(remotePath);
                            }
                        }
                    }
                }
            }
        }
        if (log.isInfoEnabled()) {
            log.info("download file is success:remotePath=" + remotePath
                    + "and localPath=" + localPath + ",file size is"
                    + v.size());
        }

        return filenames;
    }

    /**
     * 下載單個文件
     *
     * @param remotePath：遠程下載目錄(以路徑符號結束)e
     * @param remoteFileName：下載文件名
     * @param localPath：本地保存目錄(以路徑符號結束)
     * @param localFileName：保存文件名
     * @return
     */
    public boolean downloadFile(String remotePath, String remoteFileName, String localPath, String localFileName) {
        FileOutputStream fieloutput = null;
        try {
            // sftp.cd(remotePath);
            File file = new File(localPath + localFileName);
            // mkdirs(localPath + localFileName);
            fieloutput = new FileOutputStream(file);
            sftp.get(remotePath + remoteFileName, fieloutput);
            if (log.isInfoEnabled()) {
                log.info("===DownloadFile:" + remoteFileName + " success from sftp.");
            }
            return true;
        } catch (FileNotFoundException e) {
            e.printStackTrace();
        } catch (SftpException e) {
            e.printStackTrace();
        } finally {
            if (null != fieloutput) {
                try {
                    fieloutput.close();
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
        }
        return false;
    }

    /**
     * 上傳單個文件
     *
     * @param remotePath：遠程保存目錄
     * @param localPath：本地上傳目錄(以路徑符號結束)
     * @return
     */
    public static void uploadFile(String remotePath, String localPath) throws Exception {
        FileInputStream in = null;
        File localFile = new File(localPath);
        sftp.cd(remotePath);
        in = new FileInputStream(localFile);
        sftp.put(in, localFile.getName());

        if (in != null) {
            in.close();
        }
    }

    /**
     * 批量上傳文件
     *
     * @param remotePath：遠程保存目錄
     * @param localPath：本地上傳目錄(以路徑符號結束)
     * @return
     */
    public static boolean bacthUploadFile(String remotePath, String localPath) throws Exception {
        File localFile = new File(localPath);
        boolean flag = true;
        //進入遠程路徑
        try {
            if (!isDirExist(remotePath)) {
                sftp.mkdir(remotePath);
                sftp.cd(remotePath);
            } else {
                sftp.cd(remotePath);
            }
            //本地文件上傳
            File file = new File(localPath);
            //本地文件上傳方法
            copyFile(file, sftp.pwd());


        } catch (Exception e) {
            e.printStackTrace();
            flag = false;
            throw e;
        }

        return flag;
    }

    private static void copyFile(File file, String pwd) throws Exception {
        if (file.isDirectory()) {
            File[] list = file.listFiles();
            String fileName = file.getName();
            sftp.cd(pwd);
            System.out.println("正在創建目錄:" + sftp.pwd() + "/" + fileName);
            sftp.mkdir(fileName);
            System.out.println("目錄創建成功:" + sftp.pwd() + "/" + fileName);
            //遠程路徑發生改變
            pwd = pwd + "/" + file.getName();
            sftp.cd(file.getName());

            for (int i = 0; i < list.length; i++) {
                copyFile(list[i], pwd);
            }
        } else {
            //不是目錄,直接進入改變后的遠程路徑,進行上傳
            sftp.cd(pwd);

            System.out.println("正在復制文件:" + file.getAbsolutePath());
            InputStream instream = null;
            OutputStream outstream = null;
            outstream = sftp.put(file.getName());
            instream = new FileInputStream(file);

            byte b[] = new byte[1024];
            int n;
            while ((n = instream.read(b)) != -1) {
                outstream.write(b, 0, n);
            }

            outstream.flush();
            outstream.close();
            instream.close();

        }

    }

    /**
     * 刪除本地文件
     *
     * @param filePath
     * @return
     */
    public boolean deleteFile(String filePath) {
        File file = new File(filePath);
        if (!file.exists()) {
            return false;
        }

        if (!file.isFile()) {
            return false;
        }
        boolean rs = file.delete();
        if (rs && log.isInfoEnabled()) {
            log.info("delete file success from local.");
        }
        return rs;
    }

    /**
     * 創建目錄
     *
     * @param createpath
     * @return
     */
    public static void createDir(String createpath) {
        try {
            if (isDirExist(createpath)) {
                sftp.cd(createpath);
            }
            String pathArry[] = createpath.split("/");
            StringBuffer filePath = new StringBuffer("/");
            for (String path : pathArry) {
                if (path.equals("")) {
                    continue;
                }
                filePath.append(path + "/");
                if (isDirExist(filePath.toString())) {
                    sftp.cd(filePath.toString());
                } else {
                    // 建立目錄
                    sftp.mkdir(filePath.toString());
                    // 進入並設置為當前目錄
                    sftp.cd(filePath.toString());
                }

            }
        } catch (SftpException e) {
            e.printStackTrace();
        }
    }

    /**
     * 判斷目錄是否存在
     *
     * @param directory
     * @return
     */
    public static boolean isDirExist(String directory) {
        try {
            Vector<?> vector = sftp.ls(directory);
            if (null == vector) {
                return false;
            } else {
                return true;
            }
        } catch (Exception e) {
            return false;
        }
    }

    /**
     * 刪除stfp文件
     *
     * @param directory：要刪除文件所在目錄
     */
    public static void deleteSFTP(String directory) {
        try {
            if (isDirExist(directory)) {
                Vector<ChannelSftp.LsEntry> vector = sftp.ls(directory);
                if (vector.size() == 1) { // 文件，直接刪除
                    sftp.rm(directory);
                } else if (vector.size() == 2) { // 空文件夾，直接刪除
                    sftp.rmdir(directory);
                } else {
                    String fileName = "";
                    // 刪除文件夾下所有文件
                    for (ChannelSftp.LsEntry en : vector) {
                        fileName = en.getFilename();
                        if (".".equals(fileName) || "..".equals(fileName)) {
                            continue;
                        } else {
                            deleteSFTP(directory + "/" + fileName);
                        }
                    }
                    // 刪除文件夾
                    sftp.rmdir(directory);
                }
            }
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    /**
     * 如果目錄不存在就創建目錄
     *
     * @param path
     */
    public void mkdirs(String path) {
        File f = new File(path);

        String fs = f.getParent();

        f = new File(fs);

        if (!f.exists()) {
            f.mkdirs();
        }
    }



    /**
     * 測試
     */
    public static void main(String[] args) {
        // 本地存放地址
        String localPath = "F:\\java\\src\\main\\webapp\\resources";
        // Sftp下載路徑
        String sftpPath = "/home/sftp/";
        List<String> filePathList = new ArrayList<String>();
        try {
            connect("111.16.123.12", "ftptest1", "123456", 22);
            //deleteSFTP(sftpPath);
            // 上傳
            //bacthUploadFile(sftpPath, localPath);
            //deleteSFTP("/home/ftp/ff");
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            sftp.disconnect();
        }
    }
}