main #10

Closed
student-20690 wants to merge 26 commits from (deleted):main into main
2 changed files with 14 additions and 26 deletions
Showing only changes of commit 9e52213068 - Show all commits

View File

@@ -58,15 +58,20 @@ public class BookingController {
) { ) {
try { try {
Booking booking = bookingService.createBooking(code, request.getDate(), request.getPlaceId()); Booking booking = bookingService.createBooking(code, request.getDate(), request.getPlaceId());
System.out.println("OK");
return ResponseEntity.status(HttpStatus.CREATED).build(); return ResponseEntity.status(HttpStatus.CREATED).build();
} catch (EmployeeNotFoundException e) { } catch (EmployeeNotFoundException e) {
System.out.println("NOT");
return ResponseEntity.status(HttpStatus.UNAUTHORIZED).build(); return ResponseEntity.status(HttpStatus.UNAUTHORIZED).build();
} catch (BookingConflictException e) { } catch (BookingConflictException e) {
System.out.println("NOT1");
return ResponseEntity.status(HttpStatus.CONFLICT).build(); return ResponseEntity.status(HttpStatus.CONFLICT).build();
} catch (InvalidBookingException e) { } catch (InvalidBookingException e) {
System.out.println("NOT2");
return ResponseEntity.status(HttpStatus.BAD_REQUEST).build(); return ResponseEntity.status(HttpStatus.BAD_REQUEST).build();
} catch (Exception e) { } catch (Exception e) {
System.out.println("NOT3");
return ResponseEntity.status(HttpStatus.INTERNAL_SERVER_ERROR).build(); return ResponseEntity.status(HttpStatus.INTERNAL_SERVER_ERROR).build();
} }
} }

View File

@@ -54,12 +54,11 @@ public class BookingServiceImpl implements BookingService {
public boolean isEmployeeExists(String code) { public boolean isEmployeeExists(String code) {
return employeeRepository.findByCode(code).isPresent(); return employeeRepository.findByCode(code).isPresent();
} }
public Map<String, List<AvailablePlaceDto>> getAvailablePlaces(String employeeCode) { @Override
// 1. Получаем сотрудника (для контекста) public Map<String, List<Booking.AvailablePlaceDto>> getAvailablePlaces(String employeeCode) {
Employee employee = employeeRepository.findByCode(employeeCode) Employee employee = employeeRepository.findByCode(employeeCode)
.orElseThrow(() -> new RuntimeException("Employee not found")); .orElseThrow(() -> new EmployeeNotFoundException("Unauthorized"));
// 2. Определяем диапазон дат: сегодня + 3 дня
LocalDate today = LocalDate.now(); LocalDate today = LocalDate.now();
List<LocalDate> targetDates = List.of( List<LocalDate> targetDates = List.of(
today, today,
@@ -68,41 +67,28 @@ public class BookingServiceImpl implements BookingService {
today.plusDays(3) today.plusDays(3)
); );
// 3. Получаем все места из БД
List<Place> allPlaces = placeRepository.findAll(); List<Place> allPlaces = placeRepository.findAll();
// 4. Находим все бронирования на целевые даты
List<Booking> bookings = bookingRepository.findByDateIn(targetDates); List<Booking> bookings = bookingRepository.findByDateIn(targetDates);
// 5. Группируем бронирования по датам
Map<LocalDate, List<Booking>> bookingsByDate = bookings.stream() Map<LocalDate, List<Booking>> bookingsByDate = bookings.stream()
.collect(Collectors.groupingBy(Booking::getDate)); .collect(Collectors.groupingBy(Booking::getDate));
Map<String, List<Booking.AvailablePlaceDto>> result = new LinkedHashMap<>(); // Используем LinkedHashMap для сохранения порядка
// 6. Формируем итоговый ответ
Map<String, List<AvailablePlaceDto>> result = new HashMap<>();
for (LocalDate date : targetDates) { for (LocalDate date : targetDates) {
String dateKey = date.toString(); // Формат: "yyyy-MM-dd" String dateKey = date.toString();
// Берём бронирования на эту дату
List<Booking> dayBookings = bookingsByDate.getOrDefault(date, List.of()); List<Booking> dayBookings = bookingsByDate.getOrDefault(date, List.of());
// Собираем ID занятых мест
Set<Long> occupiedPlaceIds = dayBookings.stream() Set<Long> occupiedPlaceIds = dayBookings.stream()
.map(booking -> booking.getPlace().getId()) .map(booking -> booking.getPlace().getId())
.collect(Collectors.toSet()); .collect(Collectors.toSet());
List<Booking.AvailablePlaceDto> available = allPlaces.stream()
// Фильтруем свободные места
List<AvailablePlaceDto> available = allPlaces.stream()
.filter(place -> !occupiedPlaceIds.contains(place.getId())) .filter(place -> !occupiedPlaceIds.contains(place.getId()))
.map(place -> { .map(place -> {
AvailablePlaceDto dto = new AvailablePlaceDto(); Booking.AvailablePlaceDto dto = new Booking.AvailablePlaceDto();
dto.setId(place.getId()); dto.setId(place.getId());
dto.setPlace(place.getPlace()); dto.setPlace(place.getPlace());
return dto; return dto;
}) })
.collect(Collectors.toList()); .collect(Collectors.toList());
result.put(dateKey, available); result.put(dateKey, available);
} }
@@ -125,10 +111,7 @@ public class BookingServiceImpl implements BookingService {
// 4. Проверяем, не занято ли место (409 если занято) // 4. Проверяем, не занято ли место (409 если занято)
checkPlaceAvailability(date, placeId); checkPlaceAvailability(date, placeId);
// 5. Проверяем, нет ли уже брони у сотрудника на эту дату (409 если есть)
checkEmployeeBookingConflict(employeeCode, date);
// 6. Создаем и сохраняем бронирование
Booking booking = Booking.builder() Booking booking = Booking.builder()
.date(date) .date(date)
.place(place) .place(place)
@@ -162,12 +145,12 @@ public class BookingServiceImpl implements BookingService {
} }
} }
private void checkEmployeeBookingConflict(String employeeCode, LocalDate date) { /* private void checkEmployeeBookingConflict(String employeeCode, LocalDate date) {
List<Booking> employeeBookingsOnDate = bookingRepository.findByDateAndEmployee_Code(date, employeeCode); List<Booking> employeeBookingsOnDate = bookingRepository.findByDateAndEmployee_Code(date, employeeCode);
if (!employeeBookingsOnDate.isEmpty()) { if (!employeeBookingsOnDate.isEmpty()) {
throw new BookingConflictException("Employee already has a booking on " + date); throw new BookingConflictException("Employee already has a booking on " + date);
} }
} } */
} }