Java工具jsch.jar实现上传下载

在做运维的时候,我们一般使用shell脚本实现文件的服务器之间定时传输,那么对于一些不会shell脚本的童鞋,就得使用万能的编程语言了,这里我们介绍一款Java操作ftp的工具,那就是jsch.jar。工具已经写好,可以根据实际情况做调整,注释很清晰。大家按需阅读:

package com.wdy.tools.utils.sftputil;
 
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.InputStream;
import java.io.OutputStream;
import java.util.Arrays;
import java.util.Collections;
import java.util.Properties;
import java.util.Vector;
 
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
 
import com.jcraft.jsch.Channel;
import com.jcraft.jsch.ChannelSftp;
import com.jcraft.jsch.JSch;
import com.jcraft.jsch.Session;
 
/**
 * 利用jsch.jar针对sftp的上传和下载
 * 需要jar包:
 * commons-lang3-3.2.1.jar
 * commons-logging.jar
 * jsch-0.1.53.jar
 * log4j-1.2.17.jar
 * @author wangdy
 *
 */
public class JschSftpUtils {
 
 private static final Log log = LogFactory.getLog(JschSftpUtils.class);
 
 /**
 * 利用JSch包实现SFTP下载文件
 * (这里是将serverPath下的指定扩展名的文件,全部下载到localPath下)
 * @param ip 主机IP
 * @param user 主机登陆用户名
 * @param psw 主机登陆密码
 * @param port 主机ssh2登陆端口,如果取默认值,传<=0的值,如-1
 * @param localPath 下载的本地目录的路径
 * @param serverPath 服务器目录
 * @param fileExetension 文件扩展名,格式如:'.txt'
 * @param isDelete 是否删除该路径下扩展名为fileExetension的远程文件
 * @throws Exception
 */
 public static void sshSftpDownload(String ip, String user, String psw, int port, String serverPath, String localPath, String fileExtension, boolean isDelete) throws Exception {
 Session session = null;
 Channel channel = null;
 JSch jsch = new JSch();
 if (port <= 0) {
 // 连接服务器,采用默认端口
 session = jsch.getSession(user, ip);
 } else {
 // 采用指定的端口连接服务器
 session = jsch.getSession(user, ip, port);
 }
 // 如果服务器连接不上,则抛出异常
 if (session == null) {
 throw new Exception("session is null");
 }
 // 设置登陆主机的密码
 session.setPassword(psw);// 设置密码
 // 设置第一次登陆的时候提示,可选值:(ask | yes | no)
// session.setConfig("StrictHostKeyChecking", "no");
 Properties sshConfig = new Properties();
 sshConfig.put("StrictHostKeyChecking", "no");
 session.setConfig(sshConfig);
 // 设置登陆超时时间ms
// session.connect(640);
 session.connect();
 log.info("Session connected.");
 log.info("Opening Channel.");
 try {
 // 创建sftp通信通道
 channel = (Channel) session.openChannel("sftp");
 channel.connect();
// channel.connect(1000);
 ChannelSftp sftp = (ChannelSftp) channel;
 log.info("Connected to " + ip + ".");
 // 进入服务器指定的文件夹
 sftp.cd(serverPath);
 /**
 * 列出服务器指定的文件列表(可以加参数,指明要下载的文件类型)
 * 说明:如果fileExtension不为空,则下载指定扩展名的文件集;
 * 如果fileExtension为"",则下载该目录下所有类型的文件,如果是文件夹的话,会报错,如果您路径下有以.连接的文件夹,请注意,这是不可以的,程序会在过滤到该文件夹时中断
 */
 Vector v = null;
 if (fileExtension != null && !"".equals(fileExtension)) {
 v = sftp.ls("*"+fileExtension);
 }else {
 try {
  v = sftp.ls("*.*");//ls -al | grep \"^-\"只显示文件---// ls -al | grep "^d"只显示目录包括.和..
 } catch (Exception e) {
  log.info("您操作的是一个文件夹!");
 }
 }
 
 for (int i = 0; i < v.size(); i++) {
// log.info("fileInfos: "+v.get(i));
 String[] fileInfos = v.get(i).toString().replaceAll("\t", " ").split(" ");
 String fileName = fileInfos[fileInfos.length-1];
 log.info("fileName is: "+fileName);
 // 以下代码实现从服务器下载一个文件到 本地
 InputStream instream = sftp.get(fileName);
 OutputStream outstream = new FileOutputStream(new File(localPath+"/"+fileName));
 byte b[] = new byte[1024];
 int n;
 while ((n = instream.read(b)) != -1) {
  outstream.write(b, 0, n);
 }
 outstream.flush();
 outstream.close();
 instream.close();
 log.info("文件["+fileName+"]下载成功!---->>>>下载到目录["+localPath+"]中.");
 //下载成功后,删除文件
 if (isDelete) {
  deleteOneFile(serverPath, fileName, sftp);
  log.info("文件["+fileName+"]删除成功!");
 }
 }
 } catch (Exception e) {
 e.printStackTrace();
 } finally {
 session.disconnect();
 channel.disconnect();
 }
 }
 
