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