From 07f4f6ad9592628ae6e0e77fdd4492b20af098fa Mon Sep 17 00:00:00 2001
From: 이민희 <mhlee@maprex.co.kr>
Date: 월, 17 1월 2022 17:18:14 +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