loginservlet.java

来自「图书馆管理的源代码文件」· Java 代码 · 共 54 行

package com.nitpro.webmvc.servlet;

import java.io.IOException;

import javax.servlet.RequestDispatcher;
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.nitpro.webmvc.bean.User;
import com.nitpro.webmvc.model.UserModel;

public class LoginServlet extends HttpServlet {
	protected void service(HttpServletRequest req, HttpServletResponse resp)
			throws ServletException, IOException {
		req.setCharacterEncoding("UTF-8");
		String method = req.getParameter("method");
		if ("login".equals(method)) {
			login(req, resp);
		} else if ("logout".equals(method)) {
			logout(req, resp);
		}
	}

	private void login(HttpServletRequest req, HttpServletResponse resp)
			throws ServletException, IOException {
		String userName = req.getParameter("username");
		String pwd = req.getParameter("pwd");

		UserModel um = new UserModel();
		User u = um.login(userName, pwd);
		if (u == null) {
			req.setAttribute("msg", "用户名或者密码错误!");
			RequestDispatcher rd = req.getRequestDispatcher("/users/login.jsp");
			rd.forward(req, resp);
		} else {
			HttpSession session = req.getSession();
			session.setMaxInactiveInterval(20);
			session.setAttribute("loginuser", u);
			resp.sendRedirect("/books/book?method=all");
		}
	}

	private void logout(HttpServletRequest req, HttpServletResponse resp)
			throws ServletException, IOException {
		HttpSession session = req.getSession();
		session.removeAttribute("loginuser");
		resp.sendRedirect("/index.htm");
	}

}