[Bug] Entity 연관관계 수정

src/main/java/com/school/mohitto/domain/Diagnosis.java

@@ -51,10 +51,12 @@ public class Diagnosis extends BaseTimeEntity{
     @OneToMany(mappedBy = "diagnosis", cascade = CascadeType.ALL, orphanRemoval = true)
     private List<DiagnosisMood> diagnosisMoods = new ArrayList<>();
 
-
     @OneToMany(mappedBy = "diagnosis", cascade = CascadeType.ALL, orphanRemoval = true)
     private List<CreatedImage> createdImages = new ArrayList<>();
 
+    @OneToMany(mappedBy = "diagnosis", cascade = CascadeType.ALL, orphanRemoval = true)
+    private List<Hair> hairs = new ArrayList<>();
+
     public static Diagnosis create(User user) {
         Diagnosis diagnosis = new Diagnosis();
         diagnosis.user = user;
@@ -96,9 +98,9 @@ public void addDiagnosisImpression(DiagnosisMood diagnosisMood) {
         diagnosisMood.setDiagnosis(this);
     }
 
-
-
-
-
+    public void addHairs(List<Hair> hairs){
+        this.hairs.addAll(hairs);
+        hairs.forEach(hair -> hair.setDiagnosis(this));
+    }
 
 }

src/main/java/com/school/mohitto/domain/Hair.java

@@ -29,11 +29,19 @@ public class Hair {
     @Column(name = "styling_method")
     private String method;
 
+    @ManyToOne(fetch = FetchType.LAZY)
+    @JoinColumn(name = "model_image_id")
+    private ModelImage modelImage;
 
-    @OneToMany(mappedBy = "hair", cascade = CascadeType.ALL, orphanRemoval = true)
-    private List<ModelImage> modelImages = new ArrayList<>();
+    @ManyToOne(fetch = FetchType.LAZY)
+    @JoinColumn(name = "diagnosis_id")
+    private Diagnosis diagnosis;
 
     @OneToMany(mappedBy = "hair", cascade = CascadeType.ALL)
     private List<UserHair> userHairs = new ArrayList<>();
     // 생성자, 빌더, 비즈니스 메서드
+
+    public void setDiagnosis(Diagnosis diagnosis) {
+        this.diagnosis = diagnosis;
+    }
 }

src/main/java/com/school/mohitto/domain/ModelImage.java

@@ -5,6 +5,8 @@
 import lombok.Getter;
 import lombok.NoArgsConstructor;
 
+import java.util.List;
+
 @Entity
 @Table(name = "model_images")
 @Getter
@@ -16,10 +18,9 @@ public class ModelImage {
     @Column(name = "model_image_id")
     private Long id;
 
-    @ManyToOne(fetch = FetchType.LAZY)
-    @JoinColumn(name = "hair_id")
-    private Hair hair;
-
     @Column(name = "url", length = 255)
     private String uploadImageUrl;
+
+    @OneToMany(mappedBy = "modelImage", cascade = CascadeType.ALL, orphanRemoval = true)
+    private List<Hair> hairs;
 }