[PBS-84][Anh Dao] fix: update amenity quatity

Hoang-Nguyen-Huy · Oct 31, 2024 · 53b4df0 · 53b4df0
1 parent d8b02e4
commit 53b4df0
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/main/java/com/swp/PodBookingSystem/service/OrderDetailAmenityService.java b/src/main/java/com/swp/PodBookingSystem/service/OrderDetailAmenityService.java
@@ -75,7 +75,11 @@ public void createOrderDetailAmenity(OrderDetailAmenityRequest request) {
     }
 
     public void updateAmenityQuantityAfterCreateODA(OrderDetailAmenity orderDetailAmenity) {
-        Amenity amenity = orderDetailAmenity.getAmenity();
+        Optional<Amenity> amenityOptional = amenityRepository.findById(orderDetailAmenity.getAmenity().getId());
+        if(amenityOptional.isEmpty()) {
+            throw new RuntimeException("Amenity not found");
+        }
+        Amenity amenity = amenityOptional.get();
         if(amenity.getQuantity() < orderDetailAmenity.getQuantity()) {
             throw new RuntimeException("Not enough quantity");
         }