Refactor. baseprice 사용 로직 AccommodationEntity의 메서드 사용

CodeSquad-Airbnb-Team07 · Jun 25, 2024 · aebbb9c · aebbb9c
1 parent f19b303
commit aebbb9c
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/BE/src/main/java/team07/airbnb/controller/ProductController.java b/BE/src/main/java/team07/airbnb/controller/ProductController.java
@@ -50,7 +50,7 @@ public void createProduct(@RequestBody ProductCreateRequest request) {
     }
 
     @Tag(name = "Host")
-    @Operation(summary = "상품 생성")
+    @Operation(summary = "기간 중 다중 상품 생성")
     @PostMapping("/range")
     @ResponseStatus(CREATED)
     public void createProductRange(@RequestBody ProductRangeCreateRequest request) {

diff --git a/BE/src/main/java/team07/airbnb/entity/AccommodationEntity.java b/BE/src/main/java/team07/airbnb/entity/AccommodationEntity.java
@@ -86,8 +86,9 @@ public void addPicture(String url) {
         pictures.add(new Pictures(this, url));
     }
 
-    public void addProduct(LocalDate date, int price) {
-        products.add(ProductEntity.ofOpen(this, date, price));
+    public void addProduct(LocalDate date, Integer price) {
+        if(price != null) products.add(ProductEntity.ofOpen(this, date, price));
+        else addProduct(date);
     }
 
     public void addProduct(LocalDate date) {

diff --git a/BE/src/main/java/team07/airbnb/service/product/ProductService.java b/BE/src/main/java/team07/airbnb/service/product/ProductService.java
@@ -191,8 +191,7 @@ public void createRangeProduct(Long accommodationId, LocalDate startDate, LocalD
         Period between = Period.between(startDate, endDate);
         for (int i = 0; i <= between.getDays(); i++) {
             LocalDate nowDate = startDate.plusDays(i);
-            int price = requestPrice == null ? accommodation.getBasePricePerDay() : requestPrice;
-            accommodation.addProduct(nowDate, price);
+            accommodation.addProduct(nowDate, requestPrice);
         }
     }