userservlet.java

一个简单的同学录

package txl.servlet;

import java.io.IOException;
import java.io.PrintWriter;

import javax.servlet.RequestDispatcher;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import txl.tools.*;

public class UserServlet extends HttpServlet {

	/**
	 * Constructor of the object.
	 */
	public UserServlet() {
		super();
	}

	/**
	 * The doGet method of the servlet. <br>
	 *
	 * This method is called when a form has its tag value method equals to get.
	 * 
	 * @param request the request send by the client to the server
	 * @param response the response send by the server to the client
	 * @throws ServletException if an error occurred
	 * @throws IOException if an error occurred
	 */
	public void doGet(HttpServletRequest request, HttpServletResponse response)
			throws ServletException, IOException {
		String mod = request.getParameter("mod");
		if(mod == null){
			response.sendRedirect("/txl/userlogin/login.jsp");
		}else if(mod.equals("add")){
			String username = request.getParameter("username");
			String password = request.getParameter("password");
			DbBean db = new DbBean();
			db.addUser(username, password);
			db.close();
			response.sendRedirect("/txl/userlogin/login.jsp");
		}else if(mod.equals("login")){
			String username = request.getParameter("username");
			String password = request.getParameter("password");
			DbBean db = new DbBean();
			int i = db.userFind(username, password);
			db.close();
			if(i>0){
				request.getSession().setAttribute("userLogin", "true");
				request.getSession().setAttribute("username", username);
				request.getSession().setAttribute("uid", Integer.toString(i));
				response.sendRedirect("/txl/tongxun/add.jsp");
			}else{
				request.setAttribute("error", "false");
				RequestDispatcher rd = request.getRequestDispatcher("/userlogin/login.jsp");
			    rd.forward(request, response);		
			}
		}
	}

	/**
	 * The doPost method of the servlet. <br>
	 *
	 * This method is called when a form has its tag value method equals to post.
	 * 
	 * @param request the request send by the client to the server
	 * @param response the response send by the server to the client
	 * @throws ServletException if an error occurred
	 * @throws IOException if an error occurred
	 */
	public void doPost(HttpServletRequest request, HttpServletResponse response)
			throws ServletException, IOException {
       doGet(request,response);
	}

}