refactor: 수정시 검증 로직 추가 #8

변경하려는 체중의 petId 검증
myBuddyGuard · Sep 18, 2024 · cdb3ff5 · cdb3ff5
1 parent 3b79c2e
commit cdb3ff5
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/be/src/main/java/buddyguard/mybuddyguard/weight/service/WeightService.java b/be/src/main/java/buddyguard/mybuddyguard/weight/service/WeightService.java
@@ -47,11 +47,13 @@ public WeightResponse getDetailWeightRecord(Long id, Long petId) {
     }
 
     @Transactional
-    public void updateWeightRecord(Long id, WeightUpdateRequest request) {
+    public void updateWeightRecord(Long id, Long petId, WeightUpdateRequest request) {
 
         Weight weight = weightRepository.findById(id)
                 .orElseThrow(RuntimeException::new);
 
+        weight.validateOwnership(petId);
+
         weight.update( // 변경 포인트를 해당 메서드에 모아서 하나로 통일한다.
                 request.recordedAt(),
                 request.weight(),
@@ -64,7 +66,7 @@ public void updateWeightRecord(Long id, WeightUpdateRequest request) {
     public void deleteWeightRecord(Long id) {
 
         Weight weight = weightRepository.findById(id)
-                .orElseThrow(RecordNotFoundException::new); // 적절한 예외 적용하기
+                .orElseThrow(RecordNotFoundException::new);
 
         weightRepository.delete(weight);
     }