 /**
  * 利用JSch包实现SFTP上传文件
  * @param ip 主机IP
  * @param user 主机登陆用户名
  * @param psw 主机登陆密码
  * @param port 主机ssh2登陆端口,如果取默认值,传<=0的值,如-1
  * @param localPath 本地目录
  * @param serverPath 服务器目录
  * @param fileExtension 上传文件的扩展名 格式如:'.txt'
  */
 public static void sshSftpUpload(String ip, String user, String psw, int port,String localPath, String serverPath, String fileExtension) throws Exception {
 Session session = null;
 Channel channel = null;
 JSch jsch = new JSch();
 if (port <= 0) {
 // 连接服务器,采用默认端口
 session = jsch.getSession(user, ip);
 } else {
 // 采用指定的端口连接服务器
 session = jsch.getSession(user, ip, port);
 }
 // 如果服务器连接不上,则抛出异常
 if (session == null) {
 log.info("session is null,服务器连接失败");
 throw new Exception("session is null,服务器连接失败");
 }else {
 log.info("Connected Success, ip is ["+ip+"]");
 }
 // 设置登陆主机的密码
 session.setPassword(psw);// 设置密码
 // 设置第一次登陆的时候提示,可选值:(ask | yes | no)
 session.setConfig("StrictHostKeyChecking", "no");
 // 设置登陆超时时间ms
 session.connect(960);
 try {
 // 创建sftp通信通道
 channel = (Channel) session.openChannel("sftp");
 channel.connect(1000);
 ChannelSftp sftp = (ChannelSftp) channel;
 // 进入服务器指定的文件夹
 sftp.cd(serverPath);
 // 列出服务器指定的文件列表
// Vector v = sftp.ls("*.sh");
// for (int i = 0; i < v.size(); i++) {
// System.out.println(v.get(i));
// }
 // 以下代码实现从本地上传一个文件到服务器,如果要实现下载,对换一下流就可以了
 String[] files = getLocalFileNameArray(localPath);//获取所有文件名数组
 for (int i = 0; i < files.length; i++) {
 String fileName = files[i];
 if (fileExtension != null && !"".equals(fileExtension)) {//如果扩展名不为空,则上传该路径下指定扩展名的文件
  if (fileName.endsWith(fileExtension)) {
  OutputStream outstream = sftp.put(fileName);//要上传到服务器的文件,可以另外设个名字,也可以用本地的名称
  InputStream instream = new FileInputStream(new File(localPath+"/"+fileName));//读取本地文件
  byte b[] = new byte[1024];
  int n;
  while ((n = instream.read(b)) != -1) {
  outstream.write(b, 0, n);
  }
  outstream.flush();
  outstream.close();
  instream.close();
  log.info("文件["+localPath+"/"+fileName+"]上传成功!---->>>>上传到目录["+serverPath+"]中.");
  }else {
  log.info("警告:文件["+fileName+"]不是指定类型["+fileExtension+"]的文件");
  }
 }else {//如果扩展名为空,则上传该路径下的所有文件
  OutputStream outstream = sftp.put(fileName);//要上传到服务器的文件,可以另外设个名字,也可以用本地的名称
  InputStream instream = new FileInputStream(new File(localPath+"/"+fileName));//本地文件
  byte b[] = new byte[1024];
  int n;
  while ((n = instream.read(b)) != -1) {
  outstream.write(b, 0, n);
  }
  outstream.flush();
  outstream.close();
  instream.close();
  log.info("文件["+fileName+"]上传成功!---->>>>上传到目录["+serverPath+"]中.");
 }
 }
 } catch (Exception e) {
 e.printStackTrace();
 } finally {
 session.disconnect();
 channel.disconnect();
 }
 }
 
 
 /**
  * 利用JSch包实现SFTP下载、上传文件(该方法暂不适用)
  * @param ip 主机IP
  * @param user 主机登陆用户名
  * @param psw 主机登陆密码
  * @param port 主机ssh2登陆端口,如果取默认值(默认值22),传-1
  * @param privateKey 密钥文件路径
  * @param passphrase 密钥的密码
  * 
  */
 public static void sshSftp(String ip, String user, String psw ,int port ,String privateKey ,String passphrase) throws Exception{
  Session session = null;
  Channel channel = null;
  JSch jsch = new JSch();
  //设置密钥和密码
  if (privateKey != null && !"".equals(privateKey)) {
   if (passphrase != null && "".equals(passphrase)) {
    //设置带口令的密钥
    jsch.addIdentity(privateKey, passphrase);
   } else {
    //设置不带口令的密钥
    jsch.addIdentity(privateKey);
   }
  }
  if(port <=0){
   //连接服务器,采用默认端口
   session = jsch.getSession(user, ip);
  }else{
   //采用指定的端口连接服务器
   session = jsch.getSession(user, ip ,port);
  }
  //如果服务器连接不上,则抛出异常
  if (session == null) {
   throw new Exception("session is null");
  }
  //设置登陆主机的密码
  session.setPassword(psw);//设置密码 
  //设置第一次登陆的时候提示,可选值:(ask | yes | no)
  session.setConfig("StrictHostKeyChecking", "no");
  //设置登陆超时时间 
  session.connect(30000);
  try {
   //创建sftp通信通道
   channel = (Channel) session.openChannel("sftp");
   channel.connect(1000);
   ChannelSftp sftp = (ChannelSftp) channel;
   //进入服务器指定的文件夹
   sftp.cd("domains");
   //列出服务器指定的文件列表
   Vector v = sftp.ls("*.txt");
   for(int i=0;i v = null;
  if (fileExtension !=null && "".equals(fileExtension)) {
  v=sftp.ls("*"+fileExtension);
 }else {
// v=sftp.ls("");//此处有风险
 log.warn("FileExtension is not null! Please Check");
 }
  for (int i = 0; i < v.size(); i++) {
  String[] fileInfos = v.get(i).toString().replaceAll("\t", " ").split(" ");
 String fileName = fileInfos[fileInfos.length-1];
   sftp.rm(fileName);
  }
  } catch (Exception e) {
  e.printStackTrace();
  }
 }
 
 /** 
 * 删除单个文件 
 * 
 * @param directory 
 *   要删除文件所在目录 
 * @param deleteFile 
 *   要删除的文件 
 * @throws Exception  
 */ 
 public static void deleteOneFile(String directory, String deleteFile, ChannelSftp sftp) throws Exception { 
 sftp.cd(directory); 
 sftp.rm(deleteFile); 
 } 
 
 
}

这就是整个工具的内容了。

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持脚本之家。

你可能感兴趣的:(Java工具jsch.jar实现上传下载)