From e2ba375f5168608c6f86ca2a43b510b3f3db7d12 Mon Sep 17 00:00:00 2001
From: java110 <928255095@qq.com>
Date: 星期四, 20 十月 2022 13:12:45 +0800
Subject: [PATCH] 优化代码
---
java110-core/src/main/java/com/java110/core/client/FileUploadTemplate.java | 38 +++++++++++++++++++++++++++++++++-----
1 files changed, 33 insertions(+), 5 deletions(-)
diff --git a/java110-core/src/main/java/com/java110/core/client/FileUploadTemplate.java b/java110-core/src/main/java/com/java110/core/client/FileUploadTemplate.java
index cc41fc0..eabf179 100644
--- a/java110-core/src/main/java/com/java110/core/client/FileUploadTemplate.java
+++ b/java110-core/src/main/java/com/java110/core/client/FileUploadTemplate.java
@@ -11,6 +11,8 @@
@Component
public class FileUploadTemplate {
+ private static final String ROOT_PATH = "hc/";
+
@Autowired
private FtpUploadTemplate ftpUploadTemplate;
@@ -27,22 +29,48 @@
public String saveFile(InputStream inputStream,String fileName){
+ String newfileName = ROOT_PATH+ fileName;
+
String ossSwitch = MappingCache.getValue(OSSUtil.DOMAIN, OSSUtil.OSS_SWITCH);
if (OSSUtil.OSS_SWITCH_OSS.equals(ossSwitch)) {
- fileName = ossUploadTemplate.upload(inputStream, fileName);
+ newfileName = ossUploadTemplate.upload(inputStream, newfileName);
} else if (COSUtil.COS_SWITCH_COS.equals(ossSwitch)) {
- fileName = cosUploadTemplate.upload(inputStream, fileName);
+ newfileName = cosUploadTemplate.upload(inputStream, newfileName);
} else {
String ftpServer = MappingCache.getValue(FtpUploadTemplate.FTP_DOMAIN, FtpUploadTemplate.FTP_SERVER);
int ftpPort = Integer.parseInt(MappingCache.getValue(FtpUploadTemplate.FTP_DOMAIN, FtpUploadTemplate.FTP_PORT));
String ftpUserName = MappingCache.getValue(FtpUploadTemplate.FTP_DOMAIN, FtpUploadTemplate.FTP_USERNAME);
String ftpUserPassword = MappingCache.getValue(FtpUploadTemplate.FTP_DOMAIN, FtpUploadTemplate.FTP_USERPASSWORD);
- fileName = ftpUploadTemplate.upload(inputStream, ftpServer,
+ newfileName = ftpUploadTemplate.upload(inputStream, ftpServer,
ftpPort, ftpUserName,
- ftpUserPassword, fileName);
+ ftpUserPassword, newfileName);
}
- return fileName;
+ return newfileName;
+ }
+
+ public InputStream downloadFile(String fileName){
+
+ InputStream inputStream = null;
+
+ String ossSwitch = MappingCache.getValue(OSSUtil.DOMAIN, OSSUtil.OSS_SWITCH);
+
+ if (OSSUtil.OSS_SWITCH_OSS.equals(ossSwitch)) {
+ inputStream = ossUploadTemplate.download(fileName);
+ } else if (COSUtil.COS_SWITCH_COS.equals(ossSwitch)) {
+ inputStream = cosUploadTemplate.download(fileName);
+ } else {
+ String ftpServer = MappingCache.getValue(FtpUploadTemplate.FTP_DOMAIN, FtpUploadTemplate.FTP_SERVER);
+ int ftpPort = Integer.parseInt(MappingCache.getValue(FtpUploadTemplate.FTP_DOMAIN, FtpUploadTemplate.FTP_PORT));
+ String ftpUserName = MappingCache.getValue(FtpUploadTemplate.FTP_DOMAIN, FtpUploadTemplate.FTP_USERNAME);
+ String ftpUserPassword = MappingCache.getValue(FtpUploadTemplate.FTP_DOMAIN, FtpUploadTemplate.FTP_USERPASSWORD);
+
+ inputStream = ftpUploadTemplate.download(fileName, ftpServer,
+ ftpPort, ftpUserName,
+ ftpUserPassword);
+ }
+
+ return inputStream;
}
}
--
Gitblit v1.8.0