Merge pull request #50 from prgrms-be-devcourse/feature/#49

feature - 카테고리, 부품 엔티티 연결
prgrms-be-devcourse · Feb 4, 2025 · 9f38762 · 9f38762
2 parents e14010e + 7b5d822
commit 9f38762
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/backend/src/main/java/com/programmers/pcquotation/domain/categories/entity/Categories.java b/backend/src/main/java/com/programmers/pcquotation/domain/categories/entity/Categories.java
@@ -1,9 +1,16 @@
 package com.programmers.pcquotation.domain.categories.entity;
 
+import java.util.ArrayList;
+import java.util.List;
+
+import com.programmers.pcquotation.domain.items.entity.Items;
+
+import jakarta.persistence.CascadeType;
 import jakarta.persistence.Entity;
 import jakarta.persistence.GeneratedValue;
 import jakarta.persistence.GenerationType;
 import jakarta.persistence.Id;
+import jakarta.persistence.OneToMany;
 import lombok.AccessLevel;
 import lombok.AllArgsConstructor;
 import lombok.Builder;
@@ -21,6 +28,9 @@ public class Categories {
 	private Long id;
 	private String category;
 
+	@OneToMany(mappedBy = "category", cascade = CascadeType.ALL, orphanRemoval = true)
+	private List<Items> items = new ArrayList<>();
+
 	public void updateCategory(String category) {
 		this.category = category;
 	}

diff --git a/backend/src/main/java/com/programmers/pcquotation/domain/items/entity/Items.java b/backend/src/main/java/com/programmers/pcquotation/domain/items/entity/Items.java
@@ -1,9 +1,13 @@
 package com.programmers.pcquotation.domain.items.entity;
 
+import com.programmers.pcquotation.domain.categories.entity.Categories;
+
 import jakarta.persistence.Entity;
 import jakarta.persistence.GeneratedValue;
 import jakarta.persistence.GenerationType;
 import jakarta.persistence.Id;
+import jakarta.persistence.JoinColumn;
+import jakarta.persistence.ManyToOne;
 import lombok.AccessLevel;
 import lombok.AllArgsConstructor;
 import lombok.Builder;
@@ -21,4 +25,8 @@ public class Items {
 	private Long id;
 	private String name;                    // 부품 이름
 	private String imgFilename;
+
+	@ManyToOne
+	@JoinColumn(nullable = false)
+	private Categories category;
 }