OWL ITS + 탐지시스템(인터넷 진흥원)
박지현
2022-03-03 25e69beeac0e3939ba1873806710a18a5ec3091e
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;
            }
        }
    }
}