main #9

Closed
student-20690 wants to merge 26 commits from (deleted):main into main
6 changed files with 113 additions and 18 deletions
Showing only changes of commit 639b1c6c24 - Show all commits

View File

@@ -2,9 +2,13 @@ package com.example.nto.controller;
import com.example.nto.entity.Booking; import com.example.nto.entity.Booking;
import com.example.nto.service.BookingService; import com.example.nto.service.BookingService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.*; import org.springframework.web.bind.annotation.*;
import java.time.LocalDate; import java.time.LocalDate;
import java.util.List;
import java.util.Map;
/** /**
* TODO: ДОРАБОТАТЬ в рамках задания * TODO: ДОРАБОТАТЬ в рамках задания
@@ -12,9 +16,22 @@ import java.time.LocalDate;
* МОЖНО: Добавлять методы, аннотации, зависимости * МОЖНО: Добавлять методы, аннотации, зависимости
* НЕЛЬЗЯ: Изменять название класса и пакета * НЕЛЬЗЯ: Изменять название класса и пакета
*/ */
//@RestController @RestController
//@RequestMapping("/api") @RequestMapping("/api")
//public class BookingController { public class BookingController {
//
// private final BookingService bookingService;
//}
@Autowired
public BookingController(BookingService bookingService) {
this.bookingService = bookingService;
}
@GetMapping("/{code}/booking")
public ResponseEntity<Map<String, List<Booking.AvailablePlaceDto>>> getAvailablePlaces(
@PathVariable String code
) {
Map<String, List<Booking.AvailablePlaceDto>> response = bookingService.getAvailablePlaces(code);
return ResponseEntity.ok(response);
}
}

View File

@@ -37,4 +37,12 @@ public class Booking {
@JoinColumn(name = "employee_id") @JoinColumn(name = "employee_id")
@JsonBackReference @JsonBackReference
private Employee employee; private Employee employee;
@Data
public static class AvailablePlaceDto {
private long id;
private String place;
}
} }

View File

@@ -6,6 +6,7 @@ import org.springframework.stereotype.Repository;
import org.springframework.stereotype.Service; import org.springframework.stereotype.Service;
import java.sql.Date; import java.sql.Date;
import java.time.LocalDate;
import java.util.List; import java.util.List;
import java.util.Optional; import java.util.Optional;
@@ -18,7 +19,7 @@ import java.util.Optional;
@Service @Service
@Repository @Repository
public interface BookingRepository extends JpaRepository<Booking, Long> { public interface BookingRepository extends JpaRepository<Booking, Long> {
Optional<Booking> findById(long id); List<Booking> findByDateAndEmployeeByCode(List<LocalDate> dates, String code);
List<Booking> findByEmployee_Code(String code); Optional<Booking> findById(String code);
} }

View File

@@ -13,5 +13,4 @@ import java.util.Optional;
* НЕЛЬЗЯ: Изменять название класса и пакета * НЕЛЬЗЯ: Изменять название класса и пакета
*/ */
public interface PlaceRepository extends JpaRepository<Place, Long> { public interface PlaceRepository extends JpaRepository<Place, Long> {
Optional<Place> findPlaceById(long id);
} }

View File

@@ -4,6 +4,7 @@ import com.example.nto.entity.Booking;
import com.example.nto.entity.Employee; import com.example.nto.entity.Employee;
import java.util.List; import java.util.List;
import java.util.Map;
import java.util.Optional; import java.util.Optional;
/** /**
@@ -14,6 +15,7 @@ import java.util.Optional;
*/ */
public interface BookingService { public interface BookingService {
List<Booking> getAll(); List<Booking> getAll();
Optional<Booking> getById(long id); Optional<Booking> getBookingById(String code);
Map<String, List<Booking.AvailablePlaceDto>> getAvailablePlaces(String code);
} }

View File

