325 lines
9.1 KiB
Java
325 lines
9.1 KiB
Java
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());
|
||
}
|
||
}
|
||
|