From 2341721b26034622386864d0fe3e798f692c5a9e Mon Sep 17 00:00:00 2001
From: 이민희 <mhlee@maprex.co.kr>
Date: 목, 25 11월 2021 15:11:32 +0900
Subject: [PATCH] 프로젝트 담당부서 수정

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

diff --git a/src/main/java/kr/wisestone/owl/domain/Department.java b/src/main/java/kr/wisestone/owl/domain/Department.java
index 8ec34a2..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
@@ -54,4 +55,16 @@
 
         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