userservlet.java

JBuilder 2006 +SQL Server2000 +JavaBean+JSP开发的宠物医院管理系统

package com.pet.user.action;

import javax.servlet.*;
import javax.servlet.http.*;
import java.io.*;
import java.util.*;
import com.pet.user.model.DAO.UserDao;
import com.pet.user.model.DAO.impl.Userimpl;
import com.pet.user.model.DTO.UserBean;
import com.pet.vet.model.DTO.VetBean;
import com.pet.vet.model.DAO.VetDao;
import com.pet.vet.model.DAO.impl.vetimpl;
import com.pet.pets.model.DAO.PetDao;
import com.pet.pets.model.DAO.impl.Petimpl;
import com.pet.background.model.DAO.NewsDao;
import com.pet.background.model.DAO.impl.Newsimpl;

public class UserServlet extends HttpServlet {
    private static final String CONTENT_TYPE = "text/html; charset=GB2312";

    //Initialize global variables
    public void init() throws ServletException {
    }

    //Process the HTTP Get request
    public void doGet(HttpServletRequest request, HttpServletResponse response) throws
            ServletException, IOException {
        response.setContentType(CONTENT_TYPE);
        PrintWriter out = response.getWriter();
        String username = request.getParameter("username");
        String userpassword = request.getParameter("userpassword");
        String yzm = request.getParameter("yzm");
        HttpSession session = request.getSession();
        String key = session.getAttribute("key").toString();
        String select = request.getParameter("select");
        VetDao vetdao = new vetimpl();
        UserDao userdao = new Userimpl();
        if (key.equals(yzm.toUpperCase())) {
            //普通用户登录
            if (select.equals("pt")) {
                UserBean userbean = new UserBean();
                userbean = userdao.getuserbean(username, userpassword);
                if (userbean.getUname() != null) {
                    //登录成功
                    session.setAttribute("userBean", userbean);
                    PetDao petdao = new Petimpl();
                    ArrayList arrayList = new ArrayList();
                    arrayList = petdao.allType();
                    session.setAttribute("pettype",arrayList);
                    arrayList = vetdao.allvet();
                    session.setAttribute("allvet",arrayList);
                    //获得标题信息
                    ArrayList alnews = new ArrayList();
                    NewsDao newsdao = new Newsimpl();
                    alnews = newsdao.seebiaoti("pt");
                    session.setAttribute("news",alnews);
                    request.getRequestDispatcher("left.jsp").forward(request,
                            response);
                } else {
                    //用户名或密码错误返回
                    out.println("<script type=\"text/javascript\" language=\"javascript\">alert(\"用户名或密码错误！\");window.history.go(-1);</script>");
                }

            }
            //兽医登录
            else if (select.equals("vet")) {
                VetBean vb = new VetBean();
                vb = vetdao.customer(username, userpassword);
                if (vb.getVname() != null) {
                    //登录成功
                    session.setAttribute("vetBean", vb);
                    //获得标题信息
                    ArrayList alnews = new ArrayList();
                    NewsDao newsdao = new Newsimpl();
                    alnews = newsdao.seebiaoti("vet");
                    session.setAttribute("news",alnews);
                    request.getRequestDispatcher("leftvet.jsp").forward(request,
                            response);
                } else {
                    //用户名或密码错误返回
                     out.println("<script type=\"text/javascript\" language=\"javascript\">alert(\"用户名或密码错误！\");window.history.go(-1);</script>");
                }

            }
        } else {
            //验证码错误！返回
            out.println("<script type=\"text/javascript\" language=\"javascript\">alert(\"验证码错误！\");window.history.go(-1);</script>");
        }

    }

    //Process the HTTP Post request
    public void doPost(HttpServletRequest request, HttpServletResponse response) throws
            ServletException, IOException {
        doGet(request, response);
    }

    //Clean up resources
    public void destroy() {
    }
}