import java.io.File;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;
import java.util.Properties;
import java.util.Vector;
import org.apache.commons.lang.StringUtils;
import org.apache.log4j.Logger
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;
public class SFTPAccesser {
private static Logger log = Logger.getRootLogger();
/**
* connect to SFTP Server
* If can't connect to SFTP , it will re-try
* @param para
* @param count connect retry count
* @return isConnected
* @throws InterruptedException
* @throws JSchException
*/
public static Boolean connectSFTP(SFTPParameter para, Integer count) throws InterruptedException, JSchException {
Boolean ret = false;
if (para == null)
return ret;
if (count == 1) {
int interval = para.checkInterval();
Thread.sleep(interval);
}
if (count == 0)
count = 1;
if (para.isConnected)
return true;
if (StringUtils.isEmpty(para.hostName) || para.port == 0
|| para.userName == null || para.passWord == null) {
log.error("SFTP connect parameter error !");
para.release() ;
return ret;
}
ChannelSftp csftp = null;
Session session = null;
Channel channel = null;
JSch jsch = new JSch();
if(StringUtils.isNotBlank(para.sftpKeyFile)){
if(StringUtils.isBlank(para.passWord)){
jsch.addIdentity(para.sftpKeyFile);
}else{
jsch.addIdentity(para.sftpKeyFile, para.passWord);
}
}
Properties config = new java.util.Properties();
config.put("StrictHostKeyChecking", "no");
session = jsch.getSession(para.userName, para.hostName, para.port);
session.setConfig(config);
if(StringUtils.isBlank(para.sftpKeyFile)){
session.setPassword(para.passWord);
}
int i = 0;
while (true) {
i++;
try {
session.connect();
channel = session.openChannel("sftp");
channel.connect();
csftp = (ChannelSftp) channel;
ret = true;
para.isConnected = true;
para.sftp = csftp ;
log.info("Successfully connect to " + para.hostName + ":" + para.port + " with " + para.userName);
break;
} catch (Exception ex) {
log.error(ex);
para.isConnected = false;
if (i >= count) {
log.info(para.toString());
log.error(
"Can't connect SFTP Server, have already tried" + i
+ "times!");
try {
int interval = para.checkInterval();
Thread.sleep(interval);
} catch (Exception e) {
log.error(ex);
}
break;
} else {
try {
int interval = para.checkInterval();
log.error(
"Can't connect SFTP Server，waiting for "
+ interval + "seconds try again!");
Thread.sleep(interval);
} catch (InterruptedException e) {
log.error(e);
break;
}
}
}
}
return ret;
}
/**
* get SFTP Server file list .
*@param para
*@return List
* @throws SftpException
* @throws JSchException
* @throws InterruptedException
*/
public static List getFiles(SFTPParameter para) throws InterruptedException, JSchException, SftpException {
List files = new ArrayList();
if (para != null) {
if (connectSFTP(para, para.conFTPTryCount)) {
ChannelSftp cs = para.sftp;
Vector fs = null;
fs = cs.ls(para.downloadPath);
for (ChannelSftp.LsEntry entry : fs) {
if (!entry.getAttrs().isDir()) {
files.add(entry);
}
}
}
}
return files;
}
/**
* download files from SFTP
* @param para
* @return
* @throws InterruptedException
* @throws JSchException
* @throws SftpException
* @throws IOException
*/
public static Boolean downloadFiles(SFTPParameter para) throws InterruptedException, JSchException, SftpException{
Boolean ret = true ;
List files = getFiles(para) ;
if(!para.isConnected) connectSFTP(para, para.conFTPTryCount) ;
for(ChannelSftp.LsEntry file : files){
ChannelSftp cs = para.sftp;
if(!para.downloadPath.endsWith("/")) para.downloadPath = para.downloadPath + "/" ;
if(!para.savePath.endsWith("/")) para.savePath = para.savePath + "/" ;
String fromFile = para.downloadPath + file.getFilename() ;
String toFile = para.savePath + file.getFilename() ;
cs.get(fromFile, toFile);
}
return ret ;
}
/**
* upload file to SFTP
* @param para
* @param file
* @return
* @throws InterruptedException
* @throws JSchException
* @throws SftpException
*/
public static Boolean uploadFile(SFTPParameter para,File file) throws InterruptedException, JSchException, SftpException{
Boolean ret = false ;
if(!file.exists()) return ret ;
if (connectSFTP(para, para.conFTPTryCount)) {
ChannelSftp cs = para.sftp ;
if(!para.uploadPath.endsWith("/")) para.uploadPath = para.uploadPath + "/" ;
cs.put(file.getAbsolutePath(), para.uploadPath + file.getName());
ret = true ;
}
return ret ;
}
/**
* upload files to SFTP server .
*@param para
*@param file
*@return
* @throws SftpException
* @throws JSchException
* @throws InterruptedException
*/
public static Boolean uploadFiles(SFTPParameter para,List files) throws InterruptedException, JSchException, SftpException{
Boolean ret = true ;
if(files == null || files.size() <= 0) return ret ;
int i = 0 ;
for(File file : files){
if(!uploadFile(para, file)) i ++ ;
}
if(i > 0){
log.error(i + " items were uploaded failed!");
ret = false ;
}
return ret ;
}
/**
* delete FTP server file .
* @param para
* @param filepath file path must be /home/user/text.xxx
* @return true or false
* @throws SftpException
* @throws JSchException
* @throws InterruptedException
*/
public static Boolean deleteFile(SFTPParameter para,String filepath) throws InterruptedException, JSchException, SftpException{
Boolean ret = false ;
if(connectSFTP(para, para.conFTPTryCount)){
para.sftp.rm(filepath);
ret = true ;
}
return ret ;
}
/**
* delete SFTP files .
* @param para
* @return true or false
* @throws SftpException
* @throws JSchException
* @throws InterruptedException
*/
public static Boolean deleteFiles(SFTPParameter para,List list) throws InterruptedException, JSchException, SftpException{
Boolean ret = true ;
if(list == null || list.size() <= 0) ret = false;
int i = 0 ;
for(String s : list ){
if(!deleteFile(para, s)) i ++ ;
}
if(i > 0){
log.equals(i + " items were deleted failed!");
ret = false ;
}
return ret ;
}
/**
* is file exist in SFTP
* @param para
* @param path
* @param file
* @return
* @throws InterruptedException
* @throws JSchException
* @throws SftpException
*/
public static boolean isFileExist(SFTPParameter para,String path, String file) throws InterruptedException, JSchException, SftpException
{
if(connectSFTP(para, para.conFTPTryCount)){
Vector files = null;
ChannelSftp cs = para.sftp ;
files = cs.ls(path);
for (ChannelSftp.LsEntry entry : files) {
if (!entry.getAttrs().isDir()
&& entry.getFilename().equals(file)) {
return true;
}
}
}
return false;
}
/**
* is file exist in SFTP
* @param para
* @param path
* @param file
* @return
* @throws InterruptedException
* @throws JSchException
* @throws SftpException
*/
public static boolean isDirExist(SFTPParameter para,String path, String dirname) throws InterruptedException, JSchException, SftpException
{
if(connectSFTP(para, para.conFTPTryCount)){
Vector files = null;
ChannelSftp cs = para.sftp ;
files = cs.ls(path);
for (ChannelSftp.LsEntry entry : files) {
if (entry.getAttrs().isDir()
&& entry.getFilename().equals(dirname)) {
return true;
}
}
}
return false;
}
/**
* make dir
* @param para
* @param makeDir like /home/ccc/bbb
* @throws InterruptedException
* @throws JSchException
* @throws SftpException
*/
public static void makeDir(SFTPParameter para,String makeDir) throws InterruptedException, JSchException, SftpException {
if("".equals(makeDir) || makeDir == null) return;
connectSFTP(para, para.conFTPTryCount) ;
ChannelSftp cs = para.sftp ;
String[] makedirs = makeDir.split("\\/") ;
String ls = "/" ;
for(String dir : makedirs){
if( !"".equals(dir) && !isDirExist(para, ls, dir) ){
ls += dir +"/";
cs.mkdir(dir);
cs.cd(dir);
}else if(!"".equals(dir)){
ls += dir +"/";
cs.cd(ls);
}
}
}
public static void main(String[] args) {
SFTPParameter para = new SFTPParameter() ;
para.hostName ="10.37.240.35" ;
para.userName = "cfiusr" ;
para.passWord = "password" ;
para.port = 22 ;
para.downloadPath = "/home/cfiusr/test" ;
para.savePath ="D:/test";
para.uploadPath ="/home/cfiusr/test";
File file = new File("D:/appl/cmd.txt") ;
String delFile = "/home/cfiusr/test/cmd.txt";
try {
String makeDir = "/home/cfiusr/D/E/F/G/H/J/kk";
makeDir(para,makeDir) ;
para.release();
System.exit(1);
} catch (Exception e) {
e.printStackTrace();
System.exit(1);
}
}
}
import com.jcraft.jsch.ChannelSftp;
import com.jcraft.jsch.JSchException;
import com.jcraft.jsch.Session;
public class SFTPParameter {
public String hostName;
public int port = 0;
public String userName;
public String passWord;
public String sftpKeyFile ;
public String downloadPath;
public String savePath;
public String uploadPath;
public static int CON_FTP_TRYCOUNT_DEFAULT = 3;
public static int CON_FTP_TRYINTERVAL_DEFAULT = 1;
public int conFTPTryCount = 0;
public int conFTPTryInterval = 1;
public ChannelSftp sftp = null ;
public Boolean isConnected = false;
public int checkInterval() {
if (0 == this.conFTPTryInterval)
return CON_FTP_TRYINTERVAL_DEFAULT * 1000;
else
return this.conFTPTryInterval * 1000;
}
public int checkCount() {
if (this.conFTPTryCount <= 0)
return CON_FTP_TRYCOUNT_DEFAULT;
else
return this.conFTPTryCount;
}
public void release() throws JSchException {
if (sftp != null) {
Session session = sftp.getSession();
sftp.disconnect();
if (session != null) {
session.disconnect();
}
}
isConnected = false;
}
}
