视频文件sftp服务下载之SFTPUtil

sftp服务器

即安全 FTP 服务器，帮助用户通过安全文件传输协议，如 SSH 文件传输协议 FTP 或使用 SSL/TLS 传输文件。该传输可以通过服务器到服务器或客户端 – 服务器配置来实现。安全的 FTP 服务器可帮助企业通过互联网或不安全的网络安全地发送机密文件。

SFTPUtil 实现代码

package glodon.govp.core.util;

import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.util.Properties;
import java.util.Vector;

import org.apache.commons.io.IOUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.jcraft.jsch.Channel;
import com.jcraft.jsch.ChannelSftp;
import com.jcraft.jsch.JSch;
import com.jcraft.jsch.JSchException;
import com.jcraft.jsch.Session;
import com.jcraft.jsch.SftpException;
/** 
* 类说明 sftp工具类
*/
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 {  
        	com.jcraft.jsch.Logger logger = new SettleLogger();
            JSch.setLogger(logger);
            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(2500);  
              
            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 in   输入流  
     */  
    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));  
    }  
    
    /**  
     * 下载文件 
     * @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 downloadFile 下载的文件名 
     * @return 字节流
     */  
    public InputStream downloadToInputStream(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 is;  
    }  
    
    
    /** 
     * 删除文件 
     * @param directory 要删除文件所在目录 
     * @param deleteFile 要删除的文件 
     */  
    public void delete(String directory, String deleteFile) throws SftpException{  
        sftp.cd(directory);  
        sftp.rm(deleteFile);  
    }  
    
    
    /** 
     * 列出目录下的文件 
     * @param directory 要列出的目录 
     * @param sftp 
     */  
    public Vector<?> listFiles(String directory) throws SftpException {  
        return sftp.ls(directory);  
    }  
      
    //上传文件测试
    public static void main(String[] args) throws SftpException, IOException {  
        SFTPUtil sftp = new SFTPUtil("用户名", "密码", "ip地址", 22);  
        sftp.login();  
        File file = new File("D:\\图片\\t0124dd095ceb042322.jpg");  
        InputStream is = new FileInputStream(file);  
          
        sftp.upload("基础路径","文件路径", "test_sftp.jpg", is);  
        sftp.logout();  
    }  
}

SFTPUtil的使用场景

主要是应用在，用户从文件服务器上下载文件视频（加密的）

SFTPUtil使用步骤

第一步构建SFTPUtil对象

SFTPUtil sftp = new SFTPUtil(video.getFileServerUserName(), video.getFileServerUserPass(), 
				video.getFileServerIp(), Integer.valueOf(video.getFileServerPort()));

第二步登录sftp服务器
sftp.login();
第三步，向sftp服务器获取数据流

 InputStream inputStream = sftp.downloadToInputStream(video.getDownUrl(), video.getDownUrl()+video.getFilename());

第四步，浏览器下载文件

 if(inputStream != null){
        	response.reset();
            response.setContentType("application/x-download;charset=UTF-8");
            response.setHeader("Content-Disposition", "attachment;filename=\"" +URLEncoder.encode(video.getFilename(),"UTF-8") + "\"");
            response.setCharacterEncoding("UTF-8");
            //创建缓冲区
            byte[] b= new byte[2048];  
            int len = 0; 
           OutputStream stream = response.getOutputStream();
           //循环将输入流中的内容读取到缓冲区当中  
           while((len = inputStream.read(b)) != -1){
           //输出缓冲区的内容到浏览器，实现文件下载
                stream.write(b, 0, len);
            }
            //退出sftp服务器
           sftp.logout();
            stream.flush();
            stream.close();
            inputStream.close();
            inputStream =null;
}

maven依赖jar

<dependency>
		    <groupId>org.apache.poi</groupId>
		    <artifactId>poi</artifactId>
		    <version>3.9</version>
		</dependency>
		<dependency>
		    <groupId>org.apache.poi</groupId>
		    <artifactId>poi-ooxml</artifactId>
		    <version>3.9</version>
		</dependency>
		<dependency>
		    <groupId>com.jcraft</groupId>
		    <artifactId>jsch</artifactId>
		    <version>0.1.54</version>
		</dependency>
		<dependency>
	       <groupId>net.lingala.zip4j</groupId>
	       <artifactId>zip4j</artifactId>
	       <version>1.3.2</version>
		</dependency>