login.java

小型简单的旅馆预订系统. 客户:预定,更改,撤销,会员积分累积,查询预定,会员. 后台:会员更改,房间更改,预定更改,预定信息统计.

package servlet;

import java.io.IOException;
import java.io.PrintWriter;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Statement;

import javax.servlet.ServletContext;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.servlet.http.HttpSession;

/**
 * Servlet implementation class for Servlet: login
 *
 */
 public class login extends javax.servlet.http.HttpServlet implements javax.servlet.Servlet {
   static final long serialVersionUID = 1L;
   
    /* (non-Java-doc)
	 * @see javax.servlet.http.HttpServlet#HttpServlet()
	 */
	public login() {
		super();
	} 
	
	/* (non-Javadoc)
	 * @see javax.servlet.Servlet#destroy()
	 */
	public void destroy() {
		// TODO Auto-generated method stub
		super.destroy();
	}   	
	
	/* (non-Java-doc)
	 * @see javax.servlet.http.HttpServlet#doGet(HttpServletRequest request, HttpServletResponse response)
	 */
	protected void doGet(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
		// TODO Auto-generated method stub
	}  	
	
	/* (non-Java-doc)
	 * @see javax.servlet.http.HttpServlet#doPost(HttpServletRequest request, HttpServletResponse response)
	 */
	protected void doPost(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
		response.setContentType("text/html;");
        PrintWriter out=response.getWriter();
        HttpSession session=request.getSession();
        String name=(String) session.getAttribute("username");
        String password=(String) session.getAttribute("password");
        String score="0",disc="1.0";
        String sco="0";
        String relID,billStatus,payStatus,payID;
        double sc=0;
        String pass=request.getParameter("password"); //get request password
        
		userBean uber = (userBean)session.getAttribute("uber");
		roomBean rmb = (roomBean)session.getAttribute("rmb");//get session bean
		if (uber == null) { uber = new userBean(); }//set a new bean object
		else ;
		if (rmb == null) { rmb = new roomBean(); }// set a new bean object
		else ;
		resBean resb = (resBean)session.getAttribute("resb"); //get session bean
		if (resb == null) { resb = new resBean(); }//set a new bean object
		else ;
		
		payBean pb = (payBean)session.getAttribute("pb"); //get session bean
		if (pb == null) { pb = new payBean(); }//set a new bean object
		else ;
		
		userInitBean uib = (userInitBean)session.getAttribute("uib"); //get session bean
		if (uib == null) { uib = new userInitBean(); }//set a new bean object
		else ;
		
		collectionBean clb=new collectionBean();
		insert in=new insert();
		
        try {
        	ServletContext context = getServletContext();
        	Class.forName("sun.jdbc.odbc.JdbcOdbcDriver").newInstance();
        	String appDirRoot = context.getRealPath("/");
        	String driver = "Driver={Microsoft Access Driver (*.mdb)}";
        	String dbq = "DBQ=E:\\Java\\Tomcat 6.0\\webapps\\ROOT\\database\\HRSDB.mdb";
        	String connectionString = "jdbc:odbc:"+driver+";"+dbq;
        	Connection conn = DriverManager.getConnection(connectionString);

            Statement stmt = conn.createStatement();
            String query = "SELECT * FROM User where username = '" + name + "'";
            ResultSet rs = stmt.executeQuery(query);
            if (rs.next()) {
                if (rs.getString("password").equals(password)) 
                {
                    uber.setUsername(name);
                    uber.setFname(rs.getString("fname"));
                    uber.setLname(rs.getString("lname"));
                    uber.setGender(rs.getString("gender"));
                    uber.setDob(rs.getString("dob"));
                    uber.setEmail(rs.getString("email"));
                    uber.setAdd(rs.getString("address"));
                    uber.setTel(rs.getString("tel"));
                    uber.setId(rs.getString("ID"));
                    String type=rs.getString("type");
                    uber.setType(type);
                    sco=rs.getString("score");
                    sc=Double.valueOf(sco);
                    uber.setScore(sco);
                    uber.setMessage("Success");
                    if (sc>0 && sc<100)
                    {
                    	uber.setDiscount("1.0");
                    }
                    else if (sc>=100 && sc<300)
                    {
                    	uber.setDiscount("0.95");
                    }
                    else if (sc>=300 && sc<500)
                    {
                    	uber.setDiscount("0.90");
                    }
                    else if (sc>=500 && sc<1000)
                    {
                    	uber.setDiscount("0.85");
                    }
                    else if (sc>=1000 && sc<1500)
                    {
                    	uber.setDiscount("0.80");
                    }
                    else uber.setDiscount("0.78");
                    
                    rs.close();
                    stmt.close();
                    
                    session.removeAttribute("password");
                    session.removeAttribute("username");
                    if (type.equals("0"))
                    {
                    	uber.setType("Administrator");
                    	response.sendRedirect("accountOverview.jsp");
                    }
                    else
                    {
                        response.sendRedirect("accountOverview.jsp");
                        uber.setType("Customer");
                    }
                } else {
                    uber.setMessage("Incorrect password");
                    session.removeAttribute("password");
                    response.sendRedirect("memberZone.jsp");
                }
            } else {
                uber.setMessage("No such user");
                session.removeAttribute("password");
                response.sendRedirect("memberZone.jsp");
            }
            conn.close();
        }
        catch (SQLException e)
        {
        	out.println("Database!:  "+e.getCause()+"  |  "+e.getStackTrace()+"  |  "+e.getMessage());
        	//response.sendRedirect("error.jsp");
        }
        catch (Exception e) {
            out.println("Error!:  "+e.getCause()+"  |  "+e.getStackTrace()+"  |  "+e.getMessage());
            //response.sendRedirect("error.jsp");
        }
        
	}   	  	  
	
	/* (non-Javadoc)
	 * @see javax.servlet.GenericServlet#init()
	 */
	public void init() throws ServletException {
		// TODO Auto-generated method stub
		super.init();
	}   
}