hhyykk/ipms-sjy
1
1
Fork 0
mirror of https://gitee.com/hhyykk/ipms-sjy.git synced 2025-07-25 00:15:06 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

1. 修改文件名和文件路径字段长度一致，因为如果不存在文件名的情况下，会使用文件路径作为字段名，防止数据库报错字段内容太长。

Browse Source 
2. 修改type字段为ext_name，因为此字段本来就存的文件扩展名，并且新增mime_type字段，用来存储文件的类型。方便后期拓展网盘项目。
3. 修改文件路径生成方式：之前是md5，现在是sha256，降低文件碰撞概率。方便拓展网盘项目，实现秒传等功能。
This commit is contained in:
jiangqiang
2022-07-07 09:40:23 +08:00
parent d14ad888e2
commit 36d7775171
12 changed files with 58 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
yudao-module-infra/yudao-module-infra-api/src/main/java/cn/iocoder/yudao/module/infra/api/file/FileApi.java
View File

@ -14,7 +14,7 @@ public interface FileApi {
* @return 文件路径
*/
default String createFile(byte[] content) {
return createFile(null, null, content);
return createFile(null, null, "application/octet-stream", content);
}
/**
@ -25,17 +25,18 @@ public interface FileApi {
* @return 文件路径
*/
default String createFile(String path, byte[] content) {
return createFile(null, path, content);
return createFile(null, path, "application/octet-stream", content);
}
/**
* 保存文件，并返回文件的访问路径
*
* @param name 文件名称
* @param path 文件路径
* @param content 文件内容
* @param name 文件名称
* @param path 文件路径
* @param mimeType 文件类型
* @param content 文件内容
* @return 文件路径
*/
String createFile(String name, String path, byte[] content);
String createFile(String name, String path, String mimeType, byte[] content);
}

4
yudao-module-infra/yudao-module-infra-biz/src/main/java/cn/iocoder/yudao/module/infra/api/file/FileApiImpl.java
View File

@ -19,8 +19,8 @@ public class FileApiImpl implements FileApi {
private FileService fileService;
@Override
public String createFile(String name, String path, byte[] content) {
return fileService.createFile(name, path, content);
public String createFile(String name, String path, String mimeType, byte[] content) {
return fileService.createFile(name, path, mimeType, content);
}
}

2
yudao-module-infra/yudao-module-infra-biz/src/main/java/cn/iocoder/yudao/module/infra/controller/admin/file/FileController.java
View File

@ -46,7 +46,7 @@ public class FileController {
@OperateLog(logArgs = false) // 上传文件，没有记录操作日志的必要
public CommonResult<String> uploadFile(@RequestParam("file") MultipartFile file,
@RequestParam(value = "path", required = false) String path) throws Exception {
return success(fileService.createFile(file.getOriginalFilename(), path, IoUtil.readBytes(file.getInputStream())));
return success(fileService.createFile(file.getOriginalFilename(), path, file.getContentType(), IoUtil.readBytes(file.getInputStream())));
}
@DeleteMapping("/delete")

12
yudao-module-infra/yudao-module-infra-biz/src/main/java/cn/iocoder/yudao/module/infra/dal/dataobject/file/FileDO.java
View File

@ -38,7 +38,7 @@ public class FileDO extends BaseDO {
*/
private String name;
/**
* 路径，即文件名
* 路径，即文件名，唯一不可重复
*/
private String path;
/**
@ -46,11 +46,15 @@ public class FileDO extends BaseDO {
*/
private String url;
/**
* 文件类型
*
* 文件扩展名
* <p>
* 通过 {@link cn.hutool.core.io.FileTypeUtil#getType(InputStream)} 获取
*/
private String type;
private String extName;
/**
* 文件的MIME类型，默认为"application/octet-stream"
*/
private String mimeType;
/**
* 文件大小
*/

2
yudao-module-infra/yudao-module-infra-biz/src/main/java/cn/iocoder/yudao/module/infra/dal/mysql/file/FileMapper.java
View File

@ -18,7 +18,7 @@ public interface FileMapper extends BaseMapperX<FileDO> {
default PageResult<FileDO> selectPage(FilePageReqVO reqVO) {
return selectPage(reqVO, new LambdaQueryWrapperX<FileDO>()
.likeIfPresent(FileDO::getPath, reqVO.getPath())
.likeIfPresent(FileDO::getType, reqVO.getType())
.likeIfPresent(FileDO::getExtName, reqVO.getType())
.betweenIfPresent(FileDO::getCreateTime, reqVO.getBeginCreateTime(), reqVO.getEndCreateTime())
.orderByDesc(FileDO::getId));
}

3
yudao-module-infra/yudao-module-infra-biz/src/main/java/cn/iocoder/yudao/module/infra/service/file/FileService.java
View File

@ -24,10 +24,11 @@ public interface FileService {
*
* @param name 文件名称
* @param path 文件路径
* @param mimeType 文件MIME类型
* @param content 文件内容
* @return 文件路径
*/
String createFile(String name, String path, byte[] content);
String createFile(String name, String path, String mimeType,byte[] content);
/**
* 删除文件

14
yudao-module-infra/yudao-module-infra-biz/src/main/java/cn/iocoder/yudao/module/infra/service/file/FileServiceImpl.java
View File

@ -1,6 +1,7 @@
package cn.iocoder.yudao.module.infra.service.file;
import cn.hutool.core.io.FileTypeUtil;
import cn.hutool.core.io.file.FileNameUtil;
import cn.hutool.core.lang.Assert;
import cn.hutool.core.util.StrUtil;
import cn.hutool.crypto.digest.DigestUtil;
@ -39,11 +40,13 @@ public class FileServiceImpl implements FileService {
@Override
@SneakyThrows
public String createFile(String name, String path, byte[] content) {
// 计算默认的 path 名
String type = FileTypeUtil.getType(new ByteArrayInputStream(content), name);
public String createFile(String name, String path, String mimeType, byte[] content) {
//获取文件的真实扩展名
String extName = FileTypeUtil.getType(new ByteArrayInputStream(content), name);
FileNameUtil.extName(name);
if (StrUtil.isEmpty(path)) {
path = DigestUtil.md5Hex(content) + '.' + type;
//使用sha256计算文件都唯一路径，降低碰撞概率
path = DigestUtil.sha256Hex(content) + '.' + extName;
}
// 如果 name 为空，则使用 path 填充
if (StrUtil.isEmpty(name)) {
@ -61,7 +64,8 @@ public class FileServiceImpl implements FileService {
file.setName(name);
file.setPath(path);
file.setUrl(url);
file.setType(type);
file.setExtName(extName);
file.setMimeType(mimeType);
file.setSize(content.length);
fileMapper.insert(file);
return url;

8
yudao-module-infra/yudao-module-infra-biz/src/test/java/cn/iocoder/yudao/module/infra/service/file/FileServiceTest.java
View File

@ -40,7 +40,7 @@ public class FileServiceTest extends BaseDbUnitTest {
// mock 数据
FileDO dbFile = randomPojo(FileDO.class, o -> { // 等会查询到
o.setPath("yunai");
o.setType("jpg");
o.setExtName("jpg");
o.setCreateTime(buildTime(2021, 1, 15));
});
fileMapper.insert(dbFile);
@ -48,7 +48,7 @@ public class FileServiceTest extends BaseDbUnitTest {
fileMapper.insert(ObjectUtils.cloneIgnoreId(dbFile, o -> o.setPath("tudou")));
// 测试 type 不匹配
fileMapper.insert(ObjectUtils.cloneIgnoreId(dbFile, o -> {
o.setType("png");
o.setExtName("png");
}));
// 测试 createTime 不匹配
fileMapper.insert(ObjectUtils.cloneIgnoreId(dbFile, o -> {
@ -82,7 +82,7 @@ public class FileServiceTest extends BaseDbUnitTest {
when(client.getId()).thenReturn(10L);
String name = "单测文件名";
// 调用
String result = fileService.createFile(name, path, content);
String result = fileService.createFile(name, path, "application/octet-stream", content);
// 断言
assertEquals(result, url);
// 校验数据
@ -90,7 +90,7 @@ public class FileServiceTest extends BaseDbUnitTest {
assertEquals(10L, file.getConfigId());
assertEquals(path, file.getPath());
assertEquals(url, file.getUrl());
assertEquals("jpg", file.getType());
assertEquals("jpg", file.getExtName());
assertEquals(content.length, file.getSize());
}