diff --git a/AstuteSystem/src/main/java/com/astute/dao/DAO.java b/AstuteSystem/src/main/java/com/astute/dao/DAO.java index 19311c2..b39193b 100644 --- a/AstuteSystem/src/main/java/com/astute/dao/DAO.java +++ b/AstuteSystem/src/main/java/com/astute/dao/DAO.java @@ -132,6 +132,8 @@ public abstract class DAO { public abstract List getCustomers(String customerId) throws AstuteException; + public abstract Customer getCustomer(String poNumber) throws AstuteException; + public abstract String createCustomer(String customerId, String customerName, String billToDept, String add1, String add2, String city, String state, int zip, int ziplast4, String email, String phone, String fax) throws AstuteException; public abstract void updateCustomer(String customerId, String customerName, String billToDept, String add1, String add2, String city, String state, int zip, int ziplast4, String email, String phone, String fax) throws AstuteException; diff --git a/AstuteSystem/src/main/java/com/astute/dao/SqlDAO.java b/AstuteSystem/src/main/java/com/astute/dao/SqlDAO.java index b3d4ce4..d9f362b 100644 --- a/AstuteSystem/src/main/java/com/astute/dao/SqlDAO.java +++ b/AstuteSystem/src/main/java/com/astute/dao/SqlDAO.java @@ -606,7 +606,31 @@ public class SqlDAO extends DAO { e.printStackTrace(); throw new AstuteException(DB_ERROR,e.getMessage()); } -} + } + + public String getCustomerId(String poNumber) throws AstuteException { + String sql = "select po.customer_id from po " + + "where po.po_num = '" + poNumber + "'"; + String custId = null; + try { + Statement stmt = conn.createStatement(); + ResultSet resultSet = stmt.executeQuery(sql); + + if(resultSet.next()) { + custId = resultSet.getString(1); + } + } catch (SQLException e) { + e.printStackTrace(); + throw new AstuteException(DB_ERROR,e.getMessage()); + } + return custId; + } + public Customer getCustomer(String poNumber) throws AstuteException { + String customerId = getCustomerId(poNumber); + List customers = getCustomers(customerId); + return customers.get(0); + } + public String createCustomer(String customerId, String customerName, String billToDept, String add1, String add2, String city, String state, int zip, int ziplast4, String email, String phone, String fax) throws AstuteException { try {