/*

 *

 * Copyright 2000 Sun Microsystems, Inc. All Rights Reserved.

 * 

 * This software is the proprietary information of Sun Microsystems, Inc.  

 * Use is subject to license terms.

 * 

 */



import java.sql.*;

import javax.sql.*;

import java.util.*;

import javax.ejb.*;

import javax.naming.*;



public class OrderEJB implements EntityBean {



   private String orderId;

   private ArrayList lineItems;

   private String customerId;

   private double totalPrice;

   private String status;

   private Connection con;

   private String dbName = "java:comp/env/jdbc/OrderDB";

   private EntityContext context;





   public ArrayList getLineItems() {



      System.out.println("in getLineItems");

      return lineItems;

   }



   public String getCustomerId() {



      return customerId;

   }



   public double getTotalPrice() {



      return totalPrice;

   }

    

   public String getStatus() {



      return status;

   }



   public String ejbCreate(String orderId, String customerId,

       String status, double totalPrice, ArrayList lineItems)

       throws CreateException {

   

       System.out.println("in ejbCreate");

       try {

          insertOrder(orderId, customerId, status, totalPrice);

          for (int i = 0; i < lineItems.size(); i++) {

             LineItem item = (LineItem)lineItems.get(i);

             insertItem(item);

          }

       } catch (Exception ex) {

           throw new EJBException("ejbCreate: " + 

              ex.getMessage());

       }



       this.orderId = orderId;

       this.customerId = customerId;

       this.status = status;

       this.totalPrice = totalPrice;

       this.lineItems = lineItems ;



       System.out.println("about to leave ejbCreate");

       return orderId;

   }

     

   public String ejbFindByPrimaryKey(String primaryKey) 

      throws FinderException {



      boolean result;



      try {

         result = selectByPrimaryKey(primaryKey);

       } catch (Exception ex) {

           throw new EJBException("ejbFindByPrimaryKey: " + 

              ex.getMessage());

       }



      if (result) {

         return primaryKey;

      }

      else {

         throw new ObjectNotFoundException

            ("Row for id " + primaryKey + " not found.");

      }

   }



   public Collection ejbFindByProductId(String productId)

      throws FinderException {



      Collection result;



      try {

         result = selectByProductId(productId);

       } catch (Exception ex) {

           throw new EJBException("ejbFindByProductId " + 

              ex.getMessage());

       }



      if (result.isEmpty()) {

         throw new ObjectNotFoundException("No rows found.");

      }

      else {

         return result;

      }

   }



   public void ejbRemove() {



      try {

         deleteOrder(orderId);

         deleteItems(orderId);

       } catch (Exception ex) {

           throw new EJBException("ejbRemove: " + 

              ex.getMessage());

       }

   } 



   public void setEntityContext(EntityContext context) {



      this.context = context;

      try {

         makeConnection();

      } catch (Exception ex) {

          throw new EJBException("Unable to connect to database. " +

             ex.getMessage());

      }

   }



   public void unsetEntityContext() {



      try {

         con.close();

      } catch (SQLException ex) {

          throw new EJBException("unsetEntityContext: " + ex.getMessage());

      }

   }



   public void ejbActivate() {



      orderId = (String)context.getPrimaryKey();

   }



   public void ejbPassivate() {



      orderId = null;

   }

   

   public void ejbLoad() {



      try {

         loadOrder();

         loadItems();

       } catch (Exception ex) {

           throw new EJBException("ejbLoad: " +

              ex.getMessage());

       }

   }



   public void ejbStore() {



      try {

         storeOrder();

         for (int i = 0; i < lineItems.size(); i++) {

            LineItem item = (LineItem)lineItems.get(i);

            storeItem(item);

         }

       } catch (Exception ex) {

           throw new EJBException("ejbLoad: " +

              ex.getMessage());

       }

   }



   public void ejbPostCreate(String orderId, String customerId,

       String status, double totalPrice, ArrayList lineItems) { }



/*********************** Database Routines *************************/



   private void makeConnection() throws NamingException, SQLException {



      InitialContext ic = new InitialContext();

      DataSource ds = (DataSource) ic.lookup(dbName);

      con =  ds.getConnection();

   }



   private void insertOrder (String orderId, String customerId,

       String status, double totalPrice) throws SQLException {



          String insertStatement =

                "insert into orders values ( ? , ? , ? , ? )";

          PreparedStatement prepStmt = 

                con.prepareStatement(insertStatement);



          prepStmt.setString(1, orderId);

          prepStmt.setString(2, customerId);

          prepStmt.setDouble(3, totalPrice);

          prepStmt.setString(4, status);



          prepStmt.executeUpdate();

          prepStmt.close();

   }



   private void insertItem(LineItem lineItem)

