From a1487f4d4f0d0fc2a205be530b704f4b1d455fd4 Mon Sep 17 00:00:00 2001
From: Jim Martens <github@2martens.de>
Date: Mon, 24 Apr 2017 16:43:10 +0200
Subject: [PATCH] Added ability to persist contracts

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

diff --git a/02/src/de/dis2017/data/db/ORM.java b/02/src/de/dis2017/data/db/ORM.java
index 1475892..325dd6d 100644
--- a/02/src/de/dis2017/data/db/ORM.java
+++ b/02/src/de/dis2017/data/db/ORM.java
@@ -814,6 +814,94 @@ public class ORM {
         }
     }
     
+    /**
+     * Persists the given contract.
+     *
+     * @param contract the contract that should be persisted
+     */
+    public void persist(Contract contract)
+    {
+        boolean changeFinished = false;
+        try {
+            _connection.setAutoCommit(false);
+            if (contract.getContractNo() == -1) {
+                String insertSQL = "INSERT INTO CONTRACT (date, place) " +
+                                   "VALUES (?, ?)";
+                PreparedStatement pstmt = _connection.prepareStatement(insertSQL, Statement.RETURN_GENERATED_KEYS);
+                pstmt.setString(1, contract.getDate());
+                pstmt.setString(2, contract.getPlace());
+                pstmt.executeUpdate();
+                
+                ResultSet rs = pstmt.getGeneratedKeys();
+                if (rs.next()) {
+                    contract.setContractNo(rs.getInt(1));
+                }
+                rs.close();
+                pstmt.close();
+                
+                if (contract instanceof PurchaseContract) {
+                    PurchaseContract purchaseContract = (PurchaseContract) contract;
+                    String insertSQLPurchaseContract = "INSERT INTO PURCHASECONTRACT " +
+                                                       "(contractNumber, numberOfInstallments, interestRate) " +
+                                                       "VALUES (?, ?, ?)";
+                    PreparedStatement pstmtPurchase = _connection.prepareStatement(insertSQLPurchaseContract);
+                    pstmtPurchase.setInt(1, purchaseContract.getContractNo());
+                    pstmtPurchase.setInt(2, purchaseContract.getNoOfInstallments());
+                    pstmtPurchase.setInt(3, purchaseContract.getInterestRate());
+                    pstmtPurchase.executeUpdate();
+                    pstmtPurchase.close();
+                    
+                    String insertSQLSale = "INSERT INTO SALES (contractNumber, house, person) VALUES (?, ?, ?)";
+                    PreparedStatement pstmtSale = _connection.prepareStatement(insertSQLSale);
+                    pstmtSale.setInt(1, purchaseContract.getContractNo());
+                    pstmtSale.setInt(2, purchaseContract.getHouse());
+                    pstmtSale.setInt(3, purchaseContract.getPerson());
+                    pstmtSale.executeUpdate();
+                    pstmtSale.close();
+                    changeFinished = true;
+                }
+                else if (contract instanceof TenancyContract) {
+                    TenancyContract tenancyContract = (TenancyContract) contract;
+                    String insertSQLTenancyContract = "INSERT INTO TENANCYCONTRACT " +
+                                                      "(contractNumber, startDate, duration, additionalCosts) " +
+                                                      "VALUES (?, ?, ?, ?)";
+                    PreparedStatement pstmtTenancy = _connection.prepareStatement(insertSQLTenancyContract);
+                    pstmtTenancy.setInt(1, tenancyContract.getContractNo());
+                    pstmtTenancy.setTimestamp(2, tenancyContract.getStartDate());
+                    pstmtTenancy.setTimestamp(3, new Timestamp(tenancyContract.getDuration().toMillis()));
+                    pstmtTenancy.setInt(4, tenancyContract.getAdditionalCost());
+                    pstmtTenancy.executeUpdate();
+                    pstmtTenancy.close();
+    
+                    String insertSQLRental = "INSERT INTO RENTALS (contractNumber, apartment, person) VALUES (?, ?, ?)";
+                    PreparedStatement pstmtRental = _connection.prepareStatement(insertSQLRental);
+                    pstmtRental.setInt(1, tenancyContract.getContractNo());
+                    pstmtRental.setInt(2, tenancyContract.getApartment());
+                    pstmtRental.setInt(3, tenancyContract.getPerson());
+                    pstmtRental.executeUpdate();
+                    pstmtRental.close();
+                    changeFinished = true;
+                }
+            } else {
+                System.err.println("Changing contracts is not supported.");
+            }
+            if (changeFinished) {
+                _connection.commit();
+                if (!_contracts.containsKey(contract.getContractNo())) {
+                    _contracts.put(contract.getContractNo(), contract);
+                }
+            }
+            _connection.setAutoCommit(true);
+        } catch (SQLException e) {
+            try {
+                _connection.rollback();
+            } catch (SQLException e1) {
+                e1.printStackTrace();
+            }
+            e.printStackTrace();
+        }
+    }
+    
     /**
      * Persists the given person.
      *