userservlet.java

利用servlet写的用户注册登陆

package com.tarena.serv.app;

import java.io.IOException;
import java.util.List;

import javax.servlet.RequestDispatcher;
import javax.servlet.ServletContext;
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 com.tarena.action.UserAction;
import com.tarena.persitence.User;
import com.tarena.persitence.UserService;
import com.tarena.util.StringUtil;

public class UserServlet extends HttpServlet {
	UserService userService=new UserService();
	protected void doPost(HttpServletRequest request,
			HttpServletResponse response) throws ServletException, IOException {
		String path = request.getServletPath();
		path = path.substring(0, path.indexOf("."));
		String nextPage = null;
		boolean forward = true;
//		UserAction userAction = (UserAction) getServletContext().getAttribute(
//				"userAction");
		if (path.equals("/register")) {
			nextPage = register(request, response);
		} else if (path.equals("/find")) {
			nextPage = find(request, response);
		} else if (path.equals("/modify")) {
			nextPage = modify(request, response);
		} else if (path.equals("/login")) {
			nextPage = login(request, response);
			forward = false;
		} else {
			response.sendError(HttpServletResponse.SC_NOT_FOUND, "the path : "
					+ path + " is not a valid path.");
		}

		if (forward) {
			ServletContext application = getServletContext();
			RequestDispatcher dispatcher = application
					.getRequestDispatcher(nextPage);
			dispatcher.forward(request, response);
		} else {
			response.sendRedirect(request.getContextPath() + nextPage);
		}
	}
	private String register(HttpServletRequest request,
			HttpServletResponse response) throws ServletException, IOException {
		User user = createUser(request);
		try {
			userService.add(user);
			return "/html/success.html";
		} catch (Exception e) {
			e.printStackTrace();
			throw new ServletException(e);
		}
	}
	private String find(HttpServletRequest request, HttpServletResponse response)
	throws ServletException, IOException {
try {
	List<User> users = userService.findAll();
	request.setAttribute("users", users);
	return "/user/list";
} catch (Exception e) {
	e.printStackTrace();
	throw new ServletException(e);
}
}

private String modify(HttpServletRequest request,
	HttpServletResponse response) throws ServletException, IOException {
try {
	return "/user/message";
} catch (Exception e) {
	e.printStackTrace();
	throw new ServletException(e);
}
}

private String login(HttpServletRequest request, HttpServletResponse response)
	throws ServletException, IOException {
try {
	String userName = request.getParameter("userName");
	String password = request.getParameter("password");
	User userInfo = userService.find(userName, password);
	if (userInfo != null) {
		HttpSession session = request.getSession();
		session.setAttribute("userInfo", userInfo);
		return "/find.do";
	} else {
		return "/login.html";
	}
} catch (Exception e) {
	e.printStackTrace();
	throw new ServletException(e);
}
}

	private User createUser(HttpServletRequest request) {
		User user = new User();
		user.setUserName(request.getParameter("userName"));
		user.setPassword(request.getParameter("password"));
		user.setGender(request.getParameter("gender"));
		user.setDegree(request.getParameter("degree"));
		user.setIntroduction(request.getParameter("introduction"));
		String birthday = request.getParameter("birthday");
		if (birthday.trim().length() != 0) {
			user.setBirthday(java.sql.Date.valueOf(birthday));
		}
		String[] hobbiesAry = request.getParameterValues("hobbies");
		user.setHobbies(StringUtil.join(hobbiesAry, ","));
		return user;
	}
	protected void doGet(HttpServletRequest request,
			HttpServletResponse response) throws ServletException, IOException {
		doPost(request, response);
	}
	
}