From aa170291e9da8b943828d3ec4e71b914d55319ba Mon Sep 17 00:00:00 2001
From: Dominic Rogetzer <e1627756@student.tuwien.ac.at>
Date: Wed, 23 May 2018 23:36:05 +0200
Subject: Prepare vehicle-dao fix by adding RegistrationDatabaseDAO.list
 [#25963]

---
 .../dao/RegistrationDatabaseDAO.java               | 43 +++++++++++++++++++++-
 1 file changed, 42 insertions(+), 1 deletion(-)

diff --git a/src/main/java/at/ac/tuwien/sepm/assignment/groupphase/einsatzverwaltung/dao/RegistrationDatabaseDAO.java b/src/main/java/at/ac/tuwien/sepm/assignment/groupphase/einsatzverwaltung/dao/RegistrationDatabaseDAO.java
index 5c00447..13f2c0f 100644
--- a/src/main/java/at/ac/tuwien/sepm/assignment/groupphase/einsatzverwaltung/dao/RegistrationDatabaseDAO.java
+++ b/src/main/java/at/ac/tuwien/sepm/assignment/groupphase/einsatzverwaltung/dao/RegistrationDatabaseDAO.java
@@ -1,5 +1,6 @@
 package at.ac.tuwien.sepm.assignment.groupphase.einsatzverwaltung.dao;
 
+import at.ac.tuwien.sepm.assignment.groupphase.einsatzverwaltung.dto.Employee;
 import at.ac.tuwien.sepm.assignment.groupphase.einsatzverwaltung.dto.Registration;
 import at.ac.tuwien.sepm.assignment.groupphase.exception.ElementNotFoundException;
 import at.ac.tuwien.sepm.assignment.groupphase.exception.PersistenceException;
@@ -10,7 +11,9 @@ import java.sql.ResultSet;
 import java.sql.SQLException;
 import java.sql.Statement;
 import java.sql.Timestamp;
+import java.util.ArrayList;
 import java.util.HashSet;
+import java.util.List;
 import java.util.Set;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
@@ -26,20 +29,27 @@ public class RegistrationDatabaseDAO implements RegistrationDAO {
             "INSERT INTO Registration (vehicleId, employeeId, start, end, active) VALUES (?,?,?,?,?);";
     private static final String UPDATE_VEHICLE =
             "UPDATE Vehicle SET status = 'FREI_WACHE' WHERE id = ?;";
+    private static final String FETCH_REGISTRATIONS =
+            "SELECT * FROM Registration WHERE vehicleId = ?";
 
     private PreparedStatement addRegistration;
     private PreparedStatement updateVehicle;
+    private PreparedStatement fetchRegistrations;
 
     private Connection connection;
+    private EmployeeDAO employeePersistence;
 
     @Autowired
-    public RegistrationDatabaseDAO(JDBCConnectionManager connectionManager)
+    public RegistrationDatabaseDAO(
+            JDBCConnectionManager connectionManager, EmployeeDAO employeePersistence)
             throws PersistenceException {
+        this.employeePersistence = employeePersistence;
         try {
             connection = connectionManager.getConnection();
             addRegistration =
                     connection.prepareStatement(ADD_REGISTRATION, Statement.RETURN_GENERATED_KEYS);
             updateVehicle = connection.prepareStatement(UPDATE_VEHICLE);
+            fetchRegistrations = connection.prepareStatement(FETCH_REGISTRATIONS);
         } catch (SQLException e) {
             LOG.error("Could not get connection or preparation of statement failed");
             throw new PersistenceException(e);
@@ -103,4 +113,35 @@ public class RegistrationDatabaseDAO implements RegistrationDAO {
     public void remove(long id) throws ElementNotFoundException, PersistenceException {
         throw new UnsupportedOperationException();
     }
+
+    public List<Registration> list(long vehicleId) throws PersistenceException {
+        List<Registration> registrationList = new ArrayList<>();
+        try {
+            fetchRegistrations.setLong(1, vehicleId);
+            ResultSet rs = fetchRegistrations.executeQuery();
+            while (rs.next()) {
+                long employeeId = rs.getLong("employeeId");
+                // TODO: replace the following with employeePersistence.get once implemented
+                Employee emp =
+                        employeePersistence
+                                .list()
+                                .stream()
+                                .filter(employee -> employee.id() == employeeId)
+                                .findAny()
+                                .orElse(null);
+                Registration registration =
+                        Registration.builder()
+                                .id(rs.getLong("id"))
+                                .start(rs.getTimestamp("start").toInstant())
+                                .end(rs.getTimestamp("end").toInstant())
+                                .employee(emp)
+                                .build();
+                registrationList.add(registration);
+            }
+        } catch (SQLException e) {
+            throw new PersistenceException(e);
+        }
+
+        return registrationList;
+    }
 }
-- 
cgit v1.2.3-70-g09d2