bonus
/
Bonus-Cloud
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 2 Wiki Activity

提交漏交的文件

This commit is contained in:
weiweiw 2024-09-29 13:42:24 +08:00
parent fc121c14db
commit 12a7c2638a
3 changed files with 7 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
bonus-modules/bonus-file/src/main/java/com/bonus/file/service/impl/MongodbServiceImpl.java
View File

@ -26,10 +26,8 @@ import java.io.InputStream;
import java.io.OutputStream; import java.io.OutputStream;
import java.net.URLEncoder; import java.net.URLEncoder;
import java.nio.charset.StandardCharsets; import java.nio.charset.StandardCharsets;
import java.util.ArrayList; import java.util.*;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import org.bson.Document; import org.bson.Document;
/** /**
* @author wangvivi * @author wangvivi
@ -58,7 +56,8 @@ public class MongodbServiceImpl implements ISysFileService {
//验证文件扩展名和大小 //验证文件扩展名和大小
FileUploadUtils.assertAllowed(file, MimeTypeUtils.DEFAULT_ALLOWED_EXTENSION); FileUploadUtils.assertAllowed(file, MimeTypeUtils.DEFAULT_ALLOWED_EXTENSION);
String fileName = file.getOriginalFilename(); String fileName = Objects.requireNonNull(file.getOriginalFilename(), "文件名不能为空");
fileName = fileName.replace(" ", "");
long fileSize = file.getSize(); long fileSize = file.getSize();
// 判断是否使用 GridFS 存储 // 判断是否使用 GridFS 存储
@ -70,7 +69,7 @@ public class MongodbServiceImpl implements ISysFileService {
// 小文件直接存储为二进制数据 // 小文件直接存储为二进制数据
// 创建一个Map存储文件信息 // 创建一个Map存储文件信息
Map<String, Object> fileData = new HashMap<>(); Map<String, Object> fileData = new HashMap<>();
fileData.put("fileName", file.getOriginalFilename()); fileData.put("fileName", fileName);
fileData.put("fileSize", file.getSize()); fileData.put("fileSize", file.getSize());
fileData.put("contentType", file.getContentType()); fileData.put("contentType", file.getContentType());
// 将文件内容以byte[]存储 // 将文件内容以byte[]存储

1
bonus-modules/bonus-file/src/main/java/com/bonus/file/service/impl/ObsServiceImpl.java
View File

@ -54,6 +54,7 @@ public class ObsServiceImpl implements ISysFileService {
FileUploadUtils.assertAllowed(file, MimeTypeUtils.DEFAULT_ALLOWED_EXTENSION); FileUploadUtils.assertAllowed(file, MimeTypeUtils.DEFAULT_ALLOWED_EXTENSION);
String originalFilename = Objects.requireNonNull(file.getOriginalFilename(), "文件名不能为空"); String originalFilename = Objects.requireNonNull(file.getOriginalFilename(), "文件名不能为空");
originalFilename = originalFilename.replace(" ", "");
String extension = originalFilename.substring(originalFilename.lastIndexOf('.')); String extension = originalFilename.substring(originalFilename.lastIndexOf('.'));
String objectKey = UuidUtils.generateUuid() + extension; String objectKey = UuidUtils.generateUuid() + extension;
objectKey = FileUtils.generateObjectName(objectKey); objectKey = FileUtils.generateObjectName(objectKey);

1
bonus-modules/bonus-file/src/main/java/com/bonus/file/service/impl/OssServiceImpl.java
View File

@ -60,6 +60,7 @@ public class OssServiceImpl implements ISysFileService {
FileUploadUtils.assertAllowed(file, MimeTypeUtils.DEFAULT_ALLOWED_EXTENSION); FileUploadUtils.assertAllowed(file, MimeTypeUtils.DEFAULT_ALLOWED_EXTENSION);
String originalFilename = Objects.requireNonNull(file.getOriginalFilename(), "文件名不能为空"); String originalFilename = Objects.requireNonNull(file.getOriginalFilename(), "文件名不能为空");
originalFilename = originalFilename.replace(" ", "");
String extension = originalFilename.substring(originalFilename.lastIndexOf('.')); String extension = originalFilename.substring(originalFilename.lastIndexOf('.'));
String objectKey = UuidUtils.generateUuid() + extension; String objectKey = UuidUtils.generateUuid() + extension;
objectKey = FileUtils.generateObjectName(objectKey); objectKey = FileUtils.generateObjectName(objectKey);