registerservlet.java

Java程序设计高级教程（第三版）

package com.nearhills.servlets;

import java.io.IOException;

import javax.servlet.RequestDispatcher;
import javax.servlet.ServletException;
import javax.servlet.http.Cookie;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.servlet.http.HttpSession;

import com.nearhills.BadDataException;
import com.nearhills.ClubMember;

/**
 * @version 	1.0
 * @author
 */
public class RegisterServlet extends HttpServlet {

	/**
	* @see javax.servlet.http.HttpServlet#void (javax.servlet.http.HttpServletRequest, javax.servlet.http.HttpServletResponse)
	*/
	public void doPost(HttpServletRequest req, HttpServletResponse resp)
		throws ServletException, IOException { 
      HttpSession session = req.getSession(false);
      if ( session != null ) {
         session.invalidate();
      }
      ClubMember skier = null;
      try {
	  skier = getMemberData(req);
	  skier.store();
	   confirmRegistration( req, resp, skier );
      } catch( BadDataException e ) {
      	 req.setAttribute( "message", e.getMessage() );
         RequestDispatcher rd = getServletContext().
         getRequestDispatcher( 
            "/badinputMessage.jsp" );
         rd.forward( req, resp );
      } catch( Exception e ) {
      	 req.setAttribute( "exception", e );
         RequestDispatcher rd = getServletContext().
         getRequestDispatcher( 
            "/errorReport.jsp" );
         rd.forward( req, resp );
      } 
      return;
   }

   private ClubMember getMemberData( 
      HttpServletRequest req )
      throws BadDataException 
   {
      ClubMember skier = new ClubMember();
      skier.setId( 0 );
      skier.setFirstName( req.getParameter( "FNAME" ) );
      if ( skier.getFirstName().length() == 0 ) {
         throw new BadDataException( 
            "First Name Required" );
      } 
      skier.setLastName( req.getParameter( "LNAME" ) );
      if ( skier.getLastName().length() == 0 ) {
  	 throw new BadDataException( 
         "Last Name Required" );
      } 
      skier.setMiddleName( req.getParameter( "MNAME" ) );
      skier.setPhoneNumber( req.getParameter( "PHONE" ) );
      skier.setAbility( req.getParameter( "ABILITY" ) );
      return skier;
   }
 
   public void confirmRegistration( 
      HttpServletRequest req, HttpServletResponse resp,
      ClubMember skier )
      throws IOException, ServletException
   {  
      try { 
         HttpSession session = req.getSession( true );
         session.setAttribute( "member", skier );
         resp.addCookie( 
            new Cookie( "NewMember", "yes" ) );
         RequestDispatcher rd = getServletContext().
         getRequestDispatcher( 
            "/newMember.jsp" );
         rd.forward( req, resp );
      } catch ( Exception e ) {
      	 req.setAttribute( "exception", e );
         RequestDispatcher rd = getServletContext().
         getRequestDispatcher( 
            "/errorReport.jsp" );
         rd.forward( req, resp );
      }  
   }

}