main #9
@@ -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();
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -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);
|
||||||
}
|
}
|
||||||
}
|
} */
|
||||||
|
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user