SFTP工具类

悠悠 2022-05-20 04:40 186阅读 0赞

**丛菊两开他日泪，孤舟一系故园心**  
项目中，经常要用到上传文件和下载文件，这里，我们通过一个sftp连接到linux服务器。工具类如下：

import java.io.*; import java.util.Properties; import cmcc.gz.channel2.webServiceTo4A.controller.DataSyncImpController; import com.jcraft.jsch.*; import org.apache.log4j.Logger; import javax.transaction.SystemException; /** * *SFTP工具类 */ public class SFTPUtils { private final static Logger log = Logger .getLogger(SFTPUtils.class); public ChannelSftp channelSftp; public Session session; //配置文件名称，该配置文件里面设置服务器的ip地址，用户名密码等 public String attchmentProperties = "address.properties"; /** * 获取sftp连接 * @param ftpaddress * @param ftpPassword * @param ftpUserName */ public ChannelSftp getSFTPClient(String ftpaddress, String ftpPassword, String ftpUserName,String ftpPort) { //开始时间 用于计时 long startTime = System.currentTimeMillis(); JSch jsch = new JSch();// 创建JSch对象 Channel channel = null; try { session = jsch.getSession(ftpUserName, ftpaddress,Integer.valueOf(ftpPort)); // 根据用户名，主机ip，端口获取一个Session对象 session.setPassword(ftpPassword); // 设置密码 Properties config = new Properties(); config.put("StrictHostKeyChecking", "no"); session.setConfig(config); // 为Session对象设置properties // session.setTimeout(timeout); // 设置timeout时间 session.connect(); // 通过Session建立链接 channel = session.openChannel("sftp"); // 打开SFTP通道 channel.connect(); // 建立SFTP通道的连接 long endTime = System.currentTimeMillis(); log.debug("连接sftp耗时" + (endTime - startTime) + "毫秒"); System.out.println("连接sftp耗时" + (endTime - startTime) + "毫秒"); return (ChannelSftp) channel; } catch (Exception e) { e.printStackTrace(); } return null; } /** * 上传文件 */ public boolean uploadFile(InputStream in,String fileName,String moduleName){ try{ //这里是从配置文件读取IP地址和用户名密码、路径，你也可以直接写在代码上 String ftpaddress = PropertiesUtils.getSchema(attchmentProperties,"ip"); String ftpUserName = PropertiesUtils.getSchema(attchmentProperties,"username"); String ftpPassword = PropertiesUtils.getSchema(attchmentProperties,"password"); String ftpPort = PropertiesUtils.getSchema(attchmentProperties,"port"); String filePath = PropertiesUtils.getSchema(attchmentProperties,"path"); //建立连接 ChannelSftp channelSftp=getSFTPClient(ftpaddress,ftpPassword,ftpUserName,ftpPort); //判断文件夹是否存在，不存在则创建 if (isDirExist(filePath + "/" + moduleName,channelSftp)) { channelSftp.put(in, filePath + "/" + moduleName + "/"+fileName); return true; } else { //创建文件夹在上传文件 createDir(filePath + "/" + moduleName,channelSftp); channelSftp.put(in, filePath + "/" + moduleName + "/"+fileName); return true; } }catch(Exception e){ e.printStackTrace(); closeSftp(); } return false; } /** * 读取sftp文件 */ public InputStream readFile(String moudleName,String fileName){ //这里是从配置文件读取IP地址和用户名密码、路径，你也可以直接写在代码上 String ftpaddress = PropertiesUtils.getSchema(attchmentProperties,"ip"); String ftpUserName = PropertiesUtils.getSchema(attchmentProperties,"username"); String ftpPassword = PropertiesUtils.getSchema(attchmentProperties,"password"); String ftpPort = PropertiesUtils.getSchema(attchmentProperties,"port"); String filePath = PropertiesUtils.getSchema(attchmentProperties,"path"); //建立连接 ChannelSftp channelSftp=getSFTPClient(ftpaddress,ftpPassword,ftpUserName,ftpPort); try { return channelSftp.get(filePath+"/" + moudleName + "/"+fileName); } catch (SftpException e) { System.out.println("获取 ："+filePath+"/"+fileName+" 文件流错误"); closeSftp(); } return null; } /** * * 关闭sftp */ public void closeSftp() { if (channelSftp != null) { channelSftp.disconnect(); } if (session != null) { session.disconnect(); } } //测试 public static void main(String[] args) { SFTPUtils sftpUtils = new SFTPUtils(); try { //上传测试 /*FileInputStream file = new FileInputStream("D:\\abc文档 - 副本.txt"); // FileInputStream file = new FileInputStream("D:\\abc文档.txt"); sftpUtils.uploadFile(file,"abc文档1.txt","ABC"); sftpUtils.closeSftp();*/ //下载测试 InputStream inputStream = sftpUtils.readFile("ABC","abc文档.txt"); FileOutputStream fos = new FileOutputStream("D:\\aaa测试qqq.txt"); byte[] buff = new byte[1024]; int length = 0; while((length = inputStream.read(buff))!=-1){ fos.write(buff, 0, length); } inputStream.close(); fos.close(); sftpUtils.closeSftp(); } catch (Exception e) { e.printStackTrace(); } } /** * 创建一个文件目录 */ public void createDir(String createpath, ChannelSftp sftp) { try { if (isDirExist(createpath,sftp)) { sftp.cd(createpath); return; } String pathArry[] = createpath.split("/"); StringBuffer filePath = new StringBuffer("/"); for (String path : pathArry) { if (path.equals("")) { continue; } filePath.append(path + "/"); if (isDirExist(filePath.toString(),sftp)) { sftp.cd(filePath.toString()); } else { // 建立目录 sftp.mkdir(filePath.toString()); // 进入并设置为当前目录 sftp.cd(filePath.toString()); } } sftp.cd(createpath); } catch (SftpException e) { // throw new SystemException("创建路径错误：" + createpath); } } //判断目录是否存在 public boolean isDirExist(String directory, ChannelSftp sftp) { boolean isDirExistFlag = false; try { SftpATTRS sftpATTRS = sftp.lstat(directory); isDirExistFlag = true; return sftpATTRS.isDir(); } catch (Exception e) { if (e.getMessage().toLowerCase().equals("no such file")) { isDirExistFlag = false; } } return isDirExistFlag; } }