feat : 메모 감정어 분석 추가

Happy-HOBAK · Jul 12, 2024 · 5c2159a · 5c2159a
1 parent f6e091e
commit 5c2159a
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/src/main/java/com/hobak/happinessql/domain/record/application/RecordCreateService.java b/src/main/java/com/hobak/happinessql/domain/record/application/RecordCreateService.java
@@ -4,10 +4,12 @@
 import com.hobak.happinessql.domain.activity.exception.ActivityNotFoundException;
 import com.hobak.happinessql.domain.activity.repository.ActivityRepository;
 import com.hobak.happinessql.domain.record.converter.RecordConverter;
+import com.hobak.happinessql.domain.record.domain.Analysis;
 import com.hobak.happinessql.domain.record.domain.Location;
 import com.hobak.happinessql.domain.record.domain.Record;
 import com.hobak.happinessql.domain.record.domain.RecordImg;
 import com.hobak.happinessql.domain.record.dto.RecordCreateRequestDto;
+import com.hobak.happinessql.domain.record.repository.AnalysisRepository;
 import com.hobak.happinessql.domain.record.repository.LocationRepository;
 import com.hobak.happinessql.domain.record.repository.RecordImgRepository;
 import com.hobak.happinessql.domain.record.repository.RecordRepository;
@@ -19,6 +21,9 @@
 import org.springframework.stereotype.Service;
 import org.springframework.web.multipart.MultipartFile;
 
+import java.util.List;
+import java.util.Map;
+
 @Service
 @RequiredArgsConstructor
 public class RecordCreateService {
@@ -30,6 +35,8 @@ public class RecordCreateService {
 
     private final AwsS3Service awsS3Service;
     private final UserFindService userFindService;
+    private final AnalysisRepository analysisRepository;
+    private final SentimentAnalyzeService sentimentAnalyzeService;
 
 
     @Transactional
@@ -54,6 +61,16 @@ public Long createRecord(Long userId, RecordCreateRequestDto requestDto, Multipa
                     .build();
             recordImgRepository.save(recordImg);
         }
+        if (requestDto.getMemo() != null && !requestDto.getMemo().isEmpty()) {
+            Map<String, List<String>> analyzeResult = sentimentAnalyzeService.getAnalyzeResult(requestDto.getMemo());
+            Analysis analysis = Analysis.builder()
+                    .record(newRecord)
+                    .positiveSentiments(analyzeResult.get("positive"))
+                    .negativeSentiments(analyzeResult.get("negative"))
+                    .build();
+
+            analysisRepository.save(analysis);
+        }
 
         return newRecord.getRecordId();
     }