This commit is contained in:
@@ -7,8 +7,12 @@ import org.springframework.stereotype.Component;
|
|||||||
import ru.soune.nocopy.dto.*;
|
import ru.soune.nocopy.dto.*;
|
||||||
import ru.soune.nocopy.dto.file.*;
|
import ru.soune.nocopy.dto.file.*;
|
||||||
import ru.soune.nocopy.entity.AuthToken;
|
import ru.soune.nocopy.entity.AuthToken;
|
||||||
|
import ru.soune.nocopy.entity.file.FileEntity;
|
||||||
|
import ru.soune.nocopy.entity.file.FileStatus;
|
||||||
|
import ru.soune.nocopy.exception.FileEntityNotFoundException;
|
||||||
import ru.soune.nocopy.exception.NotFoundAuthToken;
|
import ru.soune.nocopy.exception.NotFoundAuthToken;
|
||||||
import ru.soune.nocopy.repository.AuthTokenRepository;
|
import ru.soune.nocopy.repository.AuthTokenRepository;
|
||||||
|
import ru.soune.nocopy.repository.FileEntityRepository;
|
||||||
import ru.soune.nocopy.service.file.FileEntityService;
|
import ru.soune.nocopy.service.file.FileEntityService;
|
||||||
|
|
||||||
import java.util.Arrays;
|
import java.util.Arrays;
|
||||||
@@ -18,11 +22,14 @@ import java.util.List;
|
|||||||
@Component
|
@Component
|
||||||
@RequiredArgsConstructor
|
@RequiredArgsConstructor
|
||||||
public class FileEntityHandler implements RequestHandler {
|
public class FileEntityHandler implements RequestHandler {
|
||||||
|
|
||||||
private final FileEntityService fileEntityService;
|
private final FileEntityService fileEntityService;
|
||||||
|
|
||||||
private final AuthTokenRepository authTokenRepository;
|
private final AuthTokenRepository authTokenRepository;
|
||||||
|
|
||||||
private final ObjectMapper objectMapper;
|
private final ObjectMapper objectMapper;
|
||||||
|
|
||||||
|
private final FileEntityRepository fileEntityRepository;
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public BaseResponse handle(BaseRequest request) {
|
public BaseResponse handle(BaseRequest request) {
|
||||||
try {
|
try {
|
||||||
@@ -236,7 +243,9 @@ public class FileEntityHandler implements RequestHandler {
|
|||||||
private BaseResponse handleDeleteFile(BaseRequest request, FileEntityRequest fileRequest) {
|
private BaseResponse handleDeleteFile(BaseRequest request, FileEntityRequest fileRequest) {
|
||||||
try {
|
try {
|
||||||
Long userId = getUserIdFromToken(fileRequest.getToken());
|
Long userId = getUserIdFromToken(fileRequest.getToken());
|
||||||
FileEntityResponse fileInfo = fileEntityService.getById(fileRequest.getFileId(), request.getVersion());
|
String fileId = fileRequest.getFileId();
|
||||||
|
FileEntityResponse fileInfo = fileEntityService.getById(fileId, request.getVersion());
|
||||||
|
DeleteFileResponse response;
|
||||||
|
|
||||||
if (!fileInfo.getUserId().equals(userId)) {
|
if (!fileInfo.getUserId().equals(userId)) {
|
||||||
return new BaseResponse(request.getMsgId(),
|
return new BaseResponse(request.getMsgId(),
|
||||||
@@ -245,18 +254,27 @@ public class FileEntityHandler implements RequestHandler {
|
|||||||
null);
|
null);
|
||||||
}
|
}
|
||||||
|
|
||||||
fileEntityService.markAsDeleted(fileRequest.getFileId());
|
FileEntity fileEntity = fileEntityRepository.findById(fileId)
|
||||||
|
.orElseThrow(() -> new FileEntityNotFoundException(fileId));
|
||||||
|
|
||||||
DeleteFileResponse response = DeleteFileResponse.builder()
|
if (fileEntity.getStatus().equals(FileStatus.DELETED)) {
|
||||||
|
fileEntityService.deleteFromDisk(fileEntity);
|
||||||
|
response = DeleteFileResponse.builder()
|
||||||
|
.fileId(fileRequest.getFileId())
|
||||||
|
.message("File deleted from disk")
|
||||||
|
.build();
|
||||||
|
} else {
|
||||||
|
fileEntityService.markAsDeleted(fileEntity);
|
||||||
|
response = DeleteFileResponse.builder()
|
||||||
.fileId(fileRequest.getFileId())
|
.fileId(fileRequest.getFileId())
|
||||||
.message("File marked as deleted")
|
.message("File marked as deleted")
|
||||||
.build();
|
.build();
|
||||||
|
}
|
||||||
|
|
||||||
return new BaseResponse(request.getMsgId(),
|
return new BaseResponse(request.getMsgId(),
|
||||||
MessageCode.SUCCESS.getCode(),
|
MessageCode.SUCCESS.getCode(),
|
||||||
"File marked as deleted",
|
"File marked as deleted",
|
||||||
response);
|
response);
|
||||||
|
|
||||||
} catch (NotFoundAuthToken e) {
|
} catch (NotFoundAuthToken e) {
|
||||||
return new BaseResponse(request.getMsgId(),
|
return new BaseResponse(request.getMsgId(),
|
||||||
MessageCode.INVALID_TOKEN.getCode(),
|
MessageCode.INVALID_TOKEN.getCode(),
|
||||||
|
|||||||
@@ -17,6 +17,7 @@ import java.nio.file.Files;
|
|||||||
import java.nio.file.Path;
|
import java.nio.file.Path;
|
||||||
import java.nio.file.Paths;
|
import java.nio.file.Paths;
|
||||||
import java.time.LocalDateTime;
|
import java.time.LocalDateTime;
|
||||||
|
import java.time.ZoneOffset;
|
||||||
import java.util.List;
|
import java.util.List;
|
||||||
import java.util.stream.Collectors;
|
import java.util.stream.Collectors;
|
||||||
|
|
||||||
@@ -167,15 +168,24 @@ public class FileEntityService {
|
|||||||
}
|
}
|
||||||
|
|
||||||
@Transactional
|
@Transactional
|
||||||
public void markAsDeleted(String fileId) {
|
public void markAsDeleted(FileEntity fileEntity) {
|
||||||
FileEntity fileEntity = fileEntityRepository.findById(fileId)
|
|
||||||
.orElseThrow(() -> new FileEntityNotFoundException(fileId));
|
|
||||||
|
|
||||||
fileEntity.setStatus(FileStatus.DELETED);
|
fileEntity.setStatus(FileStatus.DELETED);
|
||||||
fileEntity.setUpdatedAt(LocalDateTime.now());
|
fileEntity.setUpdatedAt(LocalDateTime.now());
|
||||||
fileEntityRepository.save(fileEntity);
|
|
||||||
|
|
||||||
log.info("FileEntity marked as deleted: {}", fileId);
|
fileEntityRepository.save(fileEntity);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Transactional
|
||||||
|
public boolean deleteFromDisk(FileEntity fileEntity) throws IOException {
|
||||||
|
Path path = Paths.get(fileEntity.getFilePath());
|
||||||
|
|
||||||
|
if (!Files.exists(path)) {
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
Files.delete(path);
|
||||||
|
|
||||||
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
@Transactional(readOnly = true)
|
@Transactional(readOnly = true)
|
||||||
|
|||||||
Reference in New Issue
Block a user