package com.smartor.common;
|
|
import com.jcraft.jsch.ChannelSftp;
|
import com.jcraft.jsch.JSch;
|
import com.jcraft.jsch.Session;
|
import lombok.extern.slf4j.Slf4j;
|
import org.junit.Test;
|
import org.springframework.beans.factory.annotation.Value;
|
import org.springframework.stereotype.Component;
|
|
import java.io.File;
|
import java.io.FileOutputStream;
|
import java.util.Vector;
|
|
@Slf4j
|
@Component
|
public class FtpService {
|
|
@Value("${FTP_SERVER}")
|
private String FTP_SERVER;
|
|
@Value("${FTP_USERNAME}")
|
private String FTP_USERNAME;
|
|
@Value("${FTP_PASSWORD}")
|
private String FTP_PASSWORD;
|
|
|
public void downloadFolder(String remoteDir, String localDir) {
|
Session session = null;
|
ChannelSftp channelSftp = null;
|
|
try {
|
// 1. 创建 JSch 实例并设置会话
|
JSch jsch = new JSch();
|
session = jsch.getSession(FTP_USERNAME, FTP_SERVER, 8094);
|
session.setPassword(FTP_PASSWORD);
|
|
// 跳过主机密钥检查
|
session.setConfig("StrictHostKeyChecking", "no");
|
|
// 2. 连接到 SFTP 服务器
|
session.connect();
|
channelSftp = (ChannelSftp) session.openChannel("sftp");
|
channelSftp.connect();
|
|
// 3. 开始递归下载
|
downloadDirectory(channelSftp, remoteDir, localDir);
|
|
} catch (Exception e) {
|
e.printStackTrace();
|
} finally {
|
// 4. 关闭连接
|
if (channelSftp != null) {
|
channelSftp.disconnect();
|
}
|
if (session != null) {
|
session.disconnect();
|
}
|
}
|
}
|
|
/**
|
* 递归下载远程目录到本地
|
*
|
* @param channelSftp SFTP 通道
|
* @param remoteDir 远程目录路径
|
* @param localDir 本地目录路径
|
* @throws Exception 异常
|
*/
|
private static void downloadDirectory(ChannelSftp channelSftp, String remoteDir, String localDir) throws Exception {
|
// 确保本地目录存在
|
File localDirectory = new File(localDir);
|
if (!localDirectory.exists()) {
|
localDirectory.mkdirs();
|
}
|
|
// 列出远程目录中的所有文件和文件夹
|
Vector<ChannelSftp.LsEntry> entries = channelSftp.ls(remoteDir);
|
|
for (ChannelSftp.LsEntry entry : entries) {
|
String fileName = entry.getFilename();
|
|
// 跳过 "." 和 ".."
|
if (".".equals(fileName) || "..".equals(fileName)) {
|
continue;
|
}
|
|
String remoteFilePath = remoteDir + "/" + fileName;
|
String localFilePath = localDir + File.separator + fileName;
|
|
if (entry.getAttrs().isDir()) {
|
// 如果是文件夹，递归下载
|
downloadDirectory(channelSftp, remoteFilePath, localFilePath);
|
} else {
|
// 如果是文件，覆盖下载
|
File localFile = new File(localFilePath);
|
|
// 如果本地文件已存在，先删除
|
if (localFile.exists()) {
|
localFile.delete();
|
}
|
|
try (FileOutputStream fos = new FileOutputStream(localFile)) {
|
channelSftp.get(remoteFilePath, fos);
|
log.info("文件下载成功：{}", remoteFilePath);
|
} catch (Exception e) {
|
System.err.println("文件下载失败: " + remoteFilePath);
|
e.printStackTrace();
|
}
|
}
|
}
|
}
|
//
|
// @Test
|
// public void aa() {
|
// downloadFolder("/", "D:\\ruoyi\\uploadPath\\upload\\vadio\\voice");
|
// }
|
}
|
