public int newCustomer(int id) throws RemoteException { Trace.info("INFO: RM::newCustomer(" + id + ") called"); // Generate a globally unique ID for the new customer int cid = Integer.parseInt( String.valueOf(id) + String.valueOf(Calendar.getInstance().get(Calendar.MILLISECOND)) + String.valueOf(Math.round(Math.random() * 100 + 1))); Customer cust = new Customer(cid); writeData(id, cust.getKey(), cust); Customer temp = cust.clone(); temp.setID(-1); temp.setType(1); writeDataToLog(id, cust.getKey(), temp); Trace.info("RM::newCustomer(" + cid + ") returns ID=" + cid); return cid; }
// deletes the entire item protected boolean deleteItem(int id, String key) { Trace.info("RM::deleteItem(" + id + ", " + key + ") called"); ReservableItem curObj = (ReservableItem) readData(id, key); Flight tempItem = new Flight(Integer.parseInt(curObj.getLocation()), curObj.getCount(), curObj.getPrice()); tempItem.setReserved(curObj.getReserved()); // Check if there is such an item in the storage if (curObj == null) { Trace.warn("RM::deleteItem(" + id + ", " + key + ") failed--item doesn't exist"); return false; } else { if (curObj.getReserved() == 0) { tempItem.setType(0); writeDataToLog(id, curObj.getKey(), tempItem); removeData(id, curObj.getKey()); Trace.info("RM::deleteItem(" + id + ", " + key + ") item deleted"); return true; } else { Trace.info( "RM::deleteItem(" + id + ", " + key + ") item can't be deleted because some customers reserved it"); return false; } } // if }
// Deletes customer from the database. public boolean deleteCustomer(int id, int customerID) throws RemoteException { Trace.info("RM::deleteCustomer(" + id + ", " + customerID + ") called"); Customer cust = (Customer) readData(id, Customer.getKey(customerID)); if (cust == null) { Trace.warn( "RM::deleteCustomer(" + id + ", " + customerID + ") failed--customer doesn't exist"); return false; } else { Customer temp = cust.clone(); // Increase the reserved numbers of all reservable items which the customer reserved. RMHashtable reservationHT = cust.getReservations(); for (Enumeration e = reservationHT.keys(); e.hasMoreElements(); ) { String reservedkey = (String) (e.nextElement()); ReservedItem reserveditem = cust.getReservedItem(reservedkey); Trace.info( "RM::deleteCustomer(" + id + ", " + customerID + ") has reserved " + reserveditem.getKey() + " " + reserveditem.getCount() + " times"); ReservableItem item = (ReservableItem) readData(id, reserveditem.getKey()); Trace.info( "RM::deleteCustomer(" + id + ", " + customerID + ") has reserved " + reserveditem.getKey() + "which is reserved" + item.getReserved() + " times and is still available " + item.getCount() + " times"); Flight tempItem = new Flight(Integer.parseInt(item.getLocation()), item.getCount(), item.getPrice()); tempItem.setReserved(item.getReserved()); tempItem.setType(0); item.setReserved(item.getReserved() - reserveditem.getCount()); item.setCount(item.getCount() + reserveditem.getCount()); if (readDataFromLog(id, item.getKey(), id) == null) writeDataToLog(id, item.getKey(), tempItem); } // remove the customer from the storage temp.setType(1); if (readDataFromLog(id, cust.getKey(), id) == null) writeDataToLog(id, cust.getKey(), temp); removeData(id, cust.getKey()); Trace.info("RM::deleteCustomer(" + id + ", " + customerID + ") succeeded"); return true; } // if }
// Create a new flight, or add seats to existing flight // NOTE: if flightPrice <= 0 and the flight already exists, it maintains its current price public boolean addFlight(int id, int flightNum, int flightSeats, int flightPrice) throws RemoteException, InvalidTransactionException, TransactionAbortedException { Trace.info( "RM::addFlight(" + id + ", " + flightNum + ", $" + flightPrice + ", " + flightSeats + ") called"); Flight curObj = (Flight) readData(id, Flight.getKey(flightNum)); if (curObj == null) { // doesn't exist...add it Flight newObj = new Flight(flightNum, flightSeats, flightPrice); writeData(id, newObj.getKey(), newObj); String key = newObj.getKey(); if (readDataFromLog(id, key, id) == null) { Flight logObj = (Flight) newObj.clone(); logObj.setCount(-1); logObj.type = 0; writeDataToLog(id, key, logObj); } Trace.info( "RM::addFlight(" + id + ") created new flight " + flightNum + ", seats=" + flightSeats + ", price=$" + flightPrice); } else { // add seats to existing flight and update the price... Flight logObj = (Flight) curObj.clone(); if (readDataFromLog(id, curObj.getKey(), id) == null) writeDataToLog(id, curObj.getKey(), logObj); curObj.setCount(curObj.getCount() + flightSeats); if (flightPrice > 0) { curObj.setPrice(flightPrice); } // if writeData(id, curObj.getKey(), curObj); Trace.info( "RM::addFlight(" + id + ") modified existing flight " + flightNum + ", seats=" + curObj.getCount() + ", price=$" + flightPrice); } // else return (true); }
// query the price of an item protected int queryPrice(int id, String key) { Trace.info("RM::queryCarsPrice(" + id + ", " + key + ") called"); ReservableItem curObj = (ReservableItem) readData(id, key); int value = 0; if (curObj != null) { value = curObj.getPrice(); } // else Trace.info("RM::queryCarsPrice(" + id + ", " + key + ") returns cost=$" + value); return value; }
// return a bill public String queryCustomerInfo(int id, int customerID) throws RemoteException { Trace.info("RM::queryCustomerInfo(" + id + ", " + customerID + ") called"); Customer cust = (Customer) readData(id, Customer.getKey(customerID)); if (cust == null) { Trace.warn( "RM::queryCustomerInfo(" + id + ", " + customerID + ") failed--customer doesn't exist"); return ""; // NOTE: don't change this--WC counts on this value indicating a customer does not // exist... } else { String s = cust.printBill(); Trace.info("RM::queryCustomerInfo(" + id + ", " + customerID + "), bill follows..."); System.out.println(s); return s; } // if }
// Returns data structure containing customer reservation info. Returns null if the // customer doesn't exist. Returns empty RMHashtable if customer exists but has no // reservations. public RMHashtable getCustomerReservations(int id, int customerID) throws RemoteException { Trace.info("RM::getCustomerReservations(" + id + ", " + customerID + ") called"); Customer cust = (Customer) readData(id, Customer.getKey(customerID)); if (cust == null) { Trace.warn( "RM::getCustomerReservations failed(" + id + ", " + customerID + ") failed--customer doesn't exist"); return null; } else { return cust.getReservations(); } // if }
// I opted to pass in customerID instead. This makes testing easier public boolean newCustomer(int id, int customerID) throws RemoteException { Trace.info("INFO: RM::newCustomer(" + id + ", " + customerID + ") called"); Customer cust = (Customer) readData(id, Customer.getKey(customerID)); if (cust == null) { cust = new Customer(customerID); writeData(id, cust.getKey(), cust); Customer temp = cust.clone(); temp.setType(1); temp.setID(-1); writeDataToLog(id, cust.getKey(), temp); Trace.info("INFO: RM::newCustomer(" + id + ", " + customerID + ") created a new customer"); return true; } else { Trace.info( "INFO: RM::newCustomer(" + id + ", " + customerID + ") failed--customer already exists"); return false; } // else }
// reserve an item protected boolean reserveItem(int id, int customerID, String key, String location) { Trace.info( "RM::reserveItem( " + id + ", customer=" + customerID + ", " + key + ", " + location + " ) called"); // Read customer object if it exists (and read lock it) Customer cust = (Customer) readData(id, Customer.getKey(customerID)); if (cust == null) { Trace.warn( "RM::reserveCar( " + id + ", " + customerID + ", " + key + ", " + location + ") failed--customer doesn't exist"); return false; } // check if the item is available ReservableItem item = (ReservableItem) readData(id, key); if (item == null) { Trace.warn( "RM::reserveItem( " + id + ", " + customerID + ", " + key + ", " + location + ") failed--item doesn't exist"); return false; } else if (item.getCount() == 0) { Trace.warn( "RM::reserveItem( " + id + ", " + customerID + ", " + key + ", " + location + ") failed--No more items"); return false; } else { Flight tempItem = new Flight(Integer.parseInt(item.getLocation()), item.getCount(), item.getPrice()); tempItem.setReserved(item.getReserved()); Customer temp = cust.clone(); temp.setType(1); if (readDataFromLog(id, cust.getKey(), id) == null) { writeDataToLog(id, cust.getKey(), temp); } cust.reserve(key, location, item.getPrice()); writeData(id, cust.getKey(), cust); // decrease the number of available items in the storage item.setCount(item.getCount() - 1); item.setReserved(item.getReserved() + 1); if (readDataFromLog(id, item.getKey(), id) == null) { tempItem.setType(0); writeDataToLog(id, item.getKey(), tempItem); } Trace.info( "RM::reserveItem( " + id + ", " + customerID + ", " + key + ", " + location + ") succeeded"); return true; } }