From dc9b09a4ee390db471ba8b7dedc375a9c7e72ad3 Mon Sep 17 00:00:00 2001
From: 이민희 <mhlee@maprex.co.kr>
Date: 월, 13 12월 2021 15:56:49 +0900
Subject: [PATCH] - 연관이슈 검색 시 다른 이슈 유형(다른 프로젝트)의 이슈도 검색 가능 하도록 수정 - 부서 삭제 코드 약간 수정

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

diff --git a/src/main/java/kr/wisestone/owl/domain/Department.java b/src/main/java/kr/wisestone/owl/domain/Department.java
index 4a09dbf..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
@@ -17,7 +18,7 @@
     private String departmentDescription;
 
     @OneToMany(mappedBy = "department", cascade = {CascadeType.ALL}, orphanRemoval = true)
-    private Set<ProjectRoleUser> projectRoleDepartments = new HashSet<>();
+    private Set<ProjectRoleDepartment> projectRoleDepartments = new HashSet<ProjectRoleDepartment>();
 
     public Department() {
     }
@@ -46,12 +47,24 @@
         this.departmentDescription = departmentDescription;
     }
 
-    /*public void addProjectRole(ProjectRole projectRole) {
+    public void addProjectRole(ProjectRole projectRole) {
         if (this.projectRoleDepartments == null) {
             this.projectRoleDepartments = new HashSet<>();
         }
-        ProjectRoleUser projectRoleUser = new ProjectRoleUser(projectRole, this);
+        ProjectRoleDepartment projectRoleDepartment = new ProjectRoleDepartment(projectRole, this);
 
-        this.projectRoleDepartments.add(projectRoleUser);
-    }*/
+        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