Imbus/MovieBookingSystem
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Transaction 2.0

Browse source
This commit is contained in:
Imbus 2024-02-09 09:24:55 +01:00
parent 92441ad91f
commit 4a863bc14e
1 changed files with 41 additions and 21 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

48
src/main/java/datamodel/Database.java
View file

@ -197,14 +197,14 @@ public Show getShowData(String movie, String date) {
} }
public Reservation addReservation(String username, String movie, String date) { public Reservation addReservation(String username, String movie, String date) {
if(!isConnected()) { if (!isConnected()) {
System.err.println("addReservation: no connection to database"); System.err.println("addReservation: no connection to database");
return null; return null;
} }
System.out.println("Adding reservation for " + username + " for " + movie + " on " + date); System.out.println("Adding reservation for " + username + " for " + movie + " on " + date);
try(Statement stmt = conn.createStatement()) { try (Statement stmt = conn.createStatement()) {
conn.setAutoCommit(false); conn.setAutoCommit(false);
Show s = getShowData(movie, date); Show s = getShowData(movie, date);
@ -213,27 +213,46 @@ public Show getShowData(String movie, String date) {
return null; return null;
} }
ResultSet rs = stmt.executeQuery("SELECT * FROM Users WHERE username = '" + username + "'"); try (PreparedStatement userQuery = conn.prepareStatement("SELECT * FROM Users WHERE username = ?");
PreparedStatement showingQuery = conn.prepareStatement("SELECT * FROM Showings JOIN Movies ON Showings.movie_id = Movies.movie_id WHERE name = ? AND show_date = ?");
PreparedStatement reservationInsert = conn.prepareStatement("INSERT INTO Reservations (user_id, showing_id) VALUES (?, ?)", Statement.RETURN_GENERATED_KEYS)) {
userQuery.setString(1, username);
try (ResultSet rs = userQuery.executeQuery()) {
if (rs.next()) { if (rs.next()) {
int user_id = rs.getInt("id"); int user_id = rs.getInt("id");
rs = stmt.executeQuery("SELECT * FROM Showings JOIN Movies ON Showings.movie_id = Movies.movie_id WHERE name = '" + movie + "' AND show_date = '" + date + "'");
if (rs.next()) { showingQuery.setString(1, movie);
System.out.println("Adding reservation"); showingQuery.setString(2, date);
int showing_id = rs.getInt("showing_id"); try (ResultSet showingRs = showingQuery.executeQuery()) {
String theater = rs.getString("theater_id"); if (showingRs.next()) {
stmt.executeUpdate("INSERT INTO Reservations (user_id, showing_id) VALUES (" + user_id + ", " + showing_id + ")"); // Getting the data from the showing query
System.out.println("Reservation added"); int showing_id = showingRs.getInt("showing_id");
String theater = showingRs.getString("theater_id");
// Preparing the reservation insert with the data previously retrieved
reservationInsert.setInt(1, user_id);
reservationInsert.setInt(2, showing_id);
reservationInsert.executeUpdate();
// Committing the transaction and getting the generated keys
conn.commit(); conn.commit();
conn.setAutoCommit(true); conn.setAutoCommit(true);
int bookingId = stmt.getGeneratedKeys().getInt(1);
// Getting the generated keys, if any
try (ResultSet generatedKeys = reservationInsert.getGeneratedKeys()) {
if (generatedKeys.next()) {
int bookingId = generatedKeys.getInt(1);
return new Reservation(bookingId, movie, date, theater); return new Reservation(bookingId, movie, date, theater);
} }
rs.close();
} }
else { }
}
} else {
System.err.println(String.format("addReservation: user %s not found", username)); System.err.println(String.format("addReservation: user %s not found", username));
} }
conn.commit(); }
}
} catch (SQLException e) { } catch (SQLException e) {
try { try {
conn.rollback(); conn.rollback();
@ -245,6 +264,7 @@ public Show getShowData(String movie, String date) {
return null; return null;
} }
public ArrayList<Reservation> getReservations(String username) { public ArrayList<Reservation> getReservations(String username) {
ArrayList<Reservation> reservations = new ArrayList<Reservation>(); ArrayList<Reservation> reservations = new ArrayList<Reservation>();