diff --git a/src/main/java/dgpena/siexample/persistence/Project.java b/src/main/java/dgpena/siexample/persistence/Project.java index 6b877d26c76f94d26282491b6cdc003cadbba875..b57a0d9d7835c9dc6ba09c2b777edb430db8c4e3 100644 --- a/src/main/java/dgpena/siexample/persistence/Project.java +++ b/src/main/java/dgpena/siexample/persistence/Project.java @@ -1,5 +1,6 @@ package dgpena.siexample.persistence; +import java.util.Collection; import java.util.Date; import java.util.HashSet; import java.util.Set; @@ -44,6 +45,22 @@ public class Project { return employees; } + + public void setEmployees(Collection employees) { + // remove my employees that are not in the new list of employees + for (Employee e: this.getEmployees()) { + if (!employees.contains(e)) { + this.removeEmployee(e); + } + } + + // add all the new employees + for (Employee e: employees) { + if (!this.getEmployees().contains(e)) { + this.addEmployee(e, new Date()); + } + } + } public void addEmployee(Employee e, Date startDate) {