2martens/dis
1
0
Fork 0
Code Issues 1 Pull Requests Projects Releases Wiki Activity

Added method to load an estate from database

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2017-04-19 12:27:21 +02:00
parent 0fe8f78067
commit 6380420b5a
1 changed files with 103 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

108
02/src/de/dis2017/data/db/ORM.java
View File

@ -1,7 +1,9 @@
package de.dis2017.data.db;
import de.dis2017.data.Apartment;
import de.dis2017.data.Estate;
import de.dis2017.data.EstateAgent;
import de.dis2017.data.House;
import org.jetbrains.annotations.Nullable;
import java.sql.*;
@ -115,13 +117,109 @@ public class ORM {
return agents;
}
/**
* Loads the estate with the given ID from database and returns the corresponding object.
*
* @param ID the id of the estate to load
* @return the Estate or null if there is no such object
*/
public Estate getEstate(int ID) {
if (_estates.containsKey(ID)) {
return _estates.get(ID);
}
String selectSQLHouse = "SELECT * FROM HOUSE LEFT JOIN ESTATE ON HOUSE.ID = ESTATE.ID " +
"WHERE HOUSE.ID = ?";
String selectSQLApartment = "SELECT * FROM APARTMENT LEFT JOIN ESTATE ON APARTMENT.ID = ESTATE.ID " +
"WHERE APARTMENT.ID = ?";
String countHouse = "SELECT COUNT(ID) AS count FROM HOUSE WHERE ID = ?";
String countApartment = "SELECT COUNT(ID) AS count FROM APARTMENT WHERE ID = ?";
try {
// try house first
PreparedStatement preparedStatementCount = _connection.prepareStatement(countHouse);
preparedStatementCount.setInt(1, ID);
ResultSet rs = preparedStatementCount.executeQuery();
rs.next();
int count = rs.getInt("count");
String type = "None";
if (count == 0) {
// try apartment next
preparedStatementCount = _connection.prepareStatement(countApartment);
preparedStatementCount.setInt(1, ID);
rs = preparedStatementCount.executeQuery();
rs.next();
count = rs.getInt("count");
if (count == 1) {
type = "Apartment";
}
}
else {
type = "House";
}
rs.close();
preparedStatementCount.close();
PreparedStatement pstmt;
Estate estate;
switch (type) {
case "House":
pstmt = _connection.prepareStatement(selectSQLHouse);
pstmt.setInt(1, ID);
rs = pstmt.executeQuery();
estate = new House();
break;
case "Apartment":
pstmt = _connection.prepareStatement(selectSQLApartment);
pstmt.setInt(1, ID);
rs = pstmt.executeQuery();
estate = new Apartment();
break;
default:
return null;
}
if (rs.next()) {
estate.setId(ID);
estate.setCity(rs.getString("city"));
estate.setPostalCode(rs.getString("postalCode"));
estate.setStreet(rs.getString("street"));
estate.setStreetNumber(rs.getInt("streetNumber"));
estate.setSquareArea(rs.getInt("squareArea"));
estate.setAgent(rs.getInt("agent"));
if (estate instanceof House) {
((House) estate).setFloors(rs.getInt("floors"));
((House) estate).setGarden(rs.getBoolean("garden"));
((House) estate).setPrice(rs.getInt("price"));
}
if (estate instanceof Apartment) {
((Apartment) estate).setRent(rs.getInt("rent"));
((Apartment) estate).setFloor(rs.getInt("floor"));
((Apartment) estate).setRooms(rs.getInt("rooms"));
((Apartment) estate).setBalcony(rs.getBoolean("balcony"));
((Apartment) estate).setBuiltinKitchen(rs.getBoolean("builtInKitchen"));
}
_estates.put(ID, estate);
}
rs.close();
pstmt.close();
} catch (SQLException e) {
e.printStackTrace();
}
return null;
}
/**
* Loads the estate agent with the given ID from database and returns the corresponding object.
*
* @param ID the ID of the agent to load
* @return the EstateAgent or null if there is no such agent
*/
public EstateAgent get(int ID) {
public EstateAgent getAgent(int ID) {
if (_agents.containsKey(ID)) {
return _agents.get(ID);
}
@ -132,7 +230,7 @@ public class ORM {
PreparedStatement pstmt = _connection.prepareStatement(selectSQL);
pstmt.setInt(1, ID);
return get(pstmt);
return getAgent(pstmt);
} catch (SQLException e) {
e.printStackTrace();
}
@ -146,7 +244,7 @@ public class ORM {
* @param username the username of the estate agent
* @return the EstateAgent or null if there is no such agent
*/
public EstateAgent get(String username) {
public EstateAgent getAgent(String username) {
if (_agentsUsername.containsKey(username)) {
return _agentsUsername.get(username);
}
@ -157,7 +255,7 @@ public class ORM {
PreparedStatement pstmt = _connection.prepareStatement(selectSQL);
pstmt.setString(1, username);
return get(pstmt);
return getAgent(pstmt);
} catch (SQLException e) {
e.printStackTrace();
}
@ -172,7 +270,7 @@ public class ORM {
* @return the EstateAgent or null
*/
@Nullable
private EstateAgent get(PreparedStatement pstmt)
private EstateAgent getAgent(PreparedStatement pstmt)
{
try {
// execute query