From b7b618b5ef673eb95ba6c0688a393a4d00647748 Mon Sep 17 00:00:00 2001 From: 이민희 <mhlee@maprex.co.kr> Date: 화, 07 12월 2021 19:54:10 +0900 Subject: [PATCH] 연관 이슈 추가 기능 클릭 이벤트 미작동 오류 해결 --- 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