From 3b10165395a56f2da479a757e1d751a65079279d Mon Sep 17 00:00:00 2001
From: 13693261870 <252740454@qq.com>
Date: 星期二, 18 十月 2022 15:04:11 +0800
Subject: [PATCH] 添加上传大文件功能
---
src/main/java/com/lf/server/service/data/UploaderService.java | 140 ++++++++++++++++++++++++++--------------------
1 files changed, 79 insertions(+), 61 deletions(-)
diff --git a/src/main/java/com/lf/server/service/data/UploaderService.java b/src/main/java/com/lf/server/service/data/UploaderService.java
index c5cb523..cdc0ce2 100644
--- a/src/main/java/com/lf/server/service/data/UploaderService.java
+++ b/src/main/java/com/lf/server/service/data/UploaderService.java
@@ -23,8 +23,8 @@
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.web.multipart.MultipartFile;
+import org.springframework.web.multipart.support.StandardMultipartHttpServletRequest;
-import javax.servlet.ServletContext;
import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
@@ -45,12 +45,6 @@
@Autowired
AttachService attachService;
-
- private final static long ONE_DAY = 24 * 60 * 60 * 1000;
-
- private final static long SIZE_MAX = 1024 * 1024 * 1024 * 1024;
-
- private final static long FILE_SIZE_MAX = 256 * 1024 * 1024 * 1024;
private final static Log log = LogFactory.getLog(UploaderService.class);
@@ -179,41 +173,92 @@
/**
* 涓婁紶鏂囦欢
*/
- public void uploadData(HttpServletRequest req, HttpServletResponse res) {
- try {
- // 澶勭悊涓枃涔辩爜闂
- req.setCharacterEncoding("utf-8");
- res.setContentType("text/html;charset=utf-8");
+ public Object uploadData(HttpServletRequest request, HttpServletResponse res) throws Exception {
+ StandardMultipartHttpServletRequest req = (StandardMultipartHttpServletRequest) request;
+ req.setCharacterEncoding("utf-8");
+ res.setContentType("application/json;charset=utf-8");
- // 妫�鏌ヨ姹傛槸/鍚︿负multipart/form-data绫诲瀷
- if (!ServletFileUpload.isMultipartContent(req)) {
- throw new RuntimeException("琛ㄥ崟鐨別nctype灞炴�т笉鏄痬ultipart/form-data绫诲瀷");
+ Map<String, String> map = getParams(req);
+ List<FileInfo> list = getFiles(req);
+
+ return list.size();
+ }
+
+ private Map<String, String> getParams(StandardMultipartHttpServletRequest req) {
+ Map<String, String> map = new HashMap<String, String>(3);
+
+ Enumeration<String> enumeration = req.getParameterNames();
+ while (enumeration.hasMoreElements()) {
+ String key = enumeration.nextElement();
+ String value = req.getParameter(key);
+ map.put(key, value);
+ }
+
+ return map;
+ }
+
+ private List<FileInfo> getFiles(StandardMultipartHttpServletRequest req) throws Exception {
+ List<FileInfo> list = new ArrayList<FileInfo>();
+
+ String path = pathHelper.getTempPath();
+ Iterator<String> iterator = req.getFileNames();
+ while (iterator.hasNext()) {
+ MultipartFile file = req.getFile(iterator.next());
+
+ FileInfo fi = new FileInfo(file.getOriginalFilename());
+ if (StringHelper.isEmpty(fi.getFileName())) {
+ continue;
}
- // 鍒涘缓涓婁紶鎵�闇�瑕佺殑涓や釜瀵硅薄锛氱鐩樻枃浠跺璞�+鏂囦欢涓婁紶瀵硅薄
- DiskFileItemFactory factory = new DiskFileItemFactory();
- ServletFileUpload sfu = new ServletFileUpload(factory);
- ServletRequestContext ctx = new ServletRequestContext(req);
+ fi.setSize(file.getSize());
+ fi.setPath(path + File.separator + fi.getFileName());
+ file.transferTo(new File(fi.getPath()));
- // 闄愬埗涓婁紶鐨勬�绘枃浠跺ぇ灏�
- sfu.setSizeMax(SIZE_MAX);
- // 闄愬埗鍗曚釜鏂囦欢鐨勫ぇ灏�
- sfu.setFileSizeMax(FILE_SIZE_MAX);
- // 璁剧疆缂栫爜鏂瑰紡
- sfu.setHeaderEncoding("utf-8");
-
- // list瀹瑰櫒鐢ㄦ潵淇濆瓨琛ㄥ崟涓殑鎵�鏈夋暟鎹俊鎭�
- List<FileItem> items = sfu.parseRequest(ctx);
- copeFileItems(items, req);
- } catch (Exception ex) {
- log.error(ex.getMessage() + ex.getStackTrace() + "\n");
+ list.add(fi);
}
+
+ return list;
+ }
+
+ public Object fileUpload(HttpServletRequest req, HttpServletResponse res) throws Exception {
+ List<FileItem> items = getFileItem(req, res);
+
+ return copeFileItems(items, req);
+ }
+
+ /**
+ * 鑾峰彇鏂囦欢椤�
+ */
+ private List<FileItem> getFileItem(HttpServletRequest req, HttpServletResponse res) throws Exception {
+ // 澶勭悊涓枃涔辩爜闂
+ req.setCharacterEncoding("utf-8");
+ // text/html;charset=utf-8
+ res.setContentType("application/json;charset=utf-8");
+
+ // 妫�鏌ヨ姹傛槸/鍚︿负multipart/form-data绫诲瀷
+ if (!ServletFileUpload.isMultipartContent(req)) {
+ throw new RuntimeException("琛ㄥ崟鐨別nctype灞炴�т笉鏄痬ultipart/form-data绫诲瀷");
+ }
+
+ // 鍒涘缓涓婁紶鎵�闇�瑕佺殑涓や釜瀵硅薄锛氱鐩樻枃浠跺璞�+鏂囦欢涓婁紶瀵硅薄
+ DiskFileItemFactory factory = new DiskFileItemFactory();
+ ServletFileUpload sfu = new ServletFileUpload(factory);
+ ServletRequestContext ctx = new ServletRequestContext(req);
+
+ // 璁剧疆缂栫爜鏂瑰紡
+ sfu.setHeaderEncoding("utf-8");
+ factory.setSizeThreshold(4096);
+
+ // 鑾峰彇琛ㄥ崟涓殑鎵�鏈夋暟鎹俊鎭�
+ List<FileItem> list = sfu.parseRequest(ctx);
+
+ return list;
}
/**
* 澶勭悊瑙ｆ瀽鍐呭锛氬鐞嗘櫘閫氳〃鍗曞煙鍜屾枃浠惰〃鍗曞煙
*/
- private void copeFileItems(List<FileItem> items, HttpServletRequest req) throws Exception {
+ private Object copeFileItems(List<FileItem> items, HttpServletRequest req) throws Exception {
Map<String, String> map = new HashMap<String, String>(3);
List<FileInfo> list = new ArrayList<FileInfo>();
@@ -231,6 +276,8 @@
list.add(fi);
}
}
+
+ return map;
}
/**
@@ -252,35 +299,6 @@
} catch (Exception ex) {
log.error(ex.getMessage() + ex.getStackTrace() + "\n");
return null;
- }
- }
-
- private void handleFileField(FileItem item, HttpServletRequest req) {
- // 鑾峰彇 鏂囦欢鏁版嵁椤逛腑鐨� 鏂囦欢鍚�
- String fileName = item.getName();
-
- // 鎺у埗鍙兘涓婁紶鍥剧墖
- String img = "image";
- if (!item.getContentType().startsWith(img)) {
- return;
- }
-
- // 鑾峰彇 褰撳墠椤圭洰涓嬬殑 /files 鐩綍鐨勭粷瀵逛綅缃�
- ServletContext ctx = req.getSession().getServletContext();
- String path = ctx.getRealPath("/files");
-
- // 鍒涘缓 file瀵硅薄
- File file = new File(path);
- if (!file.exists()) {
- // 鍒涘缓鐩綍
- file.mkdir();
- }
-
- // 灏嗘枃浠朵繚瀛樺埌鏈嶅姟鍣ㄤ笂锛圲UID鏄�氱敤鍞竴鏍囪瘑鐮侊級
- try {
- item.write(new File(file.toString(), UUID.randomUUID() + "_" + fileName));
- } catch (Exception e) {
- e.printStackTrace();
}
}
}
--
Gitblit v1.9.3