From 95f9639088ab6a5615aa348eb206d14541782000 Mon Sep 17 00:00:00 2001
From: minhee <alsdldlfrl@gmail.com>
Date: 월, 21 2월 2022 11:42:00 +0900
Subject: [PATCH] - 이슈타입 기본값 추가 시 inheritPartners 컬럼도 추가

---
 src/main/java/kr/wisestone/owl/domain/Department.java |   15 ++++++++++++++-
 1 files changed, 14 insertions(+), 1 deletions(-)

diff --git a/src/main/java/kr/wisestone/owl/domain/Department.java b/src/main/java/kr/wisestone/owl/domain/Department.java
index aaf363e..1fad1bd 100644
--- a/src/main/java/kr/wisestone/owl/domain/Department.java
+++ b/src/main/java/kr/wisestone/owl/domain/Department.java
@@ -3,6 +3,7 @@
 import javax.persistence.*;
 import java.io.Serializable;
 import java.util.HashSet;
+import java.util.Iterator;
 import java.util.Set;
 
 @Entity
@@ -48,10 +49,22 @@
 
     public void addProjectRole(ProjectRole projectRole) {
         if (this.projectRoleDepartments == null) {
-            this.projectRoleDepartments = new HashSet<ProjectRoleDepartment>();
+            this.projectRoleDepartments = new HashSet<>();
         }
         ProjectRoleDepartment projectRoleDepartment = new ProjectRoleDepartment(projectRole, this);
 
         this.projectRoleDepartments.add(projectRoleDepartment);
     }
+
+    public void removeProjectRole(ProjectRole projectRole) {
+        Iterator<ProjectRoleDepartment> iterator = this.projectRoleDepartments.iterator();
+
+        while (iterator.hasNext()) {
+            ProjectRoleDepartment projectRoleDepartment = iterator.next();
+            if (projectRole.getId().equals(projectRoleDepartment.getProjectRole().getId())) {
+                this.projectRoleDepartments.remove(projectRoleDepartment);
+                break;
+            }
+        }
+    }
 }

--
Gitblit v1.8.0