coursebean.java

j2ee tutorial

/*
 *
 * Copyright 2001 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.*;
import javax.rmi.PortableRemoteObject;

public class CourseBean implements EntityBean {

   private String courseId;
   private String name;
   private ArrayList studentIds;
   private Connection con;
   private String dbName = "java:comp/env/jdbc/CollegeDB";
   private EntityContext context;
   private EnrollerHome enrollerHome;

   public ArrayList getStudentIds() {

      System.out.println("in getStudentIds");
      return studentIds;
   }

   public String getName() {

      return name;
   }

   public void setName(String name) {

      this.name = name;
   }

   public String ejbCreate(String courseId, String name)
       throws CreateException {
   
       System.out.println("in ejbCreate");
       try {
          insertCourse(courseId, name);
       } catch (Exception ex) {
           throw new EJBException("ejbCreate: " + 
              ex.getMessage());
       }

       this.courseId = courseId;
       this.name = name;
       System.out.println("about to leave ejbCreate");
       return courseId;
   }

   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 void ejbRemove() {

      try {
         deleteCourse(courseId);
       } catch (Exception ex) {
           throw new EJBException("ejbRemove: " + 
              ex.getMessage());
       }
   } 

   public void setEntityContext(EntityContext context) {

      System.out.println("in setEntityContext");
      this.context = context;
      studentIds = new ArrayList();
      try {
         makeConnection();
         Context initial = new InitialContext();
         Object objref = initial.lookup("java:comp/env/ejb/Enroller");
         enrollerHome = (EnrollerHome)PortableRemoteObject.narrow(objref,
                                           EnrollerHome.class);
         
      } 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() {

      courseId = (String)context.getPrimaryKey();
   }

   public void ejbPassivate() {

      courseId = null;
   }
   
   public void ejbLoad() {

      System.out.println("in ejbLoad");
      try {
         loadCourse();
         loadStudentIds();
       } catch (Exception ex) {
           throw new EJBException("ejbLoad: " +
              ex.getMessage());
       }
      System.out.println("leaving ejbLoad");
   }

   private void loadStudentIds() {

      System.out.println("in loadStudentIds");
      studentIds.clear();

      try {
         Enroller enroller = enrollerHome.create();
         ArrayList a = enroller.getStudentIds(courseId);
         studentIds.addAll(a);

      } catch (Exception ex) {
          throw new EJBException("loadStudentIds: " +
             ex.getMessage());
       }
      System.out.println("leaving loadStudentIds");
   }

   public void ejbStore() {

      System.out.println("in ejbStore");
      try {
         storeCourse();
       } catch (Exception ex) {
           throw new EJBException("ejbStore: " +
              ex.getMessage());
       }
      System.out.println("leaving ejbStore");
   }

   public void ejbPostCreate(String courseId, String name) { }

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

   private void makeConnection() throws NamingException, SQLException {

      InitialContext ic = new InitialContext();
      DataSource ds = (DataSource) ic.lookup(dbName);
      con =  ds.getConnection();
   }

   private void insertCourse (String courseId, String name)
       throws SQLException {

          String insertStatement =
                "insert into course values ( ? , ? )";
          PreparedStatement prepStmt = 
                con.prepareStatement(insertStatement);

          prepStmt.setString(1, courseId);
          prepStmt.setString(2, name);

          prepStmt.executeUpdate();
          prepStmt.close();
   }

   private boolean selectByPrimaryKey(String primaryKey) 
      throws SQLException {

      String selectStatement =
            "select courseid " +
            "from course where courseid = ? ";
      PreparedStatement prepStmt =
            con.prepareStatement(selectStatement);
      prepStmt.setString(1, primaryKey);

      ResultSet rs = prepStmt.executeQuery();
      boolean result = rs.next();
      prepStmt.close();
      return result;
   }

   private void deleteCourse(String courseId) throws SQLException {

      String deleteStatement =
            "delete from course  " +
            "where courseid = ?";
      PreparedStatement prepStmt =
            con.prepareStatement(deleteStatement);

      prepStmt.setString(1, courseId);
      prepStmt.executeUpdate();
      prepStmt.close();
   }

   private void loadCourse() throws SQLException {

      String selectStatement =
            "select name " +
            "from course where courseid = ? ";
      PreparedStatement prepStmt = 
            con.prepareStatement(selectStatement);

      prepStmt.setString(1, courseId);

      ResultSet rs = prepStmt.executeQuery();

      if (rs.next()) {
         name = rs.getString(1);
         prepStmt.close();
      }
      else {
         prepStmt.close();
         throw new NoSuchEntityException("Row for courseId " + courseId +
            " not found in database.");
      }
   }

   private void storeCourse() throws SQLException {

      String updateStatement =
            "update course set name =  ? " +
            "where courseid = ?";
      PreparedStatement prepStmt = 
            con.prepareStatement(updateStatement);

      prepStmt.setString(1, name);
      prepStmt.setString(2, courseId);
      int rowCount = prepStmt.executeUpdate();
      prepStmt.close();

      if (rowCount == 0) {
         throw new EJBException("Storing row for courseId " + 
            courseId + " failed.");
      }
   }


} // CourseBean