1、添加依赖 <dependency> <groupId>com.jcraft</groupId> <artifactId>jsch</artifactId> <version>0.1.53</version> </dependency> 2、工具类代码 import java.io.ByteArrayInputStream; 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.io.UnsupportedEncodingException; 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; /** * * @ClassName: SFTPUtil * @Description: sftp连接工具类 * @date 2017年5月22日 下午11:17:21 * @version 1.0.0 */ public class SFTPUtil { private transient Logger log = LoggerFactory.getLogger(this.getClass()); private ChannelSftp sftp; private Session session; /** FTP 登录用户名*/ private String username; /** FTP 登录密码*/ private String password; /** 私钥 */ private String privateKey; /** FTP 服务器地址IP地址*/ private String host; /** FTP 端口*/ private int port; /** * 构造基于密码认证的sftp对象 * @param username * @param password * @param host * @param port */ public SFTPUtil(String username, String password, String host, int port) { this.username = username; this.password = password; this.host = host; this.port = port; } /** * 构造基于秘钥认证的sftp对象 * @param username * @param host * @param port * @param privateKey */ 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服务器 * @throws Exception */ public void login(){ try { JSch jsch = new JSch(); if (privateKey != null) { jsch.addIdentity(privateKey);// 设置私钥 log.info("sftp connect,path of private key file:{}" , privateKey); } log.info("sftp connect by host:{} username:{}",host,username); session = jsch.getSession(username, host, port); log.info("Session is build"); if (password != null) { session.setPassword(password); } Properties config = new Properties(); config.put("StrictHostKeyChecking", "no"); session.setConfig(config); session.connect(); log.info("Session is connected"); Channel channel = session.openChannel("sftp"); channel.connect(); log.info("channel is connected"); sftp = (ChannelSftp) channel; log.info(String.format("sftp server host:[%s] port:[%s] is connect successfull", host, port)); } catch (JSchException e) { log.error("Cannot connect to specified sftp server : {}:{} \n Exception message is: {}", new Object[]{host, port, e.getMessage()}); } } /** * 关闭连接 server */ public void logout(){ if (sftp != null) { if (sftp.isConnected()) { sftp.disconnect(); log.info("sftp is closed already"); } } if (session != null) { if (session.isConnected()) { session.disconnect(); log.info("sshSession is closed already"); } } } /** * 将输入流的数据上传到sftp作为文件 * @param basePath sftp存储根路径:要求登录用户有权操作 * @param directory 自定义的文件目录 * @param uploadFile 上传的文件名 * @param input 输入流 * @throws SftpException */ public void upload(String basePath, String directory, String uploadFile, InputStream input) throws SftpException{ basePath += directory; try { sftp.cd(basePath); } catch (SftpException e) { log.warn("目录不存在,开始创建目录"); String[] split = basePath.split("/"); String path = ""; for (int i = 0; i < split.length; i++) { String dir = split[i]; if(dir==null||"".equals(dir.trim())){ continue; } path += "/" + dir; try { sftp.cd(path); } catch (SftpException se){ try { sftp.mkdir(path); log.error("创建目录【 " + path + "】"); } catch (SftpException se1) { log.error("创建目录【 " + path + "】失败"); log.error(se1.getMessage()); } } } } sftp.cd(basePath); sftp.put(input, uploadFile); log.info("文件上传成功:【{}】" , uploadFile); } /** * 上传单个文件 * @param basePath sftp存储根路径:要求登录用户有权操作 * @param directory 用户的存储目录 * @param uploadFile 要上传的本地文件,包括路径 * @throws FileNotFoundException * @throws SftpException * @throws Exception */ public void upload(String basePath, String directory, String uploadFile) throws FileNotFoundException, SftpException{ File file = new File(uploadFile); upload(basePath, directory, file.getName(), new FileInputStream(file)); } /** * 上传单个文件 * @param basePath sftp存储根路径:要求登录用户有权操作 * @param directory 用户的存储目录 * @param uploadFile 要上传的本地文件,包括路径 * @throws FileNotFoundException * @throws SftpException * @throws Exception */ public void upload(String basePath, String directory, File uploadFile) throws FileNotFoundException, SftpException{ upload(basePath, directory, uploadFile.getName(), new FileInputStream(uploadFile)); } /** * 将byte[]上传到sftp,作为文件。注意:从String生成byte[]是,要指定字符集。 * @param basePath sftp存储根路径:要求登录用户有权操作 * @param directory 用户的存储目录 * @param sftpFileName 文件在sftp端的命名 * @param byteArr 要上传的字节数组 * @throws SftpException * @throws Exception */ public void upload(String basePath, String directory, String sftpFileName, byte[] byteArr) throws SftpException{ upload(basePath, directory, sftpFileName, new ByteArrayInputStream(byteArr)); } /** * 将字符串按照指定的字符编码上传到sftp * @param basePath sftp存储根路径:要求登录用户有权操作 * @param directory 用户的存储目录 * @param sftpFileName 文件在sftp端的命名 * @param dataStr 待上传的数据 * @param charsetName sftp上的文件,按该字符编码保存 * @throws UnsupportedEncodingException * @throws SftpException * @throws Exception */ public void upload(String basePath, String directory, String sftpFileName, String dataStr, String charsetName) throws UnsupportedEncodingException, SftpException{ upload(basePath, directory, sftpFileName, new ByteArrayInputStream(dataStr.getBytes(charsetName))); } /** * 下载文件 * @param directory 下载目录 * @param downloadFile 下载的文件 * @param saveFile 存在本地的路径 * @throws SftpException * @throws FileNotFoundException * @throws Exception */ 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)); log.info("file:{} is download successful" , downloadFile); } /** * 下载文件 * @param directory 下载目录 * @param downloadFile 下载的文件名 * @return 字节数组 * @throws SftpException * @throws IOException * @throws Exception */ 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); log.info("file:{} is download successful" , downloadFile); return fileData; } /** * 删除文件 * * @param directory 要删除文件所在目录 * @param deleteFile 要删除的文件 * @throws SftpException * @throws Exception */ public void delete(String directory, String deleteFile) throws SftpException{ sftp.cd(directory); sftp.rm(deleteFile); } /** * 列出目录下的文件 * * @param directory 要列出的目录 * @return * @throws SftpException */ public Vector<?> listFiles(String directory) throws SftpException { return sftp.ls(directory); } // 测试 public static void main(String[] args) throws SftpException, IOException { SFTPUtil sftp = new SFTPUtil("root", "eFVw8Fby", "139.196.125.230", 22); sftp.login(); // //byte[] buff = sftp.download("/opt", "start.sh"); // //System.out.println(Arrays.toString(buff)); File file = new File("D:\\test.txt"); // InputStream is = new FileInputStream(file); // sftp.upload("/home/tomcat", "/test", "test.txt", is); sftp.upload("/home/tomcat", "/test", file); sftp.logout(); } }
sftp 文件上传
最新推荐文章于 2024-05-21 15:34:38 发布