diff --git a/src/main/java/com/rymcu/forest/util/FileUtils.java b/src/main/java/com/rymcu/forest/util/FileUtils.java index c0ed2b7..61ee8ab 100644 --- a/src/main/java/com/rymcu/forest/util/FileUtils.java +++ b/src/main/java/com/rymcu/forest/util/FileUtils.java @@ -7,7 +7,6 @@ import org.springframework.web.multipart.MultipartFile; import java.io.*; import java.net.URLEncoder; -import java.util.Arrays; /** * 文件操作工具类 @@ -35,13 +34,20 @@ public class FileUtils { * @return */ public static String getExtend(String filename, String defExt) { - if ((filename != null) && (filename.length() > 0)) { - int i = filename.lastIndexOf('.'); - - if ((i > 0) && (i < (filename.length() - 1))) { - return (filename.substring(i + 1)).toLowerCase(); - } + if (StringUtils.isEmpty(filename)) { + return defExt.toLowerCase(); } + + int i = filename.lastIndexOf('.'); + + if ((i > 0) && (i < (filename.length() - 1))) { + String result = filename.substring(i + 1).toLowerCase(); + if (result.contains("?")) { + return result.split("\\?")[0]; + } + return result; + } + return defExt.toLowerCase(); } diff --git a/src/main/java/com/rymcu/forest/web/api/common/UploadController.java b/src/main/java/com/rymcu/forest/web/api/common/UploadController.java index 44a2a14..c0e9455 100644 --- a/src/main/java/com/rymcu/forest/web/api/common/UploadController.java +++ b/src/main/java/com/rymcu/forest/web/api/common/UploadController.java @@ -187,8 +187,7 @@ public class UploadController { String localPath = Utils.getProperty("resource.file-path") + "/" + typePath + "/"; - String orgName = url.substring(url.lastIndexOf(".")); - String fileName = System.currentTimeMillis() + FileUtils.getExtend(orgName).toLowerCase(); + String fileName = System.currentTimeMillis() + "." + FileUtils.getExtend(url); String savePath = file.getPath() + File.separator + fileName;