diff --git a/bonus-modules/bonus-file/src/main/java/com/bonus/file/service/impl/MongodbServiceImpl.java b/bonus-modules/bonus-file/src/main/java/com/bonus/file/service/impl/MongodbServiceImpl.java index 0fb2299..bc5a670 100644 --- a/bonus-modules/bonus-file/src/main/java/com/bonus/file/service/impl/MongodbServiceImpl.java +++ b/bonus-modules/bonus-file/src/main/java/com/bonus/file/service/impl/MongodbServiceImpl.java @@ -26,10 +26,8 @@ import java.io.InputStream; import java.io.OutputStream; import java.net.URLEncoder; import java.nio.charset.StandardCharsets; -import java.util.ArrayList; -import java.util.HashMap; -import java.util.List; -import java.util.Map; +import java.util.*; + import org.bson.Document; /** * @author wangvivi @@ -58,7 +56,8 @@ public class MongodbServiceImpl implements ISysFileService { //验证文件扩展名和大小 FileUploadUtils.assertAllowed(file, MimeTypeUtils.DEFAULT_ALLOWED_EXTENSION); - String fileName = file.getOriginalFilename(); + String fileName = Objects.requireNonNull(file.getOriginalFilename(), "文件名不能为空"); + fileName = fileName.replace(" ", ""); long fileSize = file.getSize(); // 判断是否使用 GridFS 存储 @@ -70,7 +69,7 @@ public class MongodbServiceImpl implements ISysFileService { // 小文件直接存储为二进制数据 // 创建一个Map存储文件信息 Map fileData = new HashMap<>(); - fileData.put("fileName", file.getOriginalFilename()); + fileData.put("fileName", fileName); fileData.put("fileSize", file.getSize()); fileData.put("contentType", file.getContentType()); // 将文件内容以byte[]存储 diff --git a/bonus-modules/bonus-file/src/main/java/com/bonus/file/service/impl/ObsServiceImpl.java b/bonus-modules/bonus-file/src/main/java/com/bonus/file/service/impl/ObsServiceImpl.java index 223c2b9..d18c978 100644 --- a/bonus-modules/bonus-file/src/main/java/com/bonus/file/service/impl/ObsServiceImpl.java +++ b/bonus-modules/bonus-file/src/main/java/com/bonus/file/service/impl/ObsServiceImpl.java @@ -54,6 +54,7 @@ public class ObsServiceImpl implements ISysFileService { FileUploadUtils.assertAllowed(file, MimeTypeUtils.DEFAULT_ALLOWED_EXTENSION); String originalFilename = Objects.requireNonNull(file.getOriginalFilename(), "文件名不能为空"); + originalFilename = originalFilename.replace(" ", ""); String extension = originalFilename.substring(originalFilename.lastIndexOf('.')); String objectKey = UuidUtils.generateUuid() + extension; objectKey = FileUtils.generateObjectName(objectKey); diff --git a/bonus-modules/bonus-file/src/main/java/com/bonus/file/service/impl/OssServiceImpl.java b/bonus-modules/bonus-file/src/main/java/com/bonus/file/service/impl/OssServiceImpl.java index 7bdaab0..e18ae2c 100644 --- a/bonus-modules/bonus-file/src/main/java/com/bonus/file/service/impl/OssServiceImpl.java +++ b/bonus-modules/bonus-file/src/main/java/com/bonus/file/service/impl/OssServiceImpl.java @@ -60,6 +60,7 @@ public class OssServiceImpl implements ISysFileService { FileUploadUtils.assertAllowed(file, MimeTypeUtils.DEFAULT_ALLOWED_EXTENSION); String originalFilename = Objects.requireNonNull(file.getOriginalFilename(), "文件名不能为空"); + originalFilename = originalFilename.replace(" ", ""); String extension = originalFilename.substring(originalFilename.lastIndexOf('.')); String objectKey = UuidUtils.generateUuid() + extension; objectKey = FileUtils.generateObjectName(objectKey);