From ba1c0ad6cffaa8a6288a4e66d0598d16f1cccc10 Mon Sep 17 00:00:00 2001
From: java110 <928255095@qq.com>
Date: 星期三, 13 十月 2021 11:00:35 +0800
Subject: [PATCH] 优化代码
---
java110-core/src/main/java/com/java110/core/client/FtpUploadTemplate.java | 144 ++++++++++++++++++++++++++++++++++++++++++-----
1 files changed, 128 insertions(+), 16 deletions(-)
diff --git a/java110-core/src/main/java/com/java110/core/client/FtpUploadTemplate.java b/java110-core/src/main/java/com/java110/core/client/FtpUploadTemplate.java
old mode 100644
new mode 100755
index 1ced81b..bd63e58
--- a/java110-core/src/main/java/com/java110/core/client/FtpUploadTemplate.java
+++ b/java110-core/src/main/java/com/java110/core/client/FtpUploadTemplate.java
@@ -16,6 +16,7 @@
@Component
public class FtpUploadTemplate {
+
private static Logger logger = LoggerFactory.getLogger(FtpUploadTemplate.class);
/*
@@ -24,15 +25,21 @@
* private static String userPassword ="MXUsssMjhssE+*=a3C4\\0";//瀵嗙爜
*/
private static String ftpPath = "uploadFiles"; // 鏂囦欢涓婁紶鐩綍
-
private static String LOCAL_CHARSET = "GBK";
private static String SERVER_CHARSET = "ISO-8859-1";
private final static String localpath = "F:/";//涓嬭浇鍒癋鐩樹笅
private final static String fileSeparator = System.getProperty("file.separator");
-
private final static String DEFAULT_IMG_SUFFIX = ".jpg";
-
private final static String IMAGE_DEFAULT_PATH = "img/";
+
+
+ public final static String FTP_DOMAIN = "FTP_DOMAIN";
+ public final static String FTP_SERVER = "FTP_SERVER";
+ public final static String FTP_PORT = "FTP_PORT";
+ public final static String FTP_USERNAME = "FTP_USERNAME";
+ public final static String FTP_USERPASSWORD = "FTP_USERPASSWORD";
+ public final static String FTP_PATH = "FTP_PATH";
+
/*
*鍥剧墖涓婁紶宸ュ叿鏂规硶
@@ -42,10 +49,13 @@
String userName, String userPassword, String ftpPath) {
String fileName = "";
FTPClient ftpClient = null;
+ ByteArrayInputStream is = null;
try {
ftpClient = new FTPClient();
// request.setCharacterEncoding("utf-8");
- ftpClient.connect(server, port);
+ if (!ftpClient.isConnected()) {
+ ftpClient.connect(server, port);
+ }
ftpClient.login(userName, userPassword);
ftpClient.enterLocalPassiveMode();
ftpClient.setFileType(FTP.BINARY_FILE_TYPE);
@@ -57,7 +67,6 @@
LOCAL_CHARSET = "UTF-8";
}
fileName = UUID.randomUUID().toString();
-
if (imageBase64.contains("data:image/png;base64,")) {
imageBase64 = imageBase64.replace("data:image/png;base64,", "");
fileName += ".png";
@@ -67,6 +76,12 @@
} else if (imageBase64.contains("data:image/webp;base64,")) {
imageBase64 = imageBase64.replace("data:image/webp;base64,", "");
fileName += ".jpg";
+ } else if (imageBase64.contains("data:application/octet-stream;base64,")) {
+ imageBase64 = imageBase64.replace("data:application/octet-stream;base64,", "");
+ fileName += ".jpg";
+ } else if (imageBase64.contains("mp4") || imageBase64.contains("MP4") || imageBase64.contains("AVI") || imageBase64.contains("avi")
+ || imageBase64.contains("WMV") || imageBase64.contains("wmv")) {
+ fileName += ".mp4";
} else {
fileName += ".jpg";
}
@@ -77,12 +92,9 @@
System.out.println("this file exist ftp");
ftpClient.deleteFile(fs[0].getName());
}
-
-
byte[] context = Base64Convert.base64ToByte(imageBase64);
- ByteArrayInputStream is = new ByteArrayInputStream(context);
+ is = new ByteArrayInputStream(context);
boolean saveFlag = ftpClient.storeFile(fileName, is);
- is.close();
if (!saveFlag) {
throw new IllegalArgumentException("瀛樺偍鏂囦欢澶辫触");
}
@@ -92,6 +104,9 @@
} finally {
try {
ftpClient.disconnect();
+ if (is != null) {
+ is.close();
+ }
} catch (IOException e) {
e.printStackTrace();
logger.error("鍏抽棴ftpClient 澶辫触", e);
@@ -99,7 +114,6 @@
}
return IMAGE_DEFAULT_PATH + DateUtil.getNowII() + "/" + fileName;
}
-
/*
*鏂囦欢涓婁紶宸ュ叿鏂规硶
@@ -111,7 +125,9 @@
try {
// request.setCharacterEncoding("utf-8");
ftpClient = new FTPClient();
- ftpClient.connect(server, port);
+ if (!ftpClient.isConnected()) {
+ ftpClient.connect(server, port);
+ }
ftpClient.login(userName, userPassword);
ftpClient.enterLocalPassiveMode();
ftpClient.setFileType(FTP.BINARY_FILE_TYPE);
@@ -138,7 +154,7 @@
throw new IllegalArgumentException("瀛樺偍鏂囦欢澶辫触");
}
} catch (Exception e) {
- // logger.error("涓婁紶鏂囦欢澶辫触", e);
+ logger.error("涓婁紶鏂囦欢澶辫触", e);
throw new IllegalArgumentException("涓婁紶鏂囦欢澶辫触");
} finally {
try {
@@ -158,7 +174,9 @@
FTPClient ftpClient = null;
try {
ftpClient = new FTPClient();
- ftpClient.connect(server, port);
+ if (!ftpClient.isConnected()) {
+ ftpClient.connect(server, port);
+ }
ftpClient.login(userName, userPassword);
ftpClient.enterLocalPassiveMode();
if (ftpClient != null) {
@@ -167,16 +185,29 @@
FTP.DEFAULT_CONTROL_ENCODING);//闃叉涔辩爜
InputStream ins = ftpClient.retrieveFileStream(f);//闇�浣跨敤file.getName鑾峰�硷紝鑻ョ敤f浼氫贡鐮�
ByteArrayOutputStream byteOut = new ByteArrayOutputStream();
- byte[] buf = new byte[204800];
+ byte[] buf = new byte[2048];
int bufsize = 0;
while (ins != null && (bufsize = ins.read(buf, 0, buf.length)) != -1) {
byteOut.write(buf, 0, bufsize);
}
- return_arraybyte = byteOut.toByteArray();
+ //return_arraybyte = byteOut.toByteArray();
+ ByteArrayInputStream fis = new ByteArrayInputStream(byteOut.toByteArray());
+ byteOut.flush();
byteOut.close();
+ byte[] buffer = new byte[fis.available()];
+ int offset = 0;
+ int numRead = 0;
+ while (offset < buffer.length && (numRead = fis.read(buffer, offset, buffer.length - offset)) >= 0) {
+ offset += numRead;
+ }
+ if (offset != buffer.length) {
+ throw new IOException("Could not completely read file ");
+ }
+ fis.close();
if (ins != null) {
ins.close();
}
+ return_arraybyte = buffer;
}
} catch (Exception e) {
e.printStackTrace();
@@ -185,6 +216,88 @@
closeConnect(ftpClient);
}
return return_arraybyte;
+ }
+
+ public static void main(String[] args) {
+ FtpUploadTemplate ftpUploadTemplate = new FtpUploadTemplate();
+ String img = ftpUploadTemplate.download("/hc/img/20200518/", "ed05abae-2eca-40ff-81a8-b586ff2e6a36.jpg",
+ "118.89.243.11", 617, "hcdemo", "45j74jpWTf7bNhnC");
+ System.out.printf("img=" + img);
+ }
+
+ public String download(String remotePath, String fileName, String server, int port, String userName, String userPassword) {
+ InputStream is = null;
+ ByteArrayOutputStream bos = null;
+ ByteArrayInputStream fis = null;
+ FTPClient ftpClient = new FTPClient();
+ try {
+ if (!ftpClient.isConnected()) {
+ ftpClient.connect(server, port);
+ }
+ ftpClient.login(userName, userPassword);
+ ftpClient.setFileType(FTPClient.BINARY_FILE_TYPE);
+ ftpClient.enterLocalPassiveMode();
+ int reply = ftpClient.getReplyCode();
+ if (!FTPReply.isPositiveCompletion(reply)) {
+ ftpClient.disconnect();
+ }
+ String f = new String(
+ (remotePath + fileName).getBytes("GBK"),
+ FTP.DEFAULT_CONTROL_ENCODING);
+ is = ftpClient.retrieveFileStream(f);// 鑾峰彇杩滅▼ftp涓婃寚瀹氭枃浠剁殑InputStream
+ if (null == is) {
+ throw new FileNotFoundException(remotePath);
+ }
+ bos = new ByteArrayOutputStream();
+ int length;
+ byte[] buf = new byte[2048];
+ while (-1 != (length = is.read(buf, 0, buf.length))) {
+ bos.write(buf, 0, length);
+ }
+ fis = new ByteArrayInputStream(
+ bos.toByteArray());
+ bos.flush();
+ byte[] buffer = new byte[fis.available()];
+ int offset = 0;
+ int numRead = 0;
+ while (offset < buffer.length && (numRead = fis.read(buffer, offset, buffer.length - offset)) >= 0) {
+ offset += numRead;
+ }
+ if (offset != buffer.length) {
+ throw new IOException("Could not completely read file ");
+ }
+ return Base64Convert.byteToBase64(buffer);
+ } catch (Exception e) {
+ logger.error("ftp閫氳繃鏂囦欢鍚嶇О鑾峰彇杩滅▼鏂囦欢娴�", e);
+ } finally {
+ if (bos != null) {
+ try {
+ bos.close();
+ } catch (IOException e) {
+ e.printStackTrace();
+ }
+ }
+ if (is != null) {
+ try {
+ is.close();
+ } catch (IOException e) {
+ e.printStackTrace();
+ }
+ }
+ if (fis != null) {
+ try {
+ fis.close();
+ } catch (IOException e) {
+ e.printStackTrace();
+ }
+ }
+ try {
+ closeConnect(ftpClient);
+ } catch (Exception e) {
+ e.printStackTrace();
+ }
+ }
+ return null;
}
public void closeConnect(FTPClient ftpClient) {
@@ -236,5 +349,4 @@
return false;
}
}
-
}
--
Gitblit v1.8.0