From 3d3902bfd34cdab80af235229fda5f825a60e7af Mon Sep 17 00:00:00 2001
From: yanghb <yanghb>
Date: 星期五, 21 四月 2023 14:41:10 +0800
Subject: [PATCH] 代码调整
---
ManagementTravel/guns-admin/src/main/java/com/stylefeng/guns/modular/system/controller/util/UploadUtil.java | 25 ++++++++++++++++++++++++-
1 files changed, 24 insertions(+), 1 deletions(-)
diff --git a/ManagementTravel/guns-admin/src/main/java/com/stylefeng/guns/modular/system/controller/util/UploadUtil.java b/ManagementTravel/guns-admin/src/main/java/com/stylefeng/guns/modular/system/controller/util/UploadUtil.java
index 461ff1f..467992f 100644
--- a/ManagementTravel/guns-admin/src/main/java/com/stylefeng/guns/modular/system/controller/util/UploadUtil.java
+++ b/ManagementTravel/guns-admin/src/main/java/com/stylefeng/guns/modular/system/controller/util/UploadUtil.java
@@ -82,7 +82,15 @@
try {
MultipartHttpServletRequest multipartRequest = (MultipartHttpServletRequest) request;
MultipartFile file = (MultipartFile) picture;
-
+ String name = file.getOriginalFilename();
+ String s = name.substring(name.lastIndexOf(".") + 1).toLowerCase();
+ if(!s.equals("jpg") && !s.equals("png") && !s.equals("jpeg")){
+ return null;
+ }
+ long size = file.getSize();
+ if(size > 524288000L){//500M限制
+ return "请上传500M以内的文件";
+ }
String pictureName = OssUploadUtil.ossUpload(request, file);
// String pictureName = ObsUploadUtil.obsUpload(super.getHttpServletRequest(), picture);
return pictureName;
@@ -96,6 +104,10 @@
*/
@RequestMapping("/file")
public String file(@RequestPart("file") MultipartFile picture) {
+ long size = picture.getSize();
+ if(size > 524288000L){//500M限制
+ return "请上传500M以内的文件";
+ }
//获取文件名后缀
String suffix = picture.getOriginalFilename().substring(picture.getOriginalFilename().lastIndexOf("."));
String pictureName = UUID.randomUUID().toString() + suffix;
@@ -116,6 +128,10 @@
*/
@RequestMapping("/imageUp")
public String imageUp(@RequestPart("upfile") MultipartFile picture, HttpServletRequest request) {
+ long size = picture.getSize();
+ if(size > 524288000L){//500M限制
+ return "请上传500M以内的文件";
+ }
String callback = request.getParameter("callback");
String pictureName = UUID.randomUUID().toString() + ".jpg";
try {
@@ -125,6 +141,13 @@
// 文件全路径
//pictureName = gunsProperties.getPictureServerAddress() + pictureName;
// pictureName = ObsUploadUtil.obsUpload(super.getHttpServletRequest(), picture);
+
+ String name = picture.getOriginalFilename();
+ String s = name.substring(name.lastIndexOf(".") + 1).toLowerCase();
+ if(!s.equals("jpg") && !s.equals("png") && !s.equals("jpeg")){
+ return null;
+ }
+
pictureName = OssUploadUtil.ossUpload(super.getHttpServletRequest(), picture);
String result = "{'original': '" + picture.getOriginalFilename() + "', 'state': 'SUCCESS', 'url': '" + pictureName + "'}";
--
Gitblit v1.7.1