mirror of https://github.com/elunez/eladmin
300 lines
10 KiB
Java
300 lines
10 KiB
Java
package me.zhengjie.utils.ftp;
|
|
|
|
import lombok.extern.slf4j.Slf4j;
|
|
import org.apache.commons.net.ftp.FTP;
|
|
import org.apache.commons.net.ftp.FTPClient;
|
|
import org.apache.commons.net.ftp.FTPFile;
|
|
|
|
import java.io.*;
|
|
import java.nio.charset.StandardCharsets;
|
|
|
|
/**
|
|
* @author pcshao.cn
|
|
* @date 20/01/2024
|
|
*/
|
|
@Slf4j
|
|
public class FtpUtil {
|
|
|
|
/**
|
|
* FTP文件上传
|
|
*
|
|
* @param ftpClient
|
|
* @param localFilePath
|
|
* @param remoteDirectory
|
|
* @return
|
|
*/
|
|
public static boolean uploadFile(FTPClient ftpClient, String localFilePath, String remoteDirectory) {
|
|
boolean done = false;
|
|
try {
|
|
// 检查远程目录是否存在
|
|
checkRemoteDir(ftpClient, remoteDirectory);
|
|
|
|
// 设置远程目录的字符编码为UTF-8
|
|
ftpClient.changeWorkingDirectory(new String(remoteDirectory.getBytes(StandardCharsets.UTF_8)));
|
|
|
|
File localFile = new File(localFilePath);
|
|
|
|
FileInputStream inputStream = new FileInputStream(localFile);
|
|
|
|
String remoteFilePath = remoteDirectory + localFile.getName();
|
|
|
|
done = ftpClient.storeFile(remoteFilePath, inputStream);
|
|
|
|
if (done) {
|
|
log.info("FtpUtil uploadFile successfully.");
|
|
} else {
|
|
log.error("FtpUtil uploadFile failed. reply: {} remotePath: {} obj: {}", ftpClient.getReplyString(), remoteFilePath, ftpClient);
|
|
}
|
|
inputStream.close();
|
|
} catch (IOException e) {
|
|
log.error(e.getMessage(), e);
|
|
}
|
|
return done;
|
|
}
|
|
|
|
/**
|
|
* FTP文件上传
|
|
*
|
|
* @param server
|
|
* @param port
|
|
* @param username
|
|
* @param password
|
|
* @param localFilePath
|
|
* @param remoteDirectory
|
|
*/
|
|
public static boolean uploadFile(String server, int port, String username, String password, String localFilePath, String remoteDirectory) {
|
|
boolean done = false;
|
|
FTPClient ftpClient = new FTPClient();
|
|
try {
|
|
// 设置控制连接的字符编码为UTF-8
|
|
ftpClient.setControlEncoding("UTF-8");
|
|
ftpClient.connect(server, port);
|
|
ftpClient.login(username, password);
|
|
ftpClient.enterLocalPassiveMode();
|
|
ftpClient.setFileType(FTP.BINARY_FILE_TYPE);
|
|
|
|
// 检查远程目录是否存在
|
|
checkRemoteDir(ftpClient, remoteDirectory);
|
|
|
|
// 设置远程目录的字符编码为UTF-8
|
|
ftpClient.changeWorkingDirectory(new String(remoteDirectory.getBytes(StandardCharsets.UTF_8)));
|
|
|
|
File localFile = new File(localFilePath);
|
|
|
|
FileInputStream inputStream = new FileInputStream(localFile);
|
|
|
|
String remoteFilePath = remoteDirectory + localFile.getName();
|
|
|
|
done = ftpClient.storeFile(remoteFilePath, inputStream);
|
|
|
|
if (done) {
|
|
log.info("FtpUtil uploadFile successfully.");
|
|
} else {
|
|
log.info("FtpUtil uploadFile failed. reply: {} remotePath: {}", ftpClient.getReplyString(), remoteFilePath);
|
|
}
|
|
inputStream.close();
|
|
} catch (IOException e) {
|
|
log.error(e.getMessage(), e);
|
|
} finally {
|
|
try {
|
|
if (ftpClient.isConnected()) {
|
|
ftpClient.logout();
|
|
ftpClient.disconnect();
|
|
}
|
|
} catch (IOException e) {
|
|
log.error(e.getMessage(), e);
|
|
}
|
|
}
|
|
return done;
|
|
}
|
|
|
|
/**
|
|
* 检查远程目录，如果不存在则创建
|
|
*
|
|
* @param ftpClient
|
|
* @param remoteDirectory
|
|
* @throws IOException
|
|
*/
|
|
private static void checkRemoteDir(FTPClient ftpClient, String remoteDirectory) throws IOException {
|
|
FTPFile[] ftpFiles = ftpClient.mlistDir(remoteDirectory);
|
|
log.info("FtpUtil mlistDir {} reply: {}", remoteDirectory, ftpClient.getReplyString());
|
|
if (ftpFiles.length <= 0) {
|
|
ftpClient.makeDirectory(remoteDirectory);
|
|
log.info("FtpUtil makeDirectory {} reply: {}", remoteDirectory, ftpClient.getReplyString());
|
|
}
|
|
}
|
|
|
|
/**
|
|
* FTP文件下载
|
|
*
|
|
* @param ftpClient
|
|
* @param remoteFilePath
|
|
* @param localDirectory
|
|
* @return
|
|
*/
|
|
public static File downloadFile(FTPClient ftpClient, String remoteFilePath, String localDirectory) {
|
|
File localFile = null;
|
|
try {
|
|
// 设置远程目录的字符编码为UTF-8
|
|
String remoteDirectory = remoteFilePath.substring(0, remoteFilePath.lastIndexOf("/") + 1);
|
|
ftpClient.changeWorkingDirectory(new String(remoteDirectory.getBytes(StandardCharsets.UTF_8)));
|
|
|
|
String remoteFileName = remoteFilePath.substring(remoteFilePath.lastIndexOf("/") + 1);
|
|
localFile = new File(localDirectory + remoteFileName);
|
|
|
|
OutputStream outputStream = new FileOutputStream(localFile);
|
|
|
|
boolean success = ftpClient.retrieveFile(remoteFilePath, outputStream);
|
|
|
|
if (success) {
|
|
log.info("FtpUtil downloaded successfully.");
|
|
} else {
|
|
log.info("FtpUtil download failed. reply: {} remotePath: {}", ftpClient.getReplyString(), remoteFilePath);
|
|
return null;
|
|
}
|
|
|
|
outputStream.close();
|
|
} catch (IOException e) {
|
|
log.error(e.getMessage(), e);
|
|
return null;
|
|
}
|
|
return localFile;
|
|
}
|
|
|
|
/**
|
|
* FTP文件下载
|
|
*
|
|
* @param server
|
|
* @param port
|
|
* @param username
|
|
* @param password
|
|
* @param remoteFilePath
|
|
* @param localDirectory
|
|
* @return
|
|
*/
|
|
public static File downloadFile(String server, int port, String username, String password, String remoteFilePath, String localDirectory) {
|
|
File localFile = null;
|
|
FTPClient ftpClient = new FTPClient();
|
|
try {
|
|
// 设置控制连接的字符编码为UTF-8
|
|
ftpClient.setControlEncoding("UTF-8");
|
|
ftpClient.connect(server, port);
|
|
ftpClient.login(username, password);
|
|
ftpClient.enterLocalPassiveMode();
|
|
ftpClient.setFileType(FTP.BINARY_FILE_TYPE);
|
|
|
|
// 设置远程目录的字符编码为UTF-8
|
|
String remoteDirectory = remoteFilePath.substring(0, remoteFilePath.lastIndexOf("/") + 1);
|
|
ftpClient.changeWorkingDirectory(new String(remoteDirectory.getBytes(StandardCharsets.UTF_8)));
|
|
|
|
String remoteFileName = remoteFilePath.substring(remoteFilePath.lastIndexOf("/") + 1);
|
|
localFile = new File(localDirectory + remoteFileName);
|
|
|
|
OutputStream outputStream = new FileOutputStream(localFile);
|
|
|
|
boolean success = ftpClient.retrieveFile(remoteFilePath, outputStream);
|
|
|
|
if (success) {
|
|
log.info("FtpUtil downloaded successfully.");
|
|
} else {
|
|
log.info("FtpUtil download failed. reply: {} remotePath: {}", ftpClient.getReplyString(), remoteFilePath);
|
|
return null;
|
|
}
|
|
|
|
outputStream.close();
|
|
} catch (IOException e) {
|
|
log.error(e.getMessage(), e);
|
|
return null;
|
|
} finally {
|
|
try {
|
|
if (ftpClient.isConnected()) {
|
|
ftpClient.logout();
|
|
ftpClient.disconnect();
|
|
}
|
|
} catch (IOException e) {
|
|
log.error(e.getMessage(), e);
|
|
}
|
|
}
|
|
return localFile;
|
|
}
|
|
|
|
/**
|
|
* FTP文件删除
|
|
*
|
|
* @param ftpClient
|
|
* @param remoteFilePath
|
|
* @return
|
|
*/
|
|
public static boolean deleteFile(FTPClient ftpClient, String remoteFilePath) {
|
|
boolean success = false;
|
|
try {
|
|
// 设置远程目录的字符编码为UTF-8
|
|
String remoteDirectory = remoteFilePath.substring(0, remoteFilePath.lastIndexOf("/") + 1);
|
|
ftpClient.changeWorkingDirectory(new String(remoteDirectory.getBytes(StandardCharsets.UTF_8)));
|
|
|
|
String remoteFileName = remoteFilePath.substring(remoteFilePath.lastIndexOf("/") + 1);
|
|
|
|
success = ftpClient.deleteFile(remoteFileName);
|
|
|
|
if (success) {
|
|
log.info("FtpUtil delete successfully.");
|
|
} else {
|
|
log.info("FtpUtil delete failed. reply: {} remotePath: {}", ftpClient.getReplyString(), remoteFilePath);
|
|
}
|
|
} catch (IOException e) {
|
|
log.error(e.getMessage(), e);
|
|
}
|
|
return success;
|
|
}
|
|
|
|
/**
|
|
* FTP文件删除
|
|
*
|
|
* @param server
|
|
* @param port
|
|
* @param username
|
|
* @param password
|
|
* @param remoteFilePath
|
|
* @return
|
|
*/
|
|
public static boolean deleteFile(String server, int port, String username, String password, String remoteFilePath) {
|
|
boolean success = false;
|
|
FTPClient ftpClient = new FTPClient();
|
|
try {
|
|
// 设置控制连接的字符编码为UTF-8
|
|
ftpClient.setControlEncoding("UTF-8");
|
|
ftpClient.connect(server, port);
|
|
ftpClient.login(username, password);
|
|
ftpClient.enterLocalPassiveMode();
|
|
ftpClient.setFileType(FTP.BINARY_FILE_TYPE);
|
|
|
|
// 设置远程目录的字符编码为UTF-8
|
|
String remoteDirectory = remoteFilePath.substring(0, remoteFilePath.lastIndexOf("/") + 1);
|
|
ftpClient.changeWorkingDirectory(new String(remoteDirectory.getBytes(StandardCharsets.UTF_8)));
|
|
|
|
String remoteFileName = remoteFilePath.substring(remoteFilePath.lastIndexOf("/") + 1);
|
|
|
|
success = ftpClient.deleteFile(remoteFileName);
|
|
|
|
if (success) {
|
|
log.info("FtpUtil delete successfully.");
|
|
} else {
|
|
log.info("FtpUtil delete failed. reply: {} remotePath: {}", ftpClient.getReplyString(), remoteFilePath);
|
|
}
|
|
|
|
} catch (IOException e) {
|
|
log.error(e.getMessage(), e);
|
|
} finally {
|
|
try {
|
|
if (ftpClient.isConnected()) {
|
|
ftpClient.logout();
|
|
ftpClient.disconnect();
|
|
}
|
|
} catch (IOException e) {
|
|
log.error(e.getMessage(), e);
|
|
}
|
|
}
|
|
return success;
|
|
}
|
|
}
|