mregister.java

网络邮件系统

/*
 * This product includes software developed by the
 * Apache Software Foundation (http://www.apache.org/).
 */
package ch05.module;

import java.sql.*;
import java.util.Hashtable;

import javax.servlet.http.*;

import ch05.CommonConst;

/**
 * 针对注册页面的后台处理类
 * @author ShenYK
 * @version 1.0
 */
public class MRegister
{
    public boolean registerNewUser ( HttpSession mySession, 
    		                           String username, 
    		                           String password, 
    		                           String realname)
    {
        //设置用户信息
        Hashtable myValues = (Hashtable)mySession.getAttribute(CommonConst.VIEWID_REGISTER);
        myValues.put("username", username);
        myValues.put("realname", realname);
        
        //尝试注册新用户
        try
        {
            //载入MySQL的JDBC驱动类
            Class.forName(CommonConst.DB_DRIVER_CLASSNAME);
            //获得数据库连接
            Connection conn = DriverManager.getConnection( CommonConst.DB_CONN_STRING );

            Statement stmt = null;
            ResultSet rs = null;
            
            try
            {
                //检查数据库中是否已经有该用户了
                stmt = conn.createStatement();
                //执行SQL语句
                String sQuery = "select count(*) from user where username='" + username + "'";
                rs = stmt.executeQuery( sQuery );
                rs.next();
                int iUserTmp =rs.getInt(1);
                //用户名已经存在
                if ( iUserTmp > 0 )
                {
                    mySession.setAttribute("errMsg","用户名已经存在！");
                    return false;
                }
                sQuery = "insert into user values('" + username + "','" + password + "','"+ realname + "')";
                stmt.executeUpdate( sQuery );
                
                mySession.setAttribute("username", username);
                mySession.setAttribute("realname",realname );
                return true;
            }
            catch(Exception e)
            {
                e.printStackTrace();
                mySession.setAttribute("errMsg","登录数据库时出现错误！");
                return false;
            }
            finally
            {
                try
                {
                    rs.close();
                    stmt.close();
                }catch(Exception ex)
                {
                }
            }
        }catch(Exception ex)
        {
            ex.printStackTrace();
            mySession.setAttribute("errMsg","登录数据库时出现错误！");
            return false;
        }
    }
}