package com.homolo.justice.lcsms.www.util;
import com.homolo.toolkit.filesystem.Storage;
import com.homolo.toolkit.filesystem.exception.FileExistsException;
import com.homolo.toolkit.filesystem.exception.FileNotFoundException;
import org.apache.commons.io.IOUtils;
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.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
/**
* FTP操作工具类.
*
*/
@Component
public class FtpUtil {
private static final Logger LOGGER = LoggerFactory.getLogger(FtpUtil.class);
@Autowired
private Storage storage;
/**
* Description: 向FTP服务器上传文件.
*
* @param host FTP服务器hostname
* @param port FTP服务器端口
* @param username FTP登录账号
* @param password FTP登录密码
* @param basePath FTP服务器基础目录
* @param filePath FTP服务器文件存放路径。例如分日期存放:/2015/01/01。文件的路径为basePath+filePath
* @param filename 上传到FTP服务器上的文件名
* @param input 输入流
* @return 成功返回true,否则返回false
*/
public boolean uploadFile(String host, int port, String username, String password, String basePath,
String filePath, String filename, InputStream input) {
boolean result = false;
FTPClient ftp = new FTPClient();
try {
int reply;
// 连接FTP服务器
ftp.connect(host, port);
// 如果采用默认端口,可以使用ftp.connect(host)的方式直接连接FTP服务器
// 登录
ftp.login(username, password);
reply = ftp.getReplyCode();
// 判断返回码是否合法
if (!FTPReply.isPositiveCompletion(reply)) {
// 不合法时断开
ftp.disconnect();
LOGGER.info("连接不合法,连接码:{}", reply);
return result;
}
//切换到上传目录
if (!ftp.changeWorkingDirectory(basePath + filePath)) {
if (StringUtils.isNotEmpty(basePath)) {
//如果目录不存在创建目录
String[] dirs = filePath.split("/");
StringBuilder tempPath = new StringBuilder();
tempPath.append(basePath);
for (String dir : dirs) {
if (null == dir || "".equals(dir)) {
continue;
}
tempPath.append("/").append(dir);
if (!ftp.changeWorkingDirectory(tempPath.toString())) {
if (!ftp.makeDirectory(tempPath.toString())) {
return result;
} else {
ftp.changeWorkingDirectory(tempPath.toString());
}
}
}
}
}
// 进入当前文件目录(如果未添加这个,可能找不到对应目录)
ftp.enterLocalPassiveMode();
ftp.setRemoteVerificationEnabled(false);
//设置上传文件的类型为二进制类型
ftp.setFileType(FTP.BINARY_FILE_TYPE);
//上传文件
if (!ftp.storeFile(filename, input)) {
LOGGER.info("上传文件失败,文件名:{}", filename);
return result;
}
input.close();
ftp.logout();
result = true;
} catch (IOException e) {
LOGGER.info("推送文件异常", e);
} finally {
if (ftp.isConnected()) {
try {
ftp.disconnect();
} catch (IOException ioe) {
LOGGER.info("连接关闭异常");
}
}
}
return result;
}
/**
* Description: 从FTP服务器下载文件.
*
* @param host FTP服务器hostname
* @param port FTP服务器端口
* @param username FTP登录账号
* @param password FTP登录密码
* @param remotePath FTP服务器上的相对路径
* @param fileName 要下载的文件名
* @param newFileName 下载后保存到服务器的新文件名
* @return .
*/
public boolean downloadFile(String host, int port, String username, String password, String remotePath,
String fileName, String newFileName) {
boolean result = false;
FTPClient ftp = new FTPClient();
try {
int reply;
ftp.connect(host, port);
// 如果采用默认端口,可以使用ftp.connect(host)的方式直接连接FTP服务器
// 登录
ftp.login(username, password);
reply = ftp.getReplyCode();
if (!FTPReply.isPositiveCompletion(reply)) {
ftp.disconnect();
return result;
}
// 转移到FTP服务器目录
ftp.changeWorkingDirectory(remotePath);
FTPFile[] fs = ftp.listFiles();
for (FTPFile ff : fs) {
if (ff.getName().equals(fileName)) {
try (InputStream inputStream = ftp.retrieveFileStream(ff.getName())) {
com.homolo.toolkit.filesystem.File file = this.storage.createFile(fileName.substring(0, (fileName.indexOf(".") + 1)), newFileName);
try (OutputStream storageOut = file.openOutputStream()) {
IOUtils.copy(inputStream, storageOut);
}
file.save();
} catch (FileNotFoundException e) {
LOGGER.info("文件不存在:", e);
result = false;
} catch (FileExistsException e) {
LOGGER.info("文件已存在:", e);
result = false;
} catch (IOException e) {
LOGGER.info("文件获取异常", e);
result = false;
}
}
}
ftp.logout();
result = true;
} catch (IOException e) {
LOGGER.info("ftp操作异常", e);
} finally {
if (ftp.isConnected()) {
try {
ftp.disconnect();
} catch (IOException ioe) {
LOGGER.info("连接关闭异常");
}
}
}
return result;
}
}
注:storage是我们公司自己封装的文件工具bean,你们可以自己操作