MemberAiText에 memberPhoto 추가

woong-nan-chanseong · Nov 11, 2023 · 8e7c523 · 8e7c523
1 parent a156ebd
commit 8e7c523
Show file tree

Hide file tree

Showing 4 changed files with 10 additions and 12 deletions.
diff --git a/src/main/java/com/example/wncsbackend/controller/PapaGoController.java b/src/main/java/com/example/wncsbackend/controller/PapaGoController.java
@@ -3,8 +3,6 @@
 import com.example.wncsbackend.service.PapaGoService;
 import java.io.IOException;
 import lombok.RequiredArgsConstructor;
-import okhttp3.Response;
-import okhttp3.ResponseBody;
 import org.springframework.http.ResponseEntity;
 import org.springframework.web.bind.annotation.GetMapping;
 import org.springframework.web.bind.annotation.RequestParam;

diff --git a/src/main/java/com/example/wncsbackend/domain/MemberAiText/MemberAiText.java b/src/main/java/com/example/wncsbackend/domain/MemberAiText/MemberAiText.java
@@ -1,6 +1,6 @@
 package com.example.wncsbackend.domain.MemberAiText;
 
-import com.example.wncsbackend.domain.Member;
+import com.example.wncsbackend.domain.MemberPhoto.MemberPhoto;
 import com.example.wncsbackend.global.common.BaseTimeEntity;
 import jakarta.persistence.Entity;
 import jakarta.persistence.FetchType;
@@ -28,13 +28,13 @@ public class MemberAiText extends BaseTimeEntity {
     private String lora;
 
     @ManyToOne(fetch = FetchType.LAZY)
-    private Member member;
+    private MemberPhoto memberPhoto;
 
-    public MemberAiText(String seed, String prompt, String lora, Member member) {
+    public MemberAiText(String seed, String prompt, String lora, MemberPhoto memberPhoto) {
         this.seed = seed;
         this.prompt = prompt;
         this.lora = lora;
-        this.member = member;
+        this.memberPhoto = memberPhoto;
     }
 
 }
diff --git a/src/main/java/com/example/wncsbackend/domain/MemberAiText/dto/MemberAiTextRequestDto.java b/src/main/java/com/example/wncsbackend/domain/MemberAiText/dto/MemberAiTextRequestDto.java
@@ -13,6 +13,6 @@ public static class MemberAiTextInfo {
         private String seed;
         private String prompt;
         private String lora;
-        private Long memberId;
+        private Long memberPhotoId;
     }
 }
diff --git a/src/main/java/com/example/wncsbackend/service/MemberAiTextService.java b/src/main/java/com/example/wncsbackend/service/MemberAiTextService.java
@@ -4,9 +4,9 @@
 import com.example.wncsbackend.domain.Member;
 import com.example.wncsbackend.domain.MemberAiText.MemberAiText;
 import com.example.wncsbackend.domain.MemberAiText.dto.MemberAiTextRequestDto.MemberAiTextInfo;
+import com.example.wncsbackend.domain.MemberPhoto.MemberPhoto;
 import com.example.wncsbackend.repository.MemberAiTextRepository;
-import com.example.wncsbackend.repository.MemberRepository;
-import java.util.Optional;
+import com.example.wncsbackend.repository.MemberPhotoRepository;
 import lombok.RequiredArgsConstructor;
 import org.springframework.stereotype.Service;
 
@@ -15,12 +15,12 @@
 public class MemberAiTextService {
 
     private final MemberAiTextRepository memberAiTextRepository;
-    private final MemberRepository memberRepository;
+    private final MemberPhotoRepository memberPhotoRepository;
 
     public String insertAiText(MemberAiTextInfo memberAiTextInfo) {
-        Member member = memberRepository.findById(memberAiTextInfo.getMemberId()).orElseThrow();
+        MemberPhoto memberPhoto = memberPhotoRepository.findById(memberAiTextInfo.getMemberPhotoId()).orElseThrow();
         MemberAiText memberAiText = new MemberAiText(memberAiTextInfo.getSeed(), memberAiTextInfo.getPrompt(),
-                memberAiTextInfo.getLora(), member);
+                memberAiTextInfo.getLora(), memberPhoto);
         memberAiTextRepository.save(memberAiText);
         return "Ai 텍스트 저장 완료";
     }