registerservlet.java

简单的流言板系统,用myeclipse进行编写的,采用简单的jstl+javabean+servlet+jsp,属于三层架构.

package me.work.servlets;

import java.io.IOException;
import java.io.PrintWriter;
import java.sql.SQLException;

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

import me.work.beans.UserBean;
import me.work.services.DBOperation;
import me.work.utils.EmailUtils;

public class RegisterServlet extends HttpServlet {

	/**
	 * serialVersionUID
	 */
	private static final long serialVersionUID = 1L;

	@Override
	protected void doGet(HttpServletRequest req, HttpServletResponse resp)
			throws ServletException, IOException {
		// invoke doPost
		doPost(req, resp);
	}

	@Override
	protected void doPost(HttpServletRequest req, HttpServletResponse resp)
			throws ServletException, IOException {
		//check the randomcode
		HttpSession session=req.getSession();
        String randomCode=(String)session.getAttribute("randomCode");
        if(null==randomCode)
        {
            resp.sendRedirect("./page/index.jsp");
            return;
        }
        
        String reqRandom=req.getParameter("random");
        
        req.setCharacterEncoding("GBK");
        resp.setContentType("text/html;charset=GBK");
        PrintWriter out=resp.getWriter();
        
        if(randomCode.equals(reqRandom))
        {
        	
        	String method = req.getParameter("method");
    		//get parameter user type
    		String userType = req.getParameter("uType");
    		if ("Register".equals(method)) {
    				// do register user
    			String userName = req.getParameter("uName");
    			String userPassword = req.getParameter("uPassword");
    			String passwordConfirm = req.getParameter("pConfirm");
    			String userMail = req.getParameter("uMail");
    			// check paramters
    			if(userName == null ||"".equals(userName.trim()) ){
    				// set the error message
    				req.setAttribute("error",
    						"Please input user Name");
    				// forward to register page to show the error to user
    				req.getRequestDispatcher("./pages/register.jsp").forward(req,
    						resp);
    				return;
    			}
    			if(userPassword == null || "".equals(userPassword.trim())){
    				// set the error message
    				req.setAttribute("error",
    						"Please input password");
    				// forward to register page to show the error to user
    				req.getRequestDispatcher("./pages/register.jsp").forward(req,
    						resp);
    				return;
    			}		
    			// Check passwords
    			if (!userPassword.equals(passwordConfirm)) {
    				req.setAttribute("uName", userName.trim());
    				req.setAttribute("uMail",userMail.trim());
    				req.setAttribute("error", "passwords do not match!");
    				// forward to register page to show the error to user
    				req.getRequestDispatcher("./pages/register.jsp").forward(req,
    						resp);
    				return;
    			}
    			// Check email
    			if (!EmailUtils.isEmail(userMail)) {
    				req.setAttribute("uName", userName.trim());
    				req.setAttribute("uPassword", userPassword.trim());
    				req.setAttribute("pConfirm", passwordConfirm.trim());
    				req.setAttribute("error", "please input legal email!");
    				// forward to register page to show the error to user
    				req.getRequestDispatcher("./pages/register.jsp").forward(req,
    						resp);
    				return;
    			}
    			// create a user bean
    			UserBean user = new UserBean();
    			// set user bean's info
    			user.setUserEmail(userMail);
    			user.setUserName(userName);
    			user.setUserPassword(userPassword);
    			// create a DBOperation to operate DB
    			DBOperation dbo = new DBOperation();

    			try {
    				if (dbo.registerUser(user)) {
    					// set user name into session
    					//HttpSession session = req.getSession();
    					session.setAttribute("userName", user.getUserName());
    					// forward to userControl
    					req.getRequestDispatcher("./pages/userControl.jsp").forward(req,
    							resp);
    					return;
    				}
    			} catch (ClassNotFoundException e) {
    				// TODO Auto-generated catch block
    				e.printStackTrace();
    			} catch (SQLException e) {
    				// TODO Auto-generated catch block
    				e.printStackTrace();
    			}

    		} else {
    			// return to login.jsp
    			req.getRequestDispatcher("./pages/index.jsp").forward(req, resp);
    			return;
    		}
        }
        else
        {
            req.setAttribute("error", "check false,please write again!");
			// forward to register page to show the error to user
			req.getRequestDispatcher("./pages/register.jsp").forward(req,
					resp);
			return;
        }
        out.close();
        //the end of random check
		
		
		
		
	}

}