[feat] 저장 취소 시 storecount 감소 메서드 추가

TripComeTrue · Jan 26, 2024 · 7958cc2 · 7958cc2
1 parent 0f08345
commit 7958cc2
Show file tree

Hide file tree

Showing 4 changed files with 33 additions and 0 deletions.
diff --git a/src/main/java/com/haejwo/tripcometrue/domain/city/entity/City.java b/src/main/java/com/haejwo/tripcometrue/domain/city/entity/City.java
@@ -67,5 +67,11 @@ public void incrementStoreCount() {
             this.storeCount++;
         }
     }
+
+    public void decrementStoreCount() {
+        if(this.storeCount > 0) {
+            this.storeCount--;
+        }
+    }
 }
 
diff --git a/src/main/java/com/haejwo/tripcometrue/domain/place/entity/Place.java b/src/main/java/com/haejwo/tripcometrue/domain/place/entity/Place.java
@@ -90,6 +90,12 @@ public void incrementStoreCount() {
         }
     }
 
+    public void decrementStoreCount() {
+        if(this.storedCount > 0) {
+            this.storedCount--;
+        }
+    }
+
     public void incrementReviewCount() {
         if(this.reviewCount == null) {
             this.reviewCount = 1;

diff --git a/src/main/java/com/haejwo/tripcometrue/domain/store/service/StoreService.java b/src/main/java/com/haejwo/tripcometrue/domain/store/service/StoreService.java
@@ -109,6 +109,11 @@ public void unstoreCity(PrincipalDetails principalDetails, Long cityId) {
         Long memberId = principalDetails.getMember().getId();
         CityStore cityStore = cityStoreRepository.findByMemberIdAndCityId(memberId, cityId)
                 .orElseThrow(() -> new StoreNotFoundException(ErrorCode.STORE_NOT_FOUND));
+
+        City city = cityRepository.findById(cityId)
+            .orElseThrow();
+        city.decrementStoreCount();
+
         cityStoreRepository.delete(cityStore);
     }
 
@@ -117,6 +122,11 @@ public void unstorePlace(PrincipalDetails principalDetails, Long placeId) {
         Long memberId = principalDetails.getMember().getId();
         PlaceStore placeStore = placeStoreRepository.findByMemberIdAndPlaceId(memberId, placeId)
                 .orElseThrow(() -> new StoreNotFoundException(ErrorCode.STORE_NOT_FOUND));
+
+        Place place = placeRepository.findById(placeId)
+                .orElseThrow(()-> new PlaceNotFoundException());
+        place.decrementStoreCount();
+
         placeStoreRepository.delete(placeStore);
     }
 
@@ -125,6 +135,11 @@ public void unstoreTripRecord(PrincipalDetails principalDetails, Long tripRecord
         Long memberId = principalDetails.getMember().getId();
         TripRecordStore tripRecordStore = tripRecordStoreRepository.findByMemberIdAndTripRecordId(memberId, tripRecordId)
                 .orElseThrow(() -> new StoreNotFoundException(ErrorCode.STORE_NOT_FOUND));
+
+        TripRecord tripRecord = tripRecordRepository.findById(tripRecordId)
+                .orElseThrow(()-> new TripRecordNotFoundException());
+        tripRecord.decrementStoreCount();
+
         tripRecordStoreRepository.delete(tripRecordStore);
     }
 

diff --git a/src/main/java/com/haejwo/tripcometrue/domain/triprecord/entity/TripRecord.java b/src/main/java/com/haejwo/tripcometrue/domain/triprecord/entity/TripRecord.java
@@ -114,6 +114,12 @@ public void incrementStoreCount() {
         }
     }
 
+    public void decrementStoreCount() {
+        if(this.storeCount > 0) {
+            this.storeCount--;
+        }
+    }
+
     public void incrementReviewCount() {
         if(this.reviewCount == null) {
             this.reviewCount = 1;