Refactor: 약 재고 업데이트 api 로직 수정

neulpeum · Mar 12, 2024 · 60f37b8 · 60f37b8
2 parents cfbd797 + b75115a
commit 60f37b8
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 5 deletions.
diff --git a/src/main/java/com/medicare/neulpeum/Repository/DrugRepository.java b/src/main/java/com/medicare/neulpeum/Repository/DrugRepository.java
@@ -11,5 +11,5 @@ public interface DrugRepository extends JpaRepository<DrugInfo, Long> {
     @Override
     List<DrugInfo> findAll();
     List<DrugInfo> findByDrugName(String drugName);
-    boolean existsByDrugName(String drugName);
+    boolean existsById(Long id);
 }
diff --git a/src/main/java/com/medicare/neulpeum/controller/DrugController.java b/src/main/java/com/medicare/neulpeum/controller/DrugController.java
@@ -22,7 +22,7 @@ public ResponseEntity<?> updateDrugInfo(@RequestBody List<DrugRequestDto> drugRe
         try {
             for (DrugRequestDto drugRequestDto : drugRequestDtoList) {
                 // 해당 약물 정보가 이미 존재하는지 확인
-                if (drugService.existsByDrugName(drugRequestDto.getDrugName())) {
+                if (drugService.existsByDrugId(drugRequestDto.getDrugId())) {
                     // 이미 존재하는 경우에는 업데이트 수행
                     drugService.update(drugRequestDto);
                 } else {

diff --git a/src/main/java/com/medicare/neulpeum/service/DrugService.java b/src/main/java/com/medicare/neulpeum/service/DrugService.java
@@ -9,6 +9,6 @@ public interface DrugService {
     void save(DrugRequestDto drugRequestDto);
     List<DrugResponseDto> findAll();
     List<DrugResponseDto> findByDrugName(String drugName);
-    boolean existsByDrugName(String drugName);
+    boolean existsByDrugId(Long id);
     void update(DrugRequestDto drugRequestDto);
 }
diff --git a/src/main/java/com/medicare/neulpeum/service/DrugServiceImpl.java b/src/main/java/com/medicare/neulpeum/service/DrugServiceImpl.java
@@ -56,8 +56,8 @@ public List<DrugResponseDto> findByDrugName(String drugName) {
     }
 
     @Override
-    public boolean existsByDrugName(String drugName) {
-        return drugRepository.existsByDrugName(drugName);
+    public boolean existsByDrugId(Long id) {
+        return drugRepository.existsById(id);
     }
 
     @Override