From 879d31207ddb082357334877876879a8a2fb2551 Mon Sep 17 00:00:00 2001
From: jhjang <jhjang@maprex.co.kr>
Date: 목, 13 1월 2022 17:14:19 +0900
Subject: [PATCH] - api 스펙 변경 - 이슈 수정시 워크플로우에 있는 상태 이외에는 에러 처리 되도록 수정
---
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