       throws SQLException {



          String insertStatement =

                "insert into lineitems values ( ? , ? , ? , ? , ? )";

          PreparedStatement prepStmt = 

                con.prepareStatement(insertStatement);



          prepStmt.setInt(1, lineItem.getItemNo());

          prepStmt.setString(2, lineItem.getOrderId());

          prepStmt.setString(3, lineItem.getProductId());

          prepStmt.setDouble(4, lineItem.getUnitPrice());

          prepStmt.setInt(5, lineItem.getQuantity());



          prepStmt.executeUpdate();

          prepStmt.close();

   }



   private boolean selectByPrimaryKey(String primaryKey) 

      throws SQLException {



      String selectStatement =

            "select orderid " +

            "from orders where orderid = ? ";

      PreparedStatement prepStmt =

            con.prepareStatement(selectStatement);

      prepStmt.setString(1, primaryKey);



      ResultSet rs = prepStmt.executeQuery();

      boolean result = rs.next();

      prepStmt.close();

      return result;

   }



   private Collection selectByProductId(String productId) 

      throws SQLException {



      String selectStatement =

            "select distinct orderid " +

            "from lineitems where productid = ? ";

      PreparedStatement prepStmt = 

            con.prepareStatement(selectStatement);



      prepStmt.setString(1, productId);

      ResultSet rs = prepStmt.executeQuery();

      ArrayList a = new ArrayList();



      while (rs.next()) {

         String id = rs.getString(1);

         a.add(id);

      }



      prepStmt.close();

      return a;

   }



   private void deleteItems(String orderId) throws SQLException {



      String deleteStatement =

            "delete from lineitems  " +

            "where orderid = ?";

      PreparedStatement prepStmt =

            con.prepareStatement(deleteStatement);



      prepStmt.setString(1, orderId);

      prepStmt.executeUpdate();

      prepStmt.close();

   }



   private void deleteOrder(String orderId) throws SQLException {



      String deleteStatement =

            "delete from orders  " +

            "where orderid = ?";

      PreparedStatement prepStmt =

            con.prepareStatement(deleteStatement);



      prepStmt.setString(1, orderId);

      prepStmt.executeUpdate();

      prepStmt.close();

   }



   private void loadOrder() throws SQLException {



      String selectStatement =

            "select customerid, totalprice, status " +

            "from orders where orderid = ? ";

      PreparedStatement prepStmt = 

            con.prepareStatement(selectStatement);



      prepStmt.setString(1, orderId);



      ResultSet rs = prepStmt.executeQuery();



      if (rs.next()) {

         customerId = rs.getString(1);

         totalPrice = rs.getDouble(2);

         status = rs.getString(3);

         prepStmt.close();

      }

      else {

         prepStmt.close();

         throw new NoSuchEntityException("Row for orderId " + orderId +

            " not found in database.");

      }

   }



   private void loadItems() throws SQLException {



      String selectStatement =

            "select itemno, productid, unitprice, quantity " +

            "from  lineitems  where orderid = ? " +

            "order by itemno";

      PreparedStatement prepStmt = 

            con.prepareStatement(selectStatement);

      prepStmt.setString(1, orderId);

      ResultSet rs = prepStmt.executeQuery();



      int count = 0;

      while (rs.next()) {

         int itemNo = rs.getInt(1);

         String productId = rs.getString(2);

         double unitPrice = rs.getDouble(3);

         int quantity = rs.getInt(4);

         lineItems.set(itemNo - 1, 

            new LineItem(productId, quantity, unitPrice, itemNo, orderId));

         count++;

      }

      prepStmt.close();



      if (count == 0) {

         throw new NoSuchEntityException("No items for orderId " + orderId +

            " found in database.");

      }

   }



   private void storeOrder() throws SQLException {



      String updateStatement =

            "update orders set customerid =  ? ," +

            "totalprice = ? , status = ? " +

            "where orderid = ?";

      PreparedStatement prepStmt = 

            con.prepareStatement(updateStatement);



      prepStmt.setString(1, customerId);

      prepStmt.setDouble(2, totalPrice);

      prepStmt.setString(3, status);

      prepStmt.setString(4, orderId);

      int rowCount = prepStmt.executeUpdate();

      prepStmt.close();



      if (rowCount == 0) {

         throw new EJBException("Storing row for orderId " + 

            orderId + " failed.");

      }

   }



   private void storeItem(LineItem item) throws SQLException {



      String updateStatement =

            "update lineitems set productid =  ? ," +

            "unitprice = ? , quantity = ? " +

            "where orderid = ? and itemno = ?";

      PreparedStatement prepStmt = 

            con.prepareStatement(updateStatement);



      prepStmt.setString(1, item.getProductId());

      prepStmt.setDouble(2, item.getUnitPrice());

      prepStmt.setInt(3, item.getQuantity());

      prepStmt.setString(4, orderId);

      prepStmt.setInt(5, item.getItemNo());

      int rowCount = prepStmt.executeUpdate();

      prepStmt.close();



      if (rowCount == 0) {

         throw new EJBException("Storing itemNo " + item.getItemNo() +

            "for orderId " + orderId + " failed.");

      }

   }



} // OrderEJB