diff --git a/src/main/java/com/example/nto/controller/EmployeeController.java b/src/main/java/com/example/nto/controller/EmployeeController.java index 78bf916..a94aacb 100644 --- a/src/main/java/com/example/nto/controller/EmployeeController.java +++ b/src/main/java/com/example/nto/controller/EmployeeController.java @@ -52,13 +52,7 @@ public class EmployeeController { } Employee employee = employeeService.getEmployeeByCode(code); List bookings = bookingRepository.findByEmployeeId(employee.getId()); - Map> bookingMap = new LinkedHashMap<>(); - - for (Booking b : bookings) { - bookingMap.put( - b.getDate().toString(), - Map.of("id", b.getId(), "place", b.getPlace().getPlace())); - } + Map>> bookingMap = bookings.stream().collect(Collectors.groupingBy(b -> b.getDate().toString(),LinkedHashMap::new,Collectors.mapping(b -> Map.of("id", b.getId(),"place", b.getPlace().getPlace()), Collectors.toList()))); Map response = new LinkedHashMap<>(); response.put("name", employee.getName()); response.put("photoUrl", employee.getPhotoUrl()); @@ -73,5 +67,4 @@ public class EmployeeController { .body("Unexpected error"); } } - -} \ No newline at end of file +}