#14 [Fix] 북마크 삭제 시 발생하는 NaN 오류 수정

MeongPanNyaengPan-MPNP · Jul 31, 2022 · da10a85 · da10a85
1 parent 0e999aa
commit da10a85
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/src/main/java/com/mpnp/baechelin/store/domain/Store.java b/src/main/java/com/mpnp/baechelin/store/domain/Store.java
@@ -104,17 +104,22 @@ public Store updateBookmarkCount() {
         this.bookMarkCount = this.getBookmarkList().size();
         return this;
     }
-    public void removeReview(Review review){
+
+    public void removeReview(Review review) {
         this.reviewList.remove(review);
     }
 
+    public void removeBookmark(Bookmark bookmark) {
+        this.bookmarkList.remove(bookmark);
+    }
+
     public Store updatePointAvg() {
         this.reviewCount = reviewList.size();
         double totalPoint = 0.0;
         for (Review review : reviewList) {
             totalPoint += review.getPoint();
         }
-        this.pointAvg = Double.parseDouble(String.format("%.1f", totalPoint / reviewList.size()));
+        this.pointAvg = reviewCount == 0 ? 0 : Double.parseDouble(String.format("%.1f", totalPoint / reviewList.size()));
         return this;
     }