package com.newtouch.util;
import java.io.BufferedReader;
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.List;
import java.util.Properties;
import java.util.Vector;
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.SftpATTRS;
import com.jcraft.jsch.SftpException;
/**
* sftp 基础类库
* 类描述: 通过配置代码连接sftp及提供相关工具函数
* 说明: 1:本工具不提供除配置连接方式之外的连接方式
* 2:构造函数必须传入参数获取配置信息
* 3:本工具只支持sftp配置连接方式,不支持ftp等其它方式.
*/
public class SftpTools {
private Session session;
private Channel channel;
private String host;
private String username;
private String password;
private int port;
/** 对外可访问 ChannelSftp对象提供的所有底层方法*/
public ChannelSftp chnSftp;
/**配置的远程目录地址*/
public String cfgRemotePath;
/**配置的远程目录历史地址 */
public String cfgRemotePathHis;
/**文件类型*/
public static final int FILE_TYPE = 1;
/**目录类型*/
public static final int DIR_TYPE = 2;
/**配置的本地地址*/
public String cfgLocalPath;
/**配置的本地历史地址*/
public String cfgLocalPathHis;
/**配置的本地临时地址*/
public String cfgLocalPathTemp;
/**
* 说明:构造函数必须传入配置中的ftpPathCode对应的值,注意检查正确性
* @param sftpPathCode
* @throws Exception
*/
public SftpTools(String sftpPathCode) throws Exception {
//可以根据 sftpPathCode得到数据库或者文件配置的主机端口用户密码等信息
// setHost("127.0.0.1");
// setPort(22);
// setUsername("newtouch");
// setPassword("newtouch");
// cfgRemotePath = "";
// cfgRemotePathHis ="";
// cfgLocalPath ="";
// cfgLocalPathHis = "";
// cfgLocalPathTemp ="";
//cfgRemotePath="/";// test
}
/**
* 通过配置 打开sftp连接资源
* @throws JSchException
* @throws SftpException
*/
public void open () throws JSchException,SftpException {
// this.connect(this.getHost(), this.getPort(), this.getUsername(), this.getPassword());
}
/**
* 连接SFTP
* @param host
* @param port
* @param username
* @param password
* @throws JSchException
* @throws SftpException
*/
public void connect (String host, int port, String username, String password) throws JSchException,SftpException {
JSch jsch = new JSch();
session = jsch.getSession(username, host, port);
//System.out.println("Session created.");
session.setPassword(password);
Properties sshConfig = new Properties();
sshConfig.put("StrictHostKeyChecking", "no");
session.setConfig(sshConfig);
session.connect();
//System.out.println("Session connected.");
channel = session.openChannel("sftp");
//System.out.println("Channel is Opened!");
channel.connect();
//System.out.println("Channel is connected!");
chnSftp = (ChannelSftp) channel;
//System.out.println("Connected to " + host + " is sucess!");
}
/**
* 进入指定的目录并设置为当前目录
* @param sftpPath
* @throws Exception
*/
public void cd (String sftpPath) throws SftpException {
chnSftp.cd(sftpPath);
}
/**
* 得到当前用户当前工作目录地址
* @return 返回当前工作目录地址
*
*/
public String pwd () {
return chnSftp.pwd();
}
/**
* 根据目录地址,文件类型返回文件或目录列表
* @param directory 如:/home/newtouch/kftesthis/201006/08/
* @param fileType 如:FILE_TYPE或者DIR_TYPE
* @return 文件或者目录列表 List
* @throws SftpException
* @throws Exception
*
*/
public List<String> listFiles (String directory, int fileType) throws SftpException {
List<String> fileList = new ArrayList<String>();
if (isDirExist(directory)) {
boolean itExist = false;
Vector vector;
vector = chnSftp.ls(directory);
for (int i = 0; i < vector.size(); i++) {
Object obj = vector.get(i);
String str = obj.toString().trim();
int tag = str.lastIndexOf(":") + 3;
String strName = str.substring(tag).trim();
itExist = isDirExist(directory + "/" + strName);
if (fileType == FILE_TYPE) {
if (!(itExist)) {
fileList.add(directory + "/" + strName);
}
}
if (fileType == DIR_TYPE) {
if (itExist) {
//目录列表中去掉目录名为.和..
if (!(strName.equals(".") || strName.equals(".."))) {
fileList.add(directory + "/" + strName);
}
}
}
}
}
return fileList;
}
/**
* 判断目录是否存在
* @param directory
* @return
* @throws SftpException
*/
public boolean isDirExist (String directory) throws SftpException {
boolean isDirExistFlag = false;
try {
SftpATTRS sftpATTRS = chnSftp.lstat(directory);
isDirExistFlag = true;
return sftpATTRS.isDir();
} catch (Exception e) {
if (e.getMessage().toLowerCase().equals("no such file")) {
isDirExistFlag = false;
}
}
return isDirExistFlag;
}
/**
* 下载文件后返回流文件
* @param sftpFilePath
* @return
* @throws SftpException
*/
public InputStream getFile (String sftpFilePath) throws SftpException {
if (isFileExist(sftpFilePath)) {
return chnSftp.get(sftpFilePath);
}
return null;
}
/**
* 获取远程文件流
* @param sftpFilePath
* @return
* @throws SftpException
*/
public InputStream getInputStreamFile (String sftpFilePath) throws SftpException {
return getFile(sftpFilePath);
}
/**
* 获取远程文件字节流
* @param sftpFilePath
* @return
* @throws SftpException
* @throws IOException
*/
public ByteArrayInputStream getByteArrayInputStreamFile (String sftpFilePath) throws SftpException,IOException {
if (isFileExist(sftpFilePath)) {
byte[] srcFtpFileByte = inputStreamToByte(getFile(sftpFilePath));
ByteArrayInputStream srcFtpFileStreams = new ByteArrayInputStream(srcFtpFileByte);
return srcFtpFileStreams;
}
return null;
}
/**
* 删除远程
* 说明:返回信息定义以:分隔第一个为代码,第二个为返回信息
* @param sftpFilePath
* @return
* @throws SftpException
*/
public String delFile (String sftpFilePath) throws SftpException {
String retInfo = "";
if (isFileExist(sftpFilePath)) {
chnSftp.rm(sftpFilePath);
retInfo = "1:File deleted.";
}
else {
retInfo = "2:Delete file error,file not exist.";
}
return retInfo;
}
/**
* 移动远程文件到目标目录
* @param srcSftpFilePath
* @param distSftpFilePath
* @return 返回移动成功或者失败代码和信息
* @throws SftpException
* @throws IOException
*/
public String moveFile (String srcSftpFilePath, String distSftpFilePath) throws SftpException,IOException {
String retInfo = "";
boolean dirExist = false;
boolean fileExist = false;
fileExist = isFileExist(srcSftpFilePath);
dirExist = isDirExist(distSftpFilePath);
if (!fileExist) {
//文件不存在直接反回.
return "0:file not exist !";
}
if (!(dirExist)) {
//1建立目录
createDir(distSftpFilePath);
//2设置dirExist为true
dirExist = true;
}
if (dirExist && fileExist) {
String fileName = srcSftpFilePath.substring(srcSftpFilePath.lastIndexOf("/"), srcSftpFilePath.length());
ByteArrayInputStream srcFtpFileStreams = getByteArrayInputStreamFile(srcSftpFilePath);
//二进制流写文件
this.chnSftp.put(srcFtpFileStreams, distSftpFilePath + fileName);
this.chnSftp.rm(srcSftpFilePath);
retInfo = "1:move success!";
}
return retInfo;
}
/**
* 复制远程文件到目标目录
* @param srcSftpFilePath
* @param distSftpFilePath
* @return
* @throws SftpException
* @throws IOException
*/
public String copyFile (String srcSftpFilePath, String distSftpFilePath) throws SftpException,IOException {
String retInfo = "";
boolean dirExi
- 1
- 2
- 3
前往页