1 files changed, 10 insertions, 3 deletions
diff --git a/src/main/java/at/ac/tuwien/sepm/assignment/groupphase/einsatzverwaltung/controller/EmployeeListItemController.java b/src/main/java/at/ac/tuwien/sepm/assignment/groupphase/einsatzverwaltung/controller/EmployeeListItemController.java
index ef0e6fe..02a21d4 100644
--- a/src/main/java/at/ac/tuwien/sepm/assignment/groupphase/einsatzverwaltung/controller/EmployeeListItemController.java
+++ b/src/main/java/at/ac/tuwien/sepm/assignment/groupphase/einsatzverwaltung/controller/EmployeeListItemController.java
@@ -4,6 +4,7 @@ import at.ac.tuwien.sepm.assignment.groupphase.einsatzverwaltung.dto.Employee;
 import at.ac.tuwien.sepm.assignment.groupphase.util.SpringFXMLLoader;
 import at.ac.tuwien.sepm.assignment.groupphase.util.SpringFXMLLoader.FXMLWrapper;
 import java.io.IOException;
+import java.util.function.Consumer;
 import javafx.fxml.FXML;
 import javafx.scene.Node;
 import javafx.scene.control.Label;
@@ -23,17 +24,19 @@ public class EmployeeListItemController {
     @FXML private ImageView imgQualification;
 
     private Node rootElement;
+    private Employee employee;
 
-    private Runnable consumerEditEmployeeClicked;
+    private Consumer<Employee> consumerEditEmployeeClicked;
 
     @FXML
     public void onEditEmployeeClicked() {
         if (consumerEditEmployeeClicked != null) {
-            consumerEditEmployeeClicked.run();
+            consumerEditEmployeeClicked.accept(employee);
         }
     }
 
     private void setData(Employee employee) {
+        this.employee = employee;
         lblName.setText(employee.name());
         lblQualification.setText(employee.educationLevel().name());
         lblPilot.setText(String.format("%s Pilot", employee.isPilot() ? "ist" : "nicht"));
@@ -68,7 +71,11 @@ public class EmployeeListItemController {
         return rootElement;
     }
 
-    public void setConsumerEditEmployeeClicked(Runnable consumerEditEmployeeClicked) {
+    public Employee getEmployee() {
+        return employee;
+    }
+
+    public void setConsumerEditEmployeeClicked(Consumer<Employee> consumerEditEmployeeClicked) {
         this.consumerEditEmployeeClicked = consumerEditEmployeeClicked;
     }
 }