loginservlet.java

用Servlet做的一个小项目。用Servlet做的一个小项目。用Servlet做的一个小项目。用Servlet做的一个小项目。用Servlet做的一个小项目。

package controller;

import java.io.IOException;
import java.io.PrintWriter;
import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.List;

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 model.AddressBookBean;
import model.UserBean;
import model.dao.BaseDao;

/**
 * 登录验证
 */
public class LoginServlet extends HttpServlet {

	static Connection con = null;
	static PreparedStatement pstmt = null;
	static ResultSet rs = null;
	static UserBean userBean = new UserBean();
	static AddressBookBean bookBean;
	static List list = new ArrayList();

	public void doPost(HttpServletRequest request, HttpServletResponse response)
			throws ServletException, IOException {
		//定义session对象
		HttpSession session = request.getSession();
		// 接收网页传值
		String userName = request.getParameter("userName").trim();
		String password = request.getParameter("password").trim();
		// 定义查询语句
		String sql = "select * from USERS where userName=? and password=?";

		try {
			// 获取连接
			con = BaseDao.getCon();
			// 预处理SQL语句
			pstmt = con.prepareStatement(sql);
			// 分别给两个占位符赋值
			pstmt.setString(1, userName);
			pstmt.setString(2, password);
			// 执行查询获得结果保存在rs结果集中
			rs = pstmt.executeQuery();
			// 如果存在
			if (rs.next()) {
				// 给实体类赋值
				userBean.setUserId(rs.getInt("userId"));
				userBean.setUserName(rs.getString("userName"));
				userBean.setPassword(rs.getString("password"));
			}
			// 比较判断
			if (userName.equals(userBean.getUserName())
					&& password.equals(userBean.getPassword())) {
				// 验证通过则转向bookContent.jsp
				session.setAttribute("user", userBean);
				response.sendRedirect("bookContent.jsp");
			} else {
				response.sendRedirect("error.jsp");
			}
		} catch (SQLException ex) {
			// 异常处理
			ex.printStackTrace();
		} finally {
			// 释放资源，关闭连接
			BaseDao.CloseAll(con, pstmt, rs);
		}
	}
	
	public static void getRelation(HttpServletRequest request, HttpServletResponse response) {
		HttpSession session = request.getSession(false);
		//根据ID查找关系人
		String sqlSearch = "select * from BOOK where userId="
				+ userBean.getUserId();

		try {
			//获得连接
			con = BaseDao.getCon();
			//预处理SQL语句
			pstmt = con.prepareStatement(sqlSearch);
			//执行命令
			rs = pstmt.executeQuery();
			//如果存在循环赋值并保存进list
			while (rs.next()) {
				bookBean = new AddressBookBean();
				bookBean.setId(rs.getInt("id"));
				bookBean.setName(rs.getString("name"));
				bookBean.setSex(rs.getString("sex"));
				bookBean.setPhone(rs.getString("phone"));
				bookBean.setMobilePhone(rs.getString("mobilePhone"));
				list.add(bookBean);
			}
			//存入session中
			session.setAttribute("list", list);
		} catch (SQLException ex) {
			ex.printStackTrace();
		} finally {
			BaseDao.CloseAll(con, pstmt, rs);
		}
	}
}