login.java

本程序用jsp+servlet+javaBean+EJB实现交友系统

package com.database;

import java.sql.SQLException;

import javax.servlet.http.HttpServletRequest;

/**
 	*包括一般注册用户和管理员的登录
 	*改类核实用户名和密码正确后，才允许用户登录
 */
public class login extends DataBase{
	private String loginName;//声明登录用户名
	private String password;//声明登录密码
	private String loginType;//声明登录方式
	private String status;//声明用户类别
	private HttpServletRequest request;//建立页面请求	
	private String message="";	//声明出错信息提示
	private boolean isAdmin;//声明是否管理员登录
	private long id=0;//
	private String nickName;
	private String userGender;
	
	public long getId() {
		return id;
	}
	public String getNickName() {
		return nickName;
	}
	public String getUserGender() {
		return userGender;
	}

	public login() throws Exception{
		super();
		loginName="";
		password="";
		loginType="nickName";
		status="common";
		isAdmin=false;
	}
	
	public boolean getRequest(HttpServletRequest newrequest){
		boolean flag=false;
		try{
			request=newrequest;
//			获取loginType参数			
			loginType=request.getParameter("loginType");
			if(loginType==null||loginType.equals("")){
				loginType="nickName";
				message=message+"<br>请选择您的登录类型";
			}
//			获取loginName参数
			loginName=request.getParameter("loginName");
			if(loginName==null||loginName.equals("")){
				loginName="";
				message=message+"<br>请输入您的登录用户名";
			}
//			获取password参数
			password=request.getParameter("password");
			if(password==null||password.equals("")){
				password="";
				message=message+"<br>请输入您的密码";
			}
//			获取status参数
			status=request.getParameter("status");
			if(status==null||status.equals("")){
				status="common";
				message=message+"<br>请选择您的用户级别";
			}			
//			
			if(message.equals("")){
				flag=true;
			}
			return flag;
			}catch(Exception e){
				return flag;
			}
		}
	
	public void setAdmin(boolean isAdmin) {
		this.isAdmin = isAdmin;
	}
	
	public String getMessage() {
		return message;
	}
	
	public String getSql(HttpServletRequest req) {	
			if (isAdmin) {
				sql = "select * from my_Adminuser where " +
					"adminuser = '" + strformat.toSql(loginName) + "' and adminpass = '" + strformat.toSql(password) + "'";
			}
			else if(status.equals("common")){
				if(loginType.equals("nickName")){
					sql = "select id,nickName,userGender from my_users where " +
						"nickName = '" + strformat.toSql(loginName) + "' and password = '" + strformat.toSql(password) + "'";
				}
				else if(loginType.equals("ID")){
					sql = "select id,nickName,userGender from my_users where " +
						"id = '" + strformat.toSql(loginName) + "' and password = '" + strformat.toSql(password) + "'";
				}
			}
			else if(status.equals("vip")){
				if(loginType.equals("nickName")){
					sql = "select id,nickName,userGender from my_users where " +
						"nickName = '" + strformat.toSql(loginName) + "' and password = '" + strformat.toSql(password) 
						+ "' and Id in (select Id from vip_users)";
				}
				else if(loginType.equals("ID")){
					sql = "select id,nickName,userGender from my_users where " +
					"Id = '" + strformat.toSql(loginName) + "' and password = '" + strformat.toSql(password) 
					+ "' and Id in (select Id from vip_users)";
					}
			}
		return sql;
	}

	public boolean excute(HttpServletRequest req)throws Exception{
		boolean flag = false;
		if(getRequest(req)){
			rs = stmt.executeQuery(getSql(req));
			if (rs.next()){	
				if (!isAdmin){
					id = rs.getLong("id");
					nickName=rs.getString("nickName");
					userGender=rs.getString("userGender");
				}
				flag = true;
			}
			rs.close();
			return flag;		
		}else{
			return flag;
		}
	}
//	更新用户信息加入数据库	
	public boolean update(HttpServletRequest req)throws Exception{
		String nickName=(String)req.getAttribute("nickName");
		String password=req.getParameter("password");
		sql="update my_users set "
			+"password='"+password+"'"			
			+"where nickName='"+nickName+"'";		
		try{
			stmt.executeUpdate(sql);
			message=message+"成功！";
			stmt.close();				
			return true;
		}catch(SQLException sqle){
			System.out.println(sqle.toString());
			return false;
		}
	}	
}