package com.chinaztt.mes.common.ftp;
|
|
import com.alibaba.fastjson.JSONObject;
|
import lombok.Data;
|
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 org.springframework.cloud.context.config.annotation.RefreshScope;
|
import org.springframework.stereotype.Component;
|
|
import javax.servlet.http.HttpServletResponse;
|
import java.io.*;
|
import java.util.ArrayList;
|
import java.util.List;
|
|
/**
|
* @Author: zhangxy
|
* @Date: 2019/12/10 8:45
|
*/
|
@RefreshScope
|
@Component
|
@Data
|
@Slf4j
|
public class FtpUtils {
|
private static String GSM_CONTRACT_DIR = "/contract";
|
private static String GSM_ORDER_DIR = "/Order";
|
private static final String UNDERLINE = "/";
|
|
|
/**
|
* 通过流下载
|
* @param fileName
|
* @param contractNumber
|
* @param response
|
*/
|
public static void downloadForWeb(String fileName, String contractNumber, HttpServletResponse response) {
|
String remoteFilePath = getRemote(contractNumber) + contractNumber.replaceAll(UNDERLINE, "") + UNDERLINE + encodeFileName(fileName);
|
FTPClient ftpClient = Global.pool.getResource();
|
InputStream is = null;
|
BufferedInputStream bis = null;
|
try {
|
is = ftpClient.retrieveFileStream(remoteFilePath);
|
System.out.println(is);
|
bis = new BufferedInputStream(is);
|
OutputStream out = response.getOutputStream();
|
byte[] buf = new byte[1024];
|
int len = 0;
|
while ((len = bis.read(buf)) > 0) {
|
out.write(buf, 0, len);
|
}
|
out.flush();
|
} catch (Exception e) {
|
e.printStackTrace();
|
}
|
}
|
|
public static boolean downloadSingleFile(FTPClient ftpClient,
|
String remoteFilePath, String savePath) throws IOException {
|
File downloadFile = new File(savePath);
|
|
File parentDir = downloadFile.getParentFile();
|
if (!parentDir.exists()) {
|
parentDir.mkdir();
|
}
|
|
OutputStream outputStream = new BufferedOutputStream(
|
new FileOutputStream(downloadFile));
|
try {
|
return ftpClient.retrieveFile(remoteFilePath, outputStream);
|
} catch (IOException ex) {
|
throw ex;
|
} finally {
|
if (outputStream != null) {
|
outputStream.close();
|
}
|
}
|
}
|
|
|
public static void downloadDirectory(FTPClient ftpClient, String parentDir,
|
String currentDir, String saveDir) throws IOException {
|
String dirToList = parentDir;
|
if (!"".equals(currentDir)) {
|
dirToList += UNDERLINE + currentDir;
|
}
|
FTPFile[] subFiles = ftpClient.listFiles(dirToList);
|
if (subFiles != null && subFiles.length > 0) {
|
for (FTPFile aFile : subFiles) {
|
String currentFileName = aFile.getName();
|
if (".".equals(currentFileName) || "..".equals(currentFileName)) {
|
continue;
|
}
|
String filePath = parentDir + UNDERLINE + currentDir + UNDERLINE + encodeFileName(currentFileName);
|
if ("".equals(currentDir)) {
|
filePath = parentDir + UNDERLINE + encodeFileName(currentFileName);
|
}
|
String newDirPath = saveDir + parentDir + File.separator
|
+ currentDir + File.separator + currentFileName;
|
if ("".equals(currentDir)) {
|
newDirPath = saveDir + parentDir + File.separator
|
+ currentFileName;
|
}
|
if (aFile.isDirectory()) {
|
// create the directory in saveDir
|
File newDir = new File(newDirPath);
|
boolean created = newDir.mkdirs();
|
if (!created) {
|
log.error("COULD NOT create the directory: " + newDirPath);
|
}
|
// download the sub directory
|
downloadDirectory(ftpClient, dirToList, currentFileName,
|
saveDir);
|
} else {
|
// download the file
|
boolean success = downloadSingleFile(ftpClient, filePath,
|
newDirPath);
|
if (!success) {
|
log.error("COULD NOT download the file: " + filePath + ftpClient.getReplyString());
|
}
|
}
|
}
|
}
|
}
|
|
|
/**
|
* 获取合同对应的附件
|
*
|
* @param
|
* @param contractNumber
|
* @return
|
* @throws Exception
|
*/
|
public static List<JSONObject> getAttachmentUrlByContractId(String contractNumber) throws Exception {
|
FTPClient ftpClient = Global.pool.getResource();
|
List<JSONObject> list = new ArrayList<>();
|
try {
|
FTPFile[] files = null;
|
if (ftpClient.changeWorkingDirectory(getRemote(contractNumber) + contractNumber.replaceAll(UNDERLINE, ""))) {
|
files = ftpClient.listFiles();
|
for (int i = 0; i < files.length; i++) {
|
if (files[i].isFile()) {
|
JSONObject attachment = new JSONObject();
|
attachment.put("name", files[i].getName());
|
list.add(attachment);
|
}
|
}
|
}
|
} catch (Exception e) {
|
log.error("过程异常！", e);
|
} finally {
|
Global.pool.returnResource(ftpClient);
|
}
|
return list;
|
}
|
|
|
public static boolean uploadFile(String contractNumber, String fileName, InputStream in) throws IOException {
|
boolean flag = false;
|
FTPClient ftpClient = Global.pool.getResource();
|
if (contractNumber.indexOf(UNDERLINE) > -1) {
|
ftpClient.changeWorkingDirectory(GSM_CONTRACT_DIR);
|
contractNumber = contractNumber.replaceAll(UNDERLINE, "");
|
} else {
|
ftpClient.changeWorkingDirectory(GSM_ORDER_DIR);
|
}
|
|
ftpClient.mkd(contractNumber);
|
boolean changedir = ftpClient.changeWorkingDirectory(contractNumber);
|
if (changedir) {
|
if (ftpClient.storeFile(encodeFileName(fileName), in)) {
|
log.info("文件上传成功 " + fileName);
|
flag = true;
|
} else {
|
log.error("文件上传失败 " + fileName + " " + ftpClient.getReplyString());
|
}
|
}
|
in.close();
|
//关闭连接
|
Global.pool.returnResource(ftpClient);
|
return flag;
|
}
|
|
|
private static String encodeFileName(String fileName) {
|
try {
|
return new String(fileName.replaceAll(UNDERLINE, "").getBytes("UTF-8"), FTP.DEFAULT_CONTROL_ENCODING);
|
} catch (UnsupportedEncodingException e) {
|
e.printStackTrace();
|
}
|
return fileName;
|
}
|
|
|
public static String getRemote(String contractNumber) {
|
if (contractNumber.indexOf(UNDERLINE) > -1) {
|
return GSM_CONTRACT_DIR + UNDERLINE;
|
} else {
|
return GSM_ORDER_DIR + UNDERLINE;
|
}
|
}
|
|
|
}
|
