fix(文件上传):文件上传路径标准化处理

This commit is contained in:
MAC 2022-04-27 21:16:35 +08:00
parent 91c45b76c0
commit 4116acf0dc

View File

@ -1,9 +1,12 @@
package com.qiwenshare.file.controller; package com.qiwenshare.file.controller;
import co.elastic.clients.elasticsearch.ElasticsearchClient; import co.elastic.clients.elasticsearch.ElasticsearchClient;
import com.qiwenshare.file.api.IShareFileService;
import com.qiwenshare.file.component.FileDealComp; import com.qiwenshare.file.component.FileDealComp;
import com.qiwenshare.file.domain.ShareFile;
import com.qiwenshare.file.domain.UserFile; import com.qiwenshare.file.domain.UserFile;
import com.qiwenshare.file.io.QiwenFile; import com.qiwenshare.file.io.QiwenFile;
import com.qiwenshare.file.service.ShareFileService;
import com.qiwenshare.file.service.UserFileService; import com.qiwenshare.file.service.UserFileService;
import lombok.extern.slf4j.Slf4j; import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired; import org.springframework.beans.factory.annotation.Autowired;
@ -21,6 +24,8 @@ public class TaskController {
UserFileService userFileService; UserFileService userFileService;
@Resource @Resource
FileDealComp fileDealComp; FileDealComp fileDealComp;
@Resource
IShareFileService shareFileService;
@Autowired @Autowired
private ElasticsearchClient elasticsearchClient; private ElasticsearchClient elasticsearchClient;
@ -45,10 +50,19 @@ public class TaskController {
public void updateFilePath() { public void updateFilePath() {
List<UserFile> list = userFileService.list(); List<UserFile> list = userFileService.list();
for (UserFile userFile : list) { for (UserFile userFile : list) {
QiwenFile qiwenFile = new QiwenFile(userFile.getFilePath(), true);
String path = QiwenFile.formatPath(userFile.getFilePath()); String path = QiwenFile.formatPath(userFile.getFilePath());
userFile.setFilePath(path); userFile.setFilePath(path);
userFileService.updateById(userFile); userFileService.updateById(userFile);
} }
} }
@Scheduled(fixedRate = Long.MAX_VALUE)
public void updateShareFilePath() {
List<ShareFile> list = shareFileService.list();
for (ShareFile shareFile : list) {
String path = QiwenFile.formatPath(shareFile.getShareFilePath());
shareFile.setShareFilePath(path);
shareFileService.updateById(shareFile);
}
}
} }