Feat: ProvidedDrugInfo 엔티티에 약 사용 History 저장 기능 구현

neulpeum · May 6, 2024 · cc5c220 · cc5c220
2 parents a323a95 + ef8d48d
commit cc5c220
Show file tree

Hide file tree

Showing 5 changed files with 20 additions and 33 deletions.
diff --git a/src/main/java/com/medicare/neulpeum/dto/DrugUpdateRequestDto.java b/src/main/java/com/medicare/neulpeum/dto/DrugUpdateRequestDto.java
@@ -1,5 +1,6 @@
 package com.medicare.neulpeum.dto;
 
+import com.medicare.neulpeum.domain.entity.ConsultContentInfo;
 import lombok.Getter;
 import lombok.NoArgsConstructor;
 import lombok.Setter;
@@ -8,6 +9,7 @@
 @Setter
 @NoArgsConstructor
 public class DrugUpdateRequestDto {
+    private ConsultContentInfo consultId;
     private String drugName;
     private int usedAmount;
 

diff --git a/src/main/java/com/medicare/neulpeum/dto/ProvidedDrugRequestDto.java b/src/main/java/com/medicare/neulpeum/dto/ProvidedDrugRequestDto.java
diff --git a/src/main/java/com/medicare/neulpeum/service/DrugServiceImpl.java b/src/main/java/com/medicare/neulpeum/service/DrugServiceImpl.java
@@ -1,7 +1,9 @@
 package com.medicare.neulpeum.service;
 
 import com.medicare.neulpeum.Repository.DrugRepository;
+import com.medicare.neulpeum.Repository.ProvidedDrugRepository;
 import com.medicare.neulpeum.domain.entity.DrugInfo;
+import com.medicare.neulpeum.domain.entity.ProvidedDrugInfo;
 import com.medicare.neulpeum.dto.*;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.beans.factory.annotation.Autowired;
@@ -19,6 +21,8 @@ public class DrugServiceImpl implements DrugService{
 
     @Autowired
     DrugRepository drugRepository;
+    @Autowired
+    ProvidedDrugRepository providedDrugRepository;
 
     @Override
     public void save(DrugRequestDto drugReq) {
@@ -101,6 +105,12 @@ public void updateUsedDrug(List<DrugUpdateRequestDto> drugUpdateRequestDtoList)
                         drugRepository.save(drug);
 
                         //ProvidedDrugInfo 테이블에 저장
+                        ProvidedDrugInfo providedDrugInfo = ProvidedDrugInfo.builder()
+                                .drugId(drug)
+                                .consultId(updateRequestDto.getConsultId())
+                                .providedAmount((long) remainingAmount)
+                                .build();
+                        providedDrugRepository.save(providedDrugInfo);
 
                         break;
                     } else {
@@ -109,6 +119,14 @@ public void updateUsedDrug(List<DrugUpdateRequestDto> drugUpdateRequestDtoList)
                         remainingAmount -= usableAmount;
                         drug.setUsableAmount(0);
                         drugRepository.save(drug);
+
+                        //ProvidedDrugInfo 테이블에 저장
+                        ProvidedDrugInfo providedDrugInfo = ProvidedDrugInfo.builder()
+                                .drugId(drug)
+                                .consultId(updateRequestDto.getConsultId())
+                                .providedAmount((long) remainingAmount)
+                                .build();
+                        providedDrugRepository.save(providedDrugInfo);
                     }
                 }
             }

diff --git a/src/main/java/com/medicare/neulpeum/service/ProvidedDrugService.java b/src/main/java/com/medicare/neulpeum/service/ProvidedDrugService.java
diff --git a/src/main/java/com/medicare/neulpeum/service/ProvidedDrugServiceImpl.java b/src/main/java/com/medicare/neulpeum/service/ProvidedDrugServiceImpl.java