From d54de4856ccac362c64fb149dd6ff6e7f808a55c Mon Sep 17 00:00:00 2001
From: Jim Martens <github@2martens.de>
Date: Mon, 24 Apr 2017 17:15:34 +0200
Subject: [PATCH] Added ability to get sold houses and rented apartments

Signed-off-by: Jim Martens <github@2martens.de>
---
 02/src/de/dis2017/data/db/ORM.java | 50 +++++++++++++++++++++++++++++-
 1 file changed, 49 insertions(+), 1 deletion(-)

diff --git a/02/src/de/dis2017/data/db/ORM.java b/02/src/de/dis2017/data/db/ORM.java
index 325dd6d..95f5a43 100644
--- a/02/src/de/dis2017/data/db/ORM.java
+++ b/02/src/de/dis2017/data/db/ORM.java
@@ -76,10 +76,58 @@ public class ORM {
         } catch (SQLException e) {
             e.printStackTrace();
         }
-        
+    
         return objects;
     }
     
+    /**
+     * Returns a list with the IDs of sold houses.
+     *
+     * @return a list of houseIDs
+     */
+    public List<Integer> getSoldHouses() {
+        List<Integer> soldHouses = new ArrayList<>();
+        
+        try {
+            String            selectSQL = "SELECT house FROM SALES";
+            PreparedStatement pstmt     = _connection.prepareStatement(selectSQL);
+            ResultSet         rs        = pstmt.executeQuery();
+            while (rs.next()) {
+                soldHouses.add(rs.getInt("house"));
+            }
+            rs.close();
+            pstmt.close();
+        } catch (SQLException e) {
+            e.printStackTrace();
+        }
+        
+        return soldHouses;
+    }
+    
+    /**
+     * Returns a list with the IDs of rented apartments.
+     *
+     * @return a list of apartmentIDs
+     */
+    public List<Integer> getRentedApartments() {
+        List<Integer> rentedApartments = new ArrayList<>();
+        
+        try {
+            String            selectSQL = "SELECT apartment FROM RENTALS";
+            PreparedStatement pstmt     = _connection.prepareStatement(selectSQL);
+            ResultSet         rs        = pstmt.executeQuery();
+            while (rs.next()) {
+                rentedApartments.add(rs.getInt("apartment"));
+            }
+            rs.close();
+            pstmt.close();
+        } catch (SQLException e) {
+            e.printStackTrace();
+        }
+        
+        return rentedApartments;
+    }
+    
     /**
      * Process a select all query for estates.
      *