ak/file
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

保存分享文件bug修改

Browse Source
This commit is contained in:
马超 2021-04-10 19:57:10 +08:00
parent 50c1ba4125
commit 73efd5f858
3 changed files with 22 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
file-web/src/main/java/com/qiwenshare/file/api/IUserFileService.java
View File

@ -22,6 +22,6 @@ public interface IUserFileService extends IService<UserFile> {
List<UserFile> selectFileListLikeRightFilePath(String filePath, long userId); List<UserFile> selectFileListLikeRightFilePath(String filePath, long userId);
List<UserFile> selectFilePathTreeByUserId(Long userId); List<UserFile> selectFilePathTreeByUserId(Long userId);
void deleteUserFile(Long userFileId, Long sessionUserId); void deleteUserFile(Long userFileId, Long sessionUserId);
UserFile repeatUserFileDeal(UserFile userFile); String getRepeatFileName(UserFile userFile, String savefilePath);
} }

10
file-web/src/main/java/com/qiwenshare/file/controller/ShareController.java
View File

@ -116,6 +116,9 @@ public class ShareController {
List<UserFile> saveUserFileList = new ArrayList<>(); List<UserFile> saveUserFileList = new ArrayList<>();
for (ShareFile shareFile : fileList) { for (ShareFile shareFile : fileList) {
UserFile userFile = userFileService.getById(shareFile.getUserFileId()); UserFile userFile = userFileService.getById(shareFile.getUserFileId());
String fileName = userFile.getFileName();
String savefileName = userFileService.getRepeatFileName(userFile, savefilePath);
if (userFile.getIsDir() == 1) { if (userFile.getIsDir() == 1) {
List<UserFile> userfileList = userFileService.selectFileListLikeRightFilePath(userFile.getFilePath() + userFile.getFileName(), userFile.getUserId()); List<UserFile> userfileList = userFileService.selectFileListLikeRightFilePath(userFile.getFilePath() + userFile.getFileName(), userFile.getUserId());
log.info("查询文件列表：" + JSON.toJSONString(userfileList)); log.info("查询文件列表：" + JSON.toJSONString(userfileList));
@ -123,8 +126,7 @@ public class ShareController {
userfileList.forEach(p->{ userfileList.forEach(p->{
p.setUserFileId(null); p.setUserFileId(null);
p.setUserId(userId); p.setUserId(userId);
p.setFilePath(p.getFilePath().replaceFirst(filePath, savefilePath)); p.setFilePath(p.getFilePath().replaceFirst(filePath + fileName, savefilePath + savefileName));
p = userFileService.repeatUserFileDeal(p);
saveUserFileList.add(p); saveUserFileList.add(p);
log.info("当前文件：" + JSON.toJSONString(p)); log.info("当前文件：" + JSON.toJSONString(p));
if (p.getIsDir() == 0) { if (p.getIsDir() == 0) {
@ -134,12 +136,12 @@ public class ShareController {
} else { } else {
fileService.increaseFilePointCount(userFile.getFileId()); fileService.increaseFilePointCount(userFile.getFileId());
} }
userFile.setUserFileId(null); userFile.setUserFileId(null);
userFile.setUserId(userId); userFile.setUserId(userId);
userFile.setFilePath(savefilePath); userFile.setFilePath(savefilePath);
userFile = userFileService.repeatUserFileDeal(userFile); userFile.setFileName(savefileName);
saveUserFileList.add(userFile); saveUserFileList.add(userFile);
} }
log.info("----------" + JSON.toJSONString(saveUserFileList)); log.info("----------" + JSON.toJSONString(saveUserFileList));
userFileService.saveBatch(saveUserFileList); userFileService.saveBatch(saveUserFileList);

26
file-web/src/main/java/com/qiwenshare/file/service/UserFileService.java
View File

@ -217,38 +217,42 @@ public class UserFileService extends ServiceImpl<UserFileMapper, UserFile> impl
} }
@Override @Override
public UserFile repeatUserFileDeal(UserFile userFile) { public String getRepeatFileName(UserFile userFile, String savefilePath) {
String fileName = userFile.getFileName(); String fileName = userFile.getFileName();
String filePath = userFile.getFilePath();
String extendName = userFile.getExtendName(); String extendName = userFile.getExtendName();
Integer deleteFlag = userFile.getDeleteFlag(); Integer deleteFlag = userFile.getDeleteFlag();
Long userId = userFile.getUserId(); Long userId = userFile.getUserId();
LambdaQueryWrapper<UserFile> lambdaQueryWrapper = new LambdaQueryWrapper<>(); LambdaQueryWrapper<UserFile> lambdaQueryWrapper = new LambdaQueryWrapper<>();
lambdaQueryWrapper.eq(UserFile::getFilePath, filePath) lambdaQueryWrapper.eq(UserFile::getFilePath, savefilePath)
.eq(UserFile::getExtendName, extendName)
.eq(UserFile::getDeleteFlag, deleteFlag) .eq(UserFile::getDeleteFlag, deleteFlag)
.eq(UserFile::getUserId, userId) .eq(UserFile::getUserId, userId)
.eq(UserFile::getFileName, fileName); .eq(UserFile::getFileName, fileName);
if (userFile.getIsDir() == 0) {
lambdaQueryWrapper.eq(UserFile::getExtendName, extendName);
}
List<UserFile> list = userFileMapper.selectList(lambdaQueryWrapper); List<UserFile> list = userFileMapper.selectList(lambdaQueryWrapper);
if (list == null) { if (list == null) {
return userFile; return fileName;
} }
if (list.isEmpty()) { if (list.isEmpty()) {
return userFile; return fileName;
} }
int i = 0; int i = 0;
while (list != null && !list.isEmpty()) { while (list != null && !list.isEmpty()) {
i++; i++;
lambdaQueryWrapper.eq(UserFile::getFilePath, filePath) LambdaQueryWrapper<UserFile> lambdaQueryWrapper1 = new LambdaQueryWrapper<>();
.eq(UserFile::getExtendName, extendName) lambdaQueryWrapper1.eq(UserFile::getFilePath, savefilePath)
.eq(UserFile::getDeleteFlag, deleteFlag) .eq(UserFile::getDeleteFlag, deleteFlag)
.eq(UserFile::getUserId, userId) .eq(UserFile::getUserId, userId)
.eq(UserFile::getFileName, fileName + "(" + i + ")"); .eq(UserFile::getFileName, fileName + "(" + i + ")");
list = userFileMapper.selectList(lambdaQueryWrapper); if (userFile.getIsDir() == 0) {
lambdaQueryWrapper1.eq(UserFile::getExtendName, extendName);
} }
userFile.setFileName(fileName + "(" + i + ")"); list = userFileMapper.selectList(lambdaQueryWrapper1);
return userFile; }
return fileName + "(" + i + ")";
} }