package org.dromara.system.uitil;
|
|
import cn.hutool.extra.ftp.Ftp;
|
import org.apache.commons.lang3.StringUtils;
|
import org.apache.commons.net.ftp.FTP;
|
import org.apache.commons.net.ftp.FTPClient;
|
import org.apache.commons.net.ftp.FTPFile;
|
import org.apache.commons.net.ftp.FTPReply;
|
import org.dromara.system.domain.properties.FtpConfig;
|
import org.springframework.web.multipart.MultipartFile;
|
|
import java.io.*;
|
import java.net.UnknownHostException;
|
import java.util.ArrayList;
|
import java.util.List;
|
|
/**
|
* 实现FTP文件上传和文件下载
|
*/
|
public class FtpApche {
|
private static FTPClient ftpClient = new FTPClient();
|
private static String encoding = System.getProperty("file.encoding");
|
/**
|
* 超时时间
|
*/
|
private static final int DEFAULT_TIMEOUT = 60 * 1000;
|
|
/**
|
* 连接到ftp
|
*/
|
public static void connect(FtpConfig config) throws IOException {
|
setTimeout();
|
try {
|
ftpClient.connect(config.getUrl(), 21);
|
} catch (UnknownHostException e) {
|
throw new IOException("Can't find FTP server :" + "21");
|
}
|
|
int reply = ftpClient.getReplyCode();
|
if (!FTPReply.isPositiveCompletion(reply)) {
|
disconnect();
|
throw new IOException("Can't connect to server :" + "21");
|
}
|
|
if (!ftpClient.login(config.getUsername(), config.getPasswd())) {
|
disconnect();
|
throw new IOException("Can't login to server :" + "21");
|
}
|
|
// set data transfer mode.
|
ftpClient.setFileType(FTP.BINARY_FILE_TYPE);
|
|
// Use passive mode to pass firewalls.
|
ftpClient.enterLocalPassiveMode();
|
}
|
|
/**
|
* 设置超时时间
|
*/
|
private static void setTimeout() {
|
ftpClient.setDefaultTimeout(DEFAULT_TIMEOUT);
|
ftpClient.setConnectTimeout(DEFAULT_TIMEOUT);
|
ftpClient.setDataTimeout(DEFAULT_TIMEOUT);
|
}
|
|
/**
|
* ftp是否处于连接状态，是连接状态返回<tt>true</tt>
|
*
|
* @return boolean 是连接状态返回<tt>true</tt>
|
*/
|
public static boolean isConnected() {
|
return ftpClient.isConnected();
|
}
|
|
/**
|
* 关闭ftp连接
|
*/
|
public static void disconnect() {
|
if (null != ftpClient && ftpClient.isConnected()) {
|
try {
|
ftpClient.logout();
|
ftpClient.disconnect();
|
} catch (IOException ex) {
|
// do nothing
|
}
|
}
|
}
|
|
/**
|
* 改变工作目录
|
*
|
* @param dir ftp服务器上目录
|
* @return boolean 改变成功返回true
|
*/
|
public static boolean changeWorkingDirectory(String dir) {
|
try {
|
return ftpClient.changeWorkingDirectory(dir);
|
} catch (IOException e) {
|
e.printStackTrace();
|
}
|
return false;
|
}
|
|
/**
|
* Description: 向FTP服务器上传文件
|
*
|
* @return 成功返回true，否则返回false
|
*/
|
public static boolean uploadFile(FtpConfig config, MultipartFile file, String name) throws IOException {
|
boolean result = false;
|
InputStream input = file.getInputStream();
|
try {
|
connect(config);
|
ftpClient.setBufferSize(1024 * 1024);
|
result = ftpClient.storeFile(new String(name.getBytes(encoding), "iso-8859-1"), input);
|
if (result) {
|
System.out.println("上传成功!");
|
}
|
input.close();
|
} catch (IOException e) {
|
e.printStackTrace();
|
} finally {
|
ftpClient.logout();
|
ftpClient.disconnect();
|
}
|
return result;
|
}
|
|
/**
|
* 根据文件ftp路径名称删除文件
|
*
|
* @param ftpFileName 文件ftp路径名称
|
*/
|
public static void deleteFile(String ftpFileName) throws IOException {
|
int dele = ftpClient.dele(ftpFileName);
|
try {
|
ftpClient.logout();
|
ftpClient.disconnect();
|
} catch (IOException e) {
|
e.printStackTrace();
|
}
|
|
}
|
|
/**
|
* 根据uploadFile返回的路径，从ftp下载文件到指定输出流中
|
*
|
* @param filename 文件名称
|
*/
|
public static InputStream downloadFileFromDailyDir(String filename) throws IOException {
|
return ftpClient.retrieveFileStream(filename);
|
}
|
|
public static List<String> downloadList(FtpConfig config) throws IOException {
|
connect(config);
|
List<String> list = new ArrayList<>();
|
FTPFile[] ftpFiles = ftpClient.listFiles();
|
if (ftpFiles.length == 0) {
|
return list;
|
}
|
for (FTPFile ftpFile : ftpFiles) {
|
list.add(ftpFile.getName());
|
}
|
return list;
|
}
|
|
/**
|
* 根据名称获取文件，以字节数组返回
|
*
|
* @param ftpPath FTP服务器文件相对路径，例如：test/123
|
* @param fileName 文件名，例如：test.xls
|
* @return byte[] 字节数组对象
|
*/
|
public static byte[] getFileBytesByName(InputStream is) throws IOException {
|
//创建byte数组输出流
|
ByteArrayOutputStream byteStream = new ByteArrayOutputStream();
|
try {
|
try {
|
byte[] buffer = new byte[1024 * 1024 * 4];
|
int len = -1;
|
while ((len = is.read(buffer, 0, 1024 * 1024 * 4)) != -1) {
|
byteStream.write(buffer, 0, len);
|
}
|
} catch (Exception e) {
|
System.out.println(e);
|
}
|
} catch (Exception e) {
|
e.printStackTrace();
|
}
|
return byteStream.toByteArray();
|
}
|
|
}
|
