loginservlet.java

来自「用java编的用到」· Java 代码 · 共 74 行

package djr.servlet;

import java.io.IOException;
import java.sql.Connection;
import java.util.List;
import javax.servlet.*;
import javax.servlet.http.*;
import djr.dao.*;
import djr.db.DbFactory;
import djr.dto.*;
import djr.mail.MailManager;

public class LoginServlet extends HttpServlet
{

	public void doGet(HttpServletRequest request, HttpServletResponse response) 
			throws ServletException, IOException
	{

		doPost(request, response);
	}

	public void doPost(HttpServletRequest request, HttpServletResponse response) 
			throws ServletException, IOException
	{
		String url = request.getServletPath();

		if (url.toUpperCase().indexOf("USERLOGIN") != -1)
		{
			// 登录邮件系统
			UserDTO userDTO = new UserDTO();
			userDTO.setUserid(request.getParameter("txtUserID"));
			userDTO.setPassword(request.getParameter("pwdUserPWD"));

			Connection conn = DbFactory.getConnection();

			UserDAO userDAO = new UserDAO();
			if (userDAO.findUser(userDTO, conn))
			{
				request.getSession().setAttribute("userinfo", userDTO);
				MailManager mailMgr = new MailManager();
				mailMgr.setUser(userDTO.getUserid());
				mailMgr.setPassword(userDTO.getPassword());
				if (mailMgr.receiveMail())
				{
					EmailDTO emailDTO = new EmailDTO();
					emailDTO.setUserID(userDTO.getUserid());
					List<EmailDTO> mailList = new EmailDAO().getMailList(
							emailDTO, conn);
					request.getSession().setAttribute("eMailList", mailList);
				}

				DbFactory.closeConn(conn);

				url = "main.jsp";
			}
			else
			{
				url = "err.jsp";
			}

		}
		else if (url.toUpperCase().indexOf("LOGOUT") != -1)
		{
			// 退出邮件系统
			request.getSession().invalidate();
			url = "index.jsp";
		}

		request.getRequestDispatcher(url).forward(request, response);
	}

}