问题描述
在我们上传文件时,会遇到这些情况:
- 如果上传两次文件,可能文件的文件名不同,但是上传的文件是相同的
- 上传了文件两次
如果不进行判断,那么同样的一个文件可以会存入多次,导致存储空间被占用。
解决思路
用户上传文件,可以在数据库中有多条记录,但是在存储空间中只允许有一个文件被存放进去。
选用的办法是设置一个md5的值。当文件进行上传的过程中,现将文件上传获取到一个md5值,然后将改文件上传的md5去数据库中找寻记录,如果有相同的md5的值,那么就将刚上传的文件删掉,同时将相同md5的值的文件url写到该条记录中。
代码实现
首先数据库设计如下:
需要一个md5的字段。
application.yml中配置一个文件路径:
后端上传文件接口:
先获取到这个文件路径:
然后上传接口:
/**
* 文件上传接口
* @Param file 前端传递过来的文件
* @Return Result封装返回
* @throws IOException
* */
@PostMapping("/upload")
@ApiOperation("文件上传")
public Result<?> upload(@RequestParam MultipartFile file) throws IOException {
//获取文件上传时的文件名
String originalFilename = file.getOriginalFilename();
//获取文件类型
String type = FileUtil.extName(originalFilename);
//获取文件大小
long size = file.getSize();
//定义一个文件唯一的标识码(作为在本地存储的文件名)
String uuid = IdUtil.fastSimpleUUID();
String fileUUID = uuid + StrUtil.DOT + type;
//就是文件对象的路径名 例子:D:/studyboot/qingge/images/uuid.jpg
File uploadFile = new File(fileUploadPath + fileUUID);
//获取父目录
File parentFile = uploadFile.getParentFile();
//判断配置的文件目录是否存在,若不存在就创建一个新的文件目录
if (parentFile.exists()){
parentFile.mkdirs();
}
//当文件存在的时候再去获取文件的md5
String md5;
String url;
//把获取到的文件存储到磁盘目录去
file.transferTo(uploadFile);
//储存进去之后就能获取到文件的md5
md5 = SecureUtil.md5(uploadFile);
//从数据库查询文件的md5是否存在
Files dbFile = getFileByMd5(md5);
if (dbFile!=null){
url = dbFile.getUrl();
//由于文件已经存在,所以删除刚才上传的重复文件
uploadFile.delete();
} else {
//若数据库不存在重复文件,则不删除刚才上传的文件
url = "http://localhost:9090/file/"+ fileUUID;
}
//存储数据库中
Files saveFile = new Files();
saveFile.setName(originalFilename);
saveFile.setSize(size);
saveFile.setType(type);
saveFile.setUrl(url);
saveFile.setMd5(md5);
fileMapper.insert(saveFile);
return Result.success(url);
}
测试
我们用postman上传三张图片
查看数据库中的记录
可以发现其中的md5的值相同
那么我们去本地文件储存看看:
其实只有一张图片,那么这样会节省图片存储的空间。