From 375870fbb83f1adbd7d5afd0e5fffbc25dcac774 Mon Sep 17 00:00:00 2001
From: dgpena <dgpena@uvigo.es>
Date: Thu, 30 Nov 2017 19:52:37 +0100
Subject: [PATCH] Adds setEmployees method

---
 .../dgpena/siexample/persistence/Project.java   | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

diff --git a/src/main/java/dgpena/siexample/persistence/Project.java b/src/main/java/dgpena/siexample/persistence/Project.java
index 6b877d2..b57a0d9 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<Employee> 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) {
 
-- 
2.18.1