From bd78ee158df93958380a6cc2460316fd87ed9713 Mon Sep 17 00:00:00 2001
From: 이민희 <mhlee@maprex.co.kr>
Date: 화, 08 2월 2022 11:39:08 +0900
Subject: [PATCH] - 이슈 임포트 시 IP주소 정규식 표현 체크 - 업체 추가 시 url을 입력하지 않을 경우 중복 체크 하는 오류 수정

---
 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