From c0a828cca9ee3480c8797891732789e574ba2200 Mon Sep 17 00:00:00 2001
From: 13693261870 <252740454@qq.com>
Date: 星期五, 21 十月 2022 09:18:29 +0800
Subject: [PATCH] 1
---
src/main/java/com/lf/server/service/data/UploaderService.java | 223 +++++++++----------------------------------------------
1 files changed, 36 insertions(+), 187 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 13e7a2e..f12f164 100644
--- a/src/main/java/com/lf/server/service/data/UploaderService.java
+++ b/src/main/java/com/lf/server/service/data/UploaderService.java
@@ -1,21 +1,11 @@
package com.lf.server.service.data;
-import com.alibaba.fastjson.JSON;
-import com.lf.server.controller.all.BaseController;
-import com.lf.server.entity.all.HttpStatus;
-import com.lf.server.entity.all.ResponseMsg;
-import com.lf.server.entity.all.SettingData;
import com.lf.server.entity.ctrl.FileInfo;
-import com.lf.server.entity.data.MetaEntity;
import com.lf.server.entity.data.MetaFileEntity;
-import com.lf.server.entity.sys.AttachEntity;
import com.lf.server.entity.sys.UserEntity;
import com.lf.server.helper.FileHelper;
import com.lf.server.helper.PathHelper;
import com.lf.server.helper.StringHelper;
-import com.lf.server.helper.WebHelper;
-import com.lf.server.service.sys.AttachService;
-import org.apache.commons.codec.digest.DigestUtils;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.apache.tomcat.util.http.fileupload.FileItem;
@@ -27,14 +17,10 @@
import org.springframework.web.multipart.MultipartFile;
import org.springframework.web.multipart.support.StandardMultipartHttpServletRequest;
-import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.File;
-import java.io.FileInputStream;
-import java.io.IOException;
import java.lang.reflect.Field;
-import java.net.URLEncoder;
import java.sql.Timestamp;
import java.util.*;
@@ -47,213 +33,58 @@
@Autowired
PathHelper pathHelper;
- @Autowired
- AttachService attachService;
-
private final static Log log = LogFactory.getLog(UploaderService.class);
- private static final String NO_FILE = JSON.toJSONString(new ResponseMsg<String>(HttpStatus.NOT_FOUND, "鏂囦欢鎵句笉鍒�"));
-
/**
* 涓婁紶鏂囦欢
*/
- public ResponseMsg<String> upload(UserEntity ue, String tab, MultipartFile file, BaseController ctrl) {
- try {
- if (file == null && file.isEmpty()) {
- return ctrl.fail("鏂囦欢涓婁紶涓虹┖", null);
- }
- if (file.getSize() > SettingData.MAX_FILE_SIZE) {
- return ctrl.fail(String.format("鏂囦欢澶т簬 %d MB", SettingData.MAX_FILE_SIZE / 1024 / 1024), null);
- }
-
- // 浼犺緭鏂囦欢
- String oldName = file.getOriginalFilename();
- String filePath = pathHelper.getTempPath() + File.separator + oldName;
- File newFile = new File(filePath);
- file.transferTo(newFile);
- double sizes = FileHelper.sizeToMb(file.getSize());
-
- // 鑾峰彇MD5
- String md5 = getFileMd5(filePath);
- AttachEntity entity = attachService.selectByGuid(md5);
- if (entity != null) {
- newFile.delete();
- return ctrl.success("鏂囦欢宸插瓨鍦�", md5);
- }
-
- AttachEntity ae = getAttachEntity(ue, tab, oldName, md5, sizes);
- String targetPath = pathHelper.getConfig().getUploadPath() + File.separator + ae.getPath();
- newFile.renameTo(new File(targetPath));
-
- int rows = attachService.insert(ae);
-
- return rows > 0 ? ctrl.success(md5) : ctrl.fail("淇濆瓨鏂囦欢澶辫触", null);
- } catch (Exception ex) {
- return ctrl.fail(ex.getMessage(), null);
- }
- }
-
- /**
- * 鑾峰彇鏂囦欢 MD5 鐮�
- */
- private String getFileMd5(String filePath) throws IOException {
- FileInputStream fileStream = new FileInputStream(filePath);
- String md5 = DigestUtils.md5Hex(fileStream);
- fileStream.close();
-
- return md5;
- }
-
- /**
- * 鑾峰彇闄勪欢瀹炰綋绫�
- */
- protected AttachEntity getAttachEntity(UserEntity ue, String tab, String oldName, String md5, double sizes) {
- AttachEntity entity = new AttachEntity();
- entity.setName(oldName);
- entity.setTab(tab);
- entity.setGuid(md5);
- String subPath = PathHelper.getUploadPath() + File.separator + md5;
- entity.setPath(subPath);
- entity.setSizes(sizes);
- if (ue != null) {
- entity.setCreateUser(ue.getId());
- }
-
- return entity;
- }
-
- /**
- * 涓嬭浇鏂囦欢
- */
- public void download(String guid, HttpServletResponse res) {
- try {
- if (StringHelper.isEmpty(guid)) {
- WebHelper.write2Page(res, NO_FILE);
- return;
- }
-
- AttachEntity entity = attachService.selectByGuid(guid);
- if (entity == null) {
- WebHelper.write2Page(res, NO_FILE);
- return;
- }
-
- setDownloadResponse(entity, res);
- String filePath = pathHelper.getConfig().getUploadPath() + File.separator + entity.getPath();
-
- // 閫氳繃response瀵硅薄锛岃幏鍙栧埌杈撳嚭娴�
- ServletOutputStream outputStream = res.getOutputStream();
- // 瀹氫箟杈撳叆娴侊紝閫氳繃杈撳叆娴佽鍙栨枃浠跺唴瀹�
- FileInputStream fileInputStream = new FileInputStream(filePath);
-
- int len = 0;
- byte[] bytes = new byte[1024];
- while ((len = fileInputStream.read(bytes)) != -1) {
- // 閫氳繃杈撳叆娴佽鍙栨枃浠舵暟鎹紝鐒跺悗閫氳繃涓婅堪鐨勮緭鍑烘祦鍐欏洖娴忚鍣�
- outputStream.write(bytes, 0, len);
- outputStream.flush();
- }
-
- // 鍏抽棴璧勬簮
- outputStream.close();
- fileInputStream.close();
- } catch (Exception ex) {
- try {
- String msg = JSON.toJSONString(new ResponseMsg<String>(HttpStatus.ERROR, "鏂囦欢涓嬭浇鍑洪敊"));
- WebHelper.write2Page(res, msg);
- } catch (Exception e) {
- log.error(e.getStackTrace());
- }
- log.error(ex.getStackTrace());
- }
- }
-
- /**
- * 璁剧疆涓嬭浇鍝嶅簲淇℃伅
- */
- private void setDownloadResponse(AttachEntity entity, HttpServletResponse res) throws IOException {
- String fileName = URLEncoder.encode(entity.getName(), "UTF-8");
-
- // 璁剧疆鍝嶅簲澶翠腑鏂囦欢鐨勪笅杞芥柟寮忎负闄勪欢鏂瑰紡锛屼互鍙婅缃枃浠跺悕
- res.setHeader("Content-Disposition", "attachment; filename=" + fileName);
- // 璁剧疆鍝嶅簲澶寸殑缂栫爜鏍煎紡涓篣TF-8
- res.setCharacterEncoding("UTF-8");
-
- // 閫氳繃response瀵硅薄璁剧疆鍝嶅簲鏁版嵁鏍煎紡(濡傦細"text/plain; charset=utf-8")
- String ext = FileHelper.getExtension(entity.getName());
- String mime = FileHelper.getMime(ext);
- res.setContentType(mime);
- }
-
- /**
- * 涓婁紶鏂囦欢
- */
- public Object uploadData(UserEntity ue, HttpServletRequest request, HttpServletResponse res) throws Exception {
+ public <T> List<MetaFileEntity> uploadData(T t, UserEntity ue, HttpServletRequest request, HttpServletResponse res) throws Exception {
StandardMultipartHttpServletRequest req = (StandardMultipartHttpServletRequest) request;
req.setCharacterEncoding("utf-8");
res.setContentType("application/json;charset=utf-8");
- MetaEntity me = getMetaEntity(req);
+ setEntity(t, req);
List<MetaFileEntity> list = getFiles(req);
- //
- return list.size();
+ return list;
}
- private MetaEntity getMetaEntity(StandardMultipartHttpServletRequest req) {
- MetaEntity me = new MetaEntity();
-
+ /**
+ * 璁剧疆瀹炰綋绫�
+ * class<T> clazz
+ * T t = clazz.newInstance();
+ */
+ private <T> void setEntity(T t, StandardMultipartHttpServletRequest req) {
Enumeration<String> enumeration = req.getParameterNames();
while (enumeration.hasMoreElements()) {
String key = enumeration.nextElement();
try {
- Field field = me.getClass().getDeclaredField(key);
+ Field field = t.getClass().getDeclaredField(key);
if (field != null) {
field.setAccessible(true);
String value = req.getParameter(key);
+
switch (field.getType().toString()) {
- case "class java.lang.String":
- field.set(me, value);
- break;
case "long":
- field.set(me, Long.valueOf(value));
+ field.set(t, Long.valueOf(value));
break;
case "int":
- field.set(me, Integer.valueOf(value));
+ field.set(t, Integer.valueOf(value));
break;
case "class java.sql.Timestamp":
- field.set(me, Timestamp.valueOf(value));
+ field.set(t, Timestamp.valueOf(value));
break;
+ //case "class java.lang.String":
default:
- field.set(me, value);
+ field.set(t, value);
break;
}
}
} catch (Exception ex) {
- //
+ log.error(ex.getStackTrace());
}
}
-
- return me;
- }
-
- /**
- * 鑾峰彇鍙傛暟
- * Enumeration<String> headers = req.getHeaderNames();
- * Enumeration<String> attributes = req.getAttributeNames();
- */
- 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;
}
/**
@@ -277,7 +108,7 @@
mf.setPath(path + File.separator + mf.getName());
file.transferTo(new File(mf.getPath()));
- mf.setGuid(getFileMd5(mf.getPath()));
+ mf.setGuid(FileHelper.getFileMd5(mf.getPath()));
list.add(mf);
}
@@ -366,4 +197,22 @@
return null;
}
}
+
+ /**
+ * 鑾峰彇鍙傛暟
+ * Enumeration<String> headers = req.getHeaderNames();
+ * Enumeration<String> attributes = req.getAttributeNames();
+ */
+ 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;
+ }
}
--
Gitblit v1.9.3