SFTP工具類

1.SFTP搭建方法：

地址：

http://www.jb51.net/article/101405.htm

https://blog.csdn.net/helloloser/article/details/79399575

 

2.SFTP工具類：

import com.jcraft.jsch.*;
import org.apache.commons.io.IOUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.io.*;
import java.util.Properties;
import java.util.Vector;
/**
 * sftp工具類 實現文件上傳、下載、刪除操做
 * @author  XIHONGLEI
 * @date 2018-03-26
 */
public class SftpUtil {
    private transient Logger log = LoggerFactory.getLogger(this.getClass());

    private ChannelSftp sftp;

    private Session session;
    /** SFTP 登陸用戶名*/
    private String username;
    /** SFTP 登陸密碼*/
    private String password;
    /** 私鑰 */
    private String privateKey;
    /** SFTP 服務器地址IP地址*/
    private String host;
    /** SFTP 端口*/
    private int port;


    /**
     * 構造基於密碼認證的sftp對象   
     */
    public SftpUtil(String username, String password, String host, int port) {
        this.username = username;
        this.password = password;
        this.host = host;
        this.port = port;
    }

    /**
     * 構造基於祕鑰認證的sftp對象  
     */
    public SftpUtil(String username, String host, int port, String privateKey) {
        this.username = username;
        this.host = host;
        this.port = port;
        this.privateKey = privateKey;
    }

    public SftpUtil(){}

    /**
     * 鏈接sftp服務器  
     */
    public void login(){
        try {
            JSch jsch = new JSch();
            if (privateKey != null) {
                // 設置私鑰
                jsch.addIdentity(privateKey);
            }
            session = jsch.getSession(username, host, port);
            if (password != null) {
                session.setPassword(password);
            }
            Properties config = new Properties();
            config.put("StrictHostKeyChecking", "no");

            session.setConfig(config);
            session.connect();

            Channel channel = session.openChannel("sftp");
            channel.connect();

            sftp = (ChannelSftp) channel;
        } catch (JSchException e) {
            e.printStackTrace();
        }
    }

    /**
     * 關閉鏈接 server   
     */
    public void logout(){
        if (sftp != null) {
            if (sftp.isConnected()) {
                sftp.disconnect();
            }
        }
        if (session != null) {
            if (session.isConnected()) {
                session.disconnect();
            }
        }
    }

    /**
     * 將輸入流的數據上傳到sftp做爲文件。文件完整路徑=basePath+directory 
     * @param basePath  服務器的基礎路徑  
     * @param directory  上傳到該目錄   
     * @param sftpFileName  sftp端文件名   
     * @param input   輸入流
     */
    public void upload(String basePath,String directory, String sftpFileName, InputStream input) throws SftpException{
        try {
            sftp.cd(basePath);
            sftp.cd(directory);
        } catch (SftpException e) {
            //目錄不存在，則建立文件夾  
            String [] dirs=directory.split("/");
            String tempPath=basePath;
            for(String dir:dirs){
                if(null == dir || "".equals(dir)) {
                    continue;
                }
                tempPath+="/"+dir;
                try{
                    sftp.cd(tempPath);
                }catch(SftpException ex){
                    sftp.mkdir(tempPath);
                    sftp.cd(tempPath);
                }
            }
        }
        //上傳文件
        sftp.put(input, sftpFileName);
    }

    /**
     * 下載文件。 
     * @param directory 下載目錄   
     * @param downloadFile 下載的文件  
     * @param saveFile 存在本地的路徑  
     */
    public void download(String directory, String downloadFile, String saveFile) throws SftpException, FileNotFoundException{
        if (directory != null && !"".equals(directory)) {
            sftp.cd(directory);
        }
        File file = new File(saveFile);
        sftp.get(downloadFile, new FileOutputStream(file));
    }

    /**
     * 判斷遠程SFTP服務器上是否存在某個文件
     * @param directory 目錄
     * @param fileName 文件名
     * @return 是否存在
     */
    public boolean isExists(String directory, String fileName){
        boolean isHave = false;
        try {
            sftp.cd(directory);
            SftpATTRS attrs = sftp.stat(fileName);
            if(attrs != null){
                isHave = true;
            }
        } catch (Exception e) {}
        return  isHave;
    }

    /**
     * 下載文件  
     * @param directory 下載目錄  
     * @param downloadFile 下載的文件名  
     * @return 字節數組
     */
    public byte[] download(String directory, String downloadFile) throws SftpException, IOException{
        if (directory != null && !"".equals(directory)) {
            sftp.cd(directory);
        }
        InputStream is = sftp.get(downloadFile);
        byte[] fileData = IOUtils.toByteArray(is);
        return fileData;
    }

    /**
     * 刪除文件  
     * @param directory 要刪除文件所在目錄  
     * @param deleteFile 要刪除的文件  
     */
    public void delete(String directory, String deleteFile) throws SftpException{
        sftp.cd(directory);
        sftp.rm(deleteFile);
    }

    /**
     * 列出目錄下的文件  
     * @param directory 要列出的目錄
     */
    public Vector<?> listFiles(String directory) throws SftpException {
        return sftp.ls(directory);
    }

    /**
     * 測試Main方法
     * @param args
     * @throws SftpException
     * @throws IOException
     */
    public static void main(String[] args) throws SftpException, IOException {
        SftpUtil sftp = new SftpUtil("zhnx","Zhnx$p=!@#z@n$h&x", "139.224.145.186", 22022);
        sftp.login();
       /* File file = new File("F:\\img\\timg.jpg");
        InputStream is = new FileInputStream(file);

        sftp.upload("/IN","", "timg.jpg", is);*/
        byte[] bytes = sftp.download("/IN","timg.jpg");
        ByteUtil.saveFile(bytes,"F:\\img2","timg2.jpg");
        sftp.logout();
    }
}

import java.io.*;

/**
 * byte數組工具類實現byte[]與文件之間的相互轉換
 * @author XIHONGLEI
 * @Date 2018-03-26
 */
public class ByteUtil {
    /**
     * 得到指定文件的byte數組
     */
    public static byte[] getBytes(String filePath){
        byte[] buffer = null;
        try {
            File file = new File(filePath);
            FileInputStream fis = new FileInputStream(file);
            ByteArrayOutputStream bos = new ByteArrayOutputStream(1000);
            byte[] b = new byte[1000];
            int n;
            while ((n = fis.read(b)) != -1) {
                bos.write(b, 0, n);
            }
            fis.close();
            bos.close();
            buffer = bos.toByteArray();
        } catch (FileNotFoundException e) {
            e.printStackTrace();
        } catch (IOException e) {
            e.printStackTrace();
        }
        return buffer;
    }

    /**
     * 根據byte數組，生成文件
     */
    public static void saveFile(byte[] bfile, String filePath,String fileName) {
        BufferedOutputStream bos = null;
        FileOutputStream fos = null;
        File file = null;
        try {
            File dir = new File(filePath);
            //判斷文件目錄是否存在
            if(!dir.exists()&&dir.isDirectory()){
                dir.mkdirs();
            }
            file = new File(filePath+"\\"+fileName);
            fos = new FileOutputStream(file);
            bos = new BufferedOutputStream(fos);
            bos.write(bfile);
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            if (bos != null) {
                try {
                    bos.close();
                } catch (IOException e1) {
                    e1.printStackTrace();
                }
            }
            if (fos != null) {
                try {
                    fos.close();
                } catch (IOException e1) {
                    e1.printStackTrace();
                }
            }
        }
    }
}