This commit is contained in:
@@ -6,9 +6,12 @@ import lombok.extern.slf4j.Slf4j;
|
||||
import org.springframework.stereotype.Component;
|
||||
import ru.soune.nocopy.dto.BaseRequest;
|
||||
import ru.soune.nocopy.dto.BaseResponse;
|
||||
import ru.soune.nocopy.dto.MessageCode;
|
||||
import ru.soune.nocopy.dto.violation.ViolationStatisticsRequest;
|
||||
import ru.soune.nocopy.dto.violation.ViolationStatisticsResponse;
|
||||
import ru.soune.nocopy.entity.user.AuthToken;
|
||||
import ru.soune.nocopy.exception.FileEntityNotFoundException;
|
||||
import ru.soune.nocopy.exception.UserNotHavePermission;
|
||||
import ru.soune.nocopy.repository.AuthTokenRepository;
|
||||
import ru.soune.nocopy.service.violation.ViolationService;
|
||||
|
||||
@@ -55,16 +58,36 @@ public class ViolationStatisticsHandler implements RequestHandler {
|
||||
}
|
||||
|
||||
Long userId = tokenOptional.orElseThrow().getUser().getId();
|
||||
ViolationStatisticsResponse statistics;
|
||||
|
||||
try {
|
||||
statistics = violationService.getViolationStatistics(
|
||||
userId,
|
||||
statRequest.getFileId(),
|
||||
startDate,
|
||||
endDate);
|
||||
} catch (UserNotHavePermission e) {
|
||||
return BaseResponse.builder()
|
||||
.msgId(request.getMsgId())
|
||||
.messageBody("userId: " + userId)
|
||||
.messageCode(MessageCode.USER_NOT_HAD_PERMISSION.getCode())
|
||||
.messageDesc(MessageCode.USER_NOT_HAD_PERMISSION.getDescription())
|
||||
.build();
|
||||
} catch (FileEntityNotFoundException e) {
|
||||
return BaseResponse.builder()
|
||||
.msgId(request.getMsgId())
|
||||
.messageBody("fileId: " + statRequest.getFileId())
|
||||
.messageDesc(MessageCode.FILE_NOT_FOUND.getDescription())
|
||||
.messageCode(MessageCode.FILE_NOT_FOUND.getCode())
|
||||
.build();
|
||||
}
|
||||
|
||||
ViolationStatisticsResponse statistics = violationService.getViolationStatistics(
|
||||
userId,
|
||||
statRequest.getFileId(),
|
||||
startDate,
|
||||
endDate);
|
||||
|
||||
return BaseResponse.builder()
|
||||
.msgId(request.getMsgId())
|
||||
.messageBody(statistics)
|
||||
.messageCode(MessageCode.SUCCESS.getCode())
|
||||
.messageDesc(MessageCode.SUCCESS.getDescription())
|
||||
.build();
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user