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 getSoldHouses() { + List 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 getRentedApartments() { + List 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. *