@@ -1,13 +1,19 @@
package com.example.nto.service.impl; package com.example.nto.service.impl;
import com.example.nto.entity.Booking; import com.example.nto.entity.Booking;
import com.example.nto.entity.Employee;
import com.example.nto.entity.Place;
import com.example.nto.repository.BookingRepository; import com.example.nto.repository.BookingRepository;
import com.example.nto.repository.EmployeeRepository;
import com.example.nto.repository.PlaceRepository;
import com.example.nto.service.BookingService; import com.example.nto.service.BookingService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component; import org.springframework.stereotype.Component;
import org.springframework.stereotype.Service; import org.springframework.stereotype.Service;
import java.util.List; import java.time.LocalDate;
import java.util.Optional; import java.util.*;
import java.util.stream.Collectors;
/** /**
* TODO: ДОРАБОТАТЬ в рамках задания * TODO: ДОРАБОТАТЬ в рамках задания
@@ -20,10 +26,8 @@ import java.util.Optional;
public class BookingServiceImpl implements BookingService { public class BookingServiceImpl implements BookingService {
private final BookingRepository bookingRepository; private final BookingRepository bookingRepository;
private final PlaceRepository placeRepository;
public BookingServiceImpl(BookingRepository bookingRepository) { private final EmployeeRepository employeeRepository;
this.bookingRepository = bookingRepository;
}
@Override @Override
public List<Booking> getAll() { public List<Booking> getAll() {
@@ -31,8 +35,72 @@ public class BookingServiceImpl implements BookingService {
} }
@Override @Override
public Optional<Booking> getById(long id) { public Optional<Booking> getBookingById(String code) {
return bookingRepository.findById(id); return bookingRepository.findById(code);
} }
@Autowired
public BookingServiceImpl(BookingRepository bookingRepository, PlaceRepository placeRepository,
EmployeeRepository employeeRepository) {
this.bookingRepository = bookingRepository;
this.placeRepository = placeRepository;
this.employeeRepository = employeeRepository;
}
public Map<String, List<Booking.AvailablePlaceDto>> getAvailablePlaces(String employeeCode) {
// 1. Получаем сотрудника (для контекста)
Employee employee = employeeRepository.findByCode(employeeCode)
.orElseThrow(() -> new RuntimeException("Employee not found"));
// 2. Определяем диапазон дат: сегодня + 3 дня
LocalDate today = LocalDate.now();
List<LocalDate> targetDates = List.of(
today,
today.plusDays(1),
today.plusDays(2),
today.plusDays(3)
);
// 3. Получаем все места из БД
List<Place> allPlaces = placeRepository.findAll();
// 4. Находим все бронирования на целевые даты
List<Booking> bookings = bookingRepository.findByDateAndEmployeeByCode(targetDates, employeeCode);
// 5. Группируем бронирования по датам
Map<LocalDate, List<Booking>> bookingsByDate = bookings.stream()
.collect(Collectors.groupingBy(Booking::getDate));
// 6. Формируем итоговый ответ
Map<String, List<Booking.AvailablePlaceDto>> result = new HashMap<>();
for (LocalDate date : targetDates) {
String dateKey = date.toString(); // Формат: "yyyy-MM-dd"
// Берём бронирования на эту дату
List<Booking> dayBookings = bookingsByDate.getOrDefault(date, List.of());
// Собираем ID занятых мест
Set<Long> occupiedPlaceIds = dayBookings.stream()
.map(booking -> booking.getPlace().getId())
.collect(Collectors.toSet());
// Фильтруем свободные места
List<Booking.AvailablePlaceDto> available = allPlaces.stream()
.filter(place -> !occupiedPlaceIds.contains(place.getId()))
.map(place -> {
Booking.AvailablePlaceDto dto = new Booking.AvailablePlaceDto();
dto.setId(place.getId());
dto.setPlace(place.getPlace());
return dto;
})
.collect(Collectors.toList());
result.put(dateKey, available);
}
return result;
}
} }