package shkd.utils;
import com.jcraft.jsch.*;
import com.jcraft.jsch.ChannelSftp.LsEntry;
import kd.bos.logging.Log;
import kd.bos.logging.LogFactory;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.Properties;
import java.util.Vector;
/**
* <p>
* ClassName: SFTPConnectUtil
* </p>
* <p>
* Description: 连接SFTP
* </p>
*/
public class SFTPConnectUtil {
/**
* 日志
*/
private final static Log logger = LogFactory.getLog(SFTPConnectUtil.class);
/**
* 连接SFTP服务器
*
* @param host
* 主机
* @param port
* 端口
* @param username
* 用户名
* @param password
* 密码
* @return ChannelSftp
*/
public static ChannelSftp connectSFTP(String host, int port,
String username, String password) {
// 声明ChannelSftp对象
ChannelSftp sftp = null;
try {
// 创建JSch对象
JSch jsch = new JSch();
// 根据用户名，主机IP，端口获取一个Session对象
Session sshSession = jsch.getSession(username, host, port);
if (!isEmpty(password)) {
sshSession.setPassword(password);
}
// 新建属性类
Properties sshConfig = new Properties();
/*
* 连接新主机时，不进行公钥确认；
* 在首次连接服务器时，会弹出公钥确认的提示。这会导致某些自动化任务，由于初次连接服务器而导致自动化任务中断。
*/
sshConfig.put("StrictHostKeyChecking", "no");
// 配置连接设置
sshSession.setConfig(sshConfig);
// 建立连接
sshSession.connect();
// 打开SFTP通道
Channel channel = sshSession.openChannel("sftp");
// 建立SFTP通道的连接
channel.connect();
// 强转ChannelSftp对象
sftp = (ChannelSftp) channel;
} catch (JSchException e) {
logger.error("connectSFTP:" + e.toString());
return null;
}
return sftp;
}
/**
* 上传文件
*
* @param directory
* 上传的目录
* @param uploadFile
* 要上传的文件
* @param sftp
* @return boolean
*/
public static boolean uploadFile(String directory, File uploadFile,
ChannelSftp sftp,String fileName) {
try {
// 进入指定目录
sftp.cd(directory);
sftp.put(new FileInputStream(uploadFile), fileName);
} catch (FileNotFoundException e) {
logger.error("uploadFile -- FileNotFoundException：" + e.toString());
return false;
} catch (SftpException e) {
logger.error("uploadFile -- SftpException：" + e.toString());
return false;
}
return true;
}
/**
* 上传附件
* @param directory
* @param fileInputStream
* @param fileName
* @param sftp
* @return
*/
public static boolean uploadFile(String directory, FileInputStream fileInputStream,String fileName,
ChannelSftp sftp) {
try {
// 进入指定目录
sftp.cd(directory);
sftp.put(fileInputStream, fileName);
} catch (SftpException e) {
logger.error("uploadFile -- SftpException：" + e.toString());
return false;
}
return true;
}
/**
* 下载文件
*
* @param directory
* 下载目录
* @param downloadFile
* 下载的文件
* @param saveFile
* 存在本地的路径
* @param sftp
* @return boolean
*/
public static boolean downloadFile(String directory, String downloadFile,
File saveFile, ChannelSftp sftp) {
try {
// 进入指定目录
sftp.cd(directory);
// 下载文件到指定路径
sftp.get(downloadFile, new FileOutputStream(saveFile));
} catch (FileNotFoundException e) {
logger.error("downloadFile -- FileNotFoundException："
+ e.toString());
return false;
} catch (SftpException e) {
logger.error("downloadFile -- SftpException：" + e.toString());
return false;
}
return true;
}
/**
* 删除文件
*
* @param directory
* 要删除文件所在目录
* @param deleteFile
* 要删除的文件
* @param sftp
* @return boolean
*/
public static boolean deleteFile(String directory, String deleteFile,
ChannelSftp sftp) {
try {
// 进入指定目录
sftp.cd(directory);
// 删除文件
sftp.rm(deleteFile);
} catch (SftpException e) {
logger.error("deleteFile -- SftpException：" + e.toString());
return false;
}
return true;
}
/**
* 创建文件夹
*
* @param directory
* 目标目录
* @param creatDir
* 要创建的文件
* @param sftp
* @return boolean
*/
public static boolean createDir(String directory, String creatDir,
ChannelSftp sftp) {
try {
// 进入指定目录
sftp.cd(directory);
// 创建文件夹
sftp.mkdir(creatDir);
} catch (SftpException e) {
logger.error("deleteFile -- SftpException：" + e.toString());
return false;
}
return true;
}
/**
* 删除文件夹
*
* @param directory
* 目标目录
* @param deleteDir
* 要删除的文件
* @param sftp
* @return boolean
*/
public static boolean deleteDir(String directory, String deleteDir,
ChannelSftp sftp) {
try {
// 进入指定目录
sftp.cd(directory);
// 删除文件夹
sftp.rmdir(deleteDir);
} catch (SftpException e) {
logger.error("deleteFile -- SftpException：" + e.toString());
return false;
}
return true;
}
/**
* 列出目录下的文件
*
* @param directory
* 要列出的目录
* @param sftp
* @return Vector<LsEntry>
* @throws SftpException
*/
@SuppressWarnings("unchecked")
public static Vector<LsEntry> getListFiles(String directory,
ChannelSftp sftp) {
Vector<LsEntry> lsVec = null;
try {
// 列出指定目录下的所有文件和子目录
lsVec = sftp.ls(directory);
} catch (SftpException e) {
logger.error("getListFiles -- SftpException：" + e.toString());
return null;
}
return lsVec;
}
/**
* 删除上传与下载临时目录的文件
*
* @param file
* 要删除的文件
*/
public static void deleteFileByTemp(File file) {
// 目录是否存在
if (file.exists()) {
// 获取当前路径的文件列表
File[] listFiles = file.listFiles();
// 遍历文件列表
for (File files : listFiles) {
// 如果是文件夹，递归调用此方法
if (files.isDirectory()) {
deleteFileByTemp(files);
// 如果是文件，直接删除
} else if (files.isFile()) {
logger.info(files.getName() + "---" + files.delete());
}
}
// 删除根目录文件
logger.info(file.getName() + "---" + file.delete());
} else {
logger.info("当前路径不存在！");
}
}
/**
* 判断字符串是否为空
*
* @param str
* @return boolean
*/
public static boolean isEmpty(String str) {
if (null == str || "".equals(str)) {
return true;
}
return false;
}
/**
* 判断对象是否为空
*
* @param
* @return boolean
*/
public static boolean isEmpty(Object obj) {
if (null == obj || "".equals(obj)) {
return true;
}
return false;
}
/**
* 设置上传文件名
*
* @return String
*/
public static String setFileName() {
return new SimpleDateFormat("yyyyMMddHHmmss").format(new Date());
}
/**
* 设置上传文件夹名
*
* @return String
*/
public static String setDirName() {
return new SimpleDateFormat("yyyyMMdd").format(new Date());
}
}