checkwork.java

基于struts的网上商店源码

/*
 * 作者：管磊 
 * 功能： 验证用户名是否已经被注册。
 * 
 */
package com.mole.servlet;

import java.io.IOException;
import java.io.PrintWriter;
import java.util.ArrayList;

import javax.servlet.ServletException;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import com.mole.struts.dao.DBConnector;

public class checkWork extends HttpServlet {
	private DBConnector dbc = new DBConnector();
	private HttpServletRequest request;
	private HttpServletResponse response;

	/**
	 * Constructor of the object.
	 */
	public checkWork() {
		super();
	}

	/**
	 * Destruction of the servlet. <br>
	 */
	public void destroy() {
		super.destroy(); // Just puts "destroy" string in log
		// Put your code here
	}

	/**
	 * The doPost method of the servlet. <br>
	 * 
	 * This method is called when a form has its tag value method equals to
	 * post.
	 * 
	 * @param request
	 *            the request send by the client to the server
	 * @param response
	 *            the response send by the server to the client
	 * @throws ServletException
	 *             if an error occurred
	 * @throws IOException
	 *             if an error occurred
	 */
	public void doPost(HttpServletRequest request, HttpServletResponse response)
			throws ServletException, IOException {
		this.request = request;
		this.response = response;
		String action = request.getParameter("action");
		String userName = request.getParameter("userName");
		String cardID = request.getParameter("cardID");
		String newCardID = request.getParameter("newCardID");
		String name = request.getParameter("name");
		String pass = request.getParameter("pass");
		String role = request.getParameter("role");
		String goodsNumber = request.getParameter("goodsNumber");
		String storeID = request.getParameter("storeID");
		response.setContentType("text/html");
		response.setCharacterEncoding("UTF-8");
		if ("checkusername".equals(action) && userName != null)
			checkUserName(userName);
		else if ("checkmerchantname".equals(action) && userName != null)
			checkMerchantName(userName);
		else if ("checkID".equals(action) && cardID != null)
			checkID(cardID);
		else if ("checkNewCardID".equals(action) && newCardID != null)
			checkNewID(newCardID);
		else if ("login".equals(action) && name != null && pass != null
				&& role != null)
			login(name, pass, role);
		else if ("checkGoodsNumber".equals(action) && goodsNumber != null
				&& storeID != null)
			checkGoodsNumber(goodsNumber, storeID);
	}

	/**
	 * Initialization of the servlet. <br>
	 * 
	 * @throws ServletException
	 *             if an error occurs
	 */
	public void init() throws ServletException {
		// Put your code here
	}

	private void checkUserName(String userName) {
		String sql = "select * from Customer where LoginName='" + userName
				+ "'";
		ArrayList<Object[]> list = null;
		try {
			list = dbc.executeQuery(sql);
			if (list.size() == 0)
				((PrintWriter) response.getWriter()).print("0");
			else
				((PrintWriter) response.getWriter()).print("1");
		} catch (Exception e) {
			e.printStackTrace();
		}
	}

	private void checkID(String cardID) {
		String sql = "select CustomerID from Card where ID='" + cardID + "'";
		ArrayList<Object[]> list = null;
		try {
			list = dbc.executeQuery(sql);
			if (list.size() == 0)
				((PrintWriter) response.getWriter()).print("2");
			else if (((Object[]) list.get(0))[0] == null)
				((PrintWriter) response.getWriter()).print("0");
			else
				((PrintWriter) response.getWriter()).print("1");
		} catch (Exception e) {
			e.printStackTrace();
		}
	}

	private void checkNewID(String cardID) {
		String sql = "select CustomerID from Card where ID='" + cardID + "'";
		ArrayList<Object[]> list = null;
		try {
			list = dbc.executeQuery(sql);
			if (list.size() == 0)
				((PrintWriter) response.getWriter()).print("2");
			else if (((Object[]) list.get(0))[0] == null)
				((PrintWriter) response.getWriter()).print("0");
			else
				((PrintWriter) response.getWriter()).print("1");
		} catch (Exception e) {
			e.printStackTrace();
		}
	}

	private void checkMerchantName(String merchantName) {
		String sql = "select * from Merchant where LoginName='" + merchantName
				+ "'";
		ArrayList<Object[]> list = null;
		try {
			list = dbc.executeQuery(sql);
			if (list.size() == 0)
				((PrintWriter) response.getWriter()).print("0");
			else
				((PrintWriter) response.getWriter()).print("1");
		} catch (Exception e) {
			e.printStackTrace();
		}
	}

	private void checkGoodsNumber(String goodsNumber, String storeID) {
		String sql = "select * from Goods where StoreID='" + storeID
				+ "' and GoodsNumber='" + goodsNumber + "'";
		ArrayList<Object[]> list = null;
		try {
			list = dbc.executeQuery(sql);
			if (list.size() == 0)
				((PrintWriter) response.getWriter()).print("0");
			else
				((PrintWriter) response.getWriter()).print("1");
		} catch (Exception e) {
			e.printStackTrace();
		}
	}

	private void login(String name, String pass, String role) {
		String table = "";
		if ("0".equals(role))
			table = "Customer";
		String sql = "SELECT [Password],[ID],[Name],[BlogTitle],[BlogSubTitle],[BlogStyle],[Nickname] FROM ["
				+ table + "] WHERE [LoginName]=?";
		ArrayList<Object[]> al = null;
		try {
			al = dbc.executeQuery(sql, name);
			if (al.size() == 0)
				((PrintWriter) response.getWriter()).print("-1");
			else {
				if (!pass.equals(((Object[]) al.get(0))[0].toString()))
					((PrintWriter) response.getWriter()).print("-2");
				else {
					Object[] obj = al.get(0);
					request.getSession().setAttribute("role", role);
					request.getSession()
							.setAttribute("user", obj[1].toString());
					request.getSession()
							.setAttribute("name", obj[2].toString());
					request.getSession().setAttribute("title",
							obj[3].toString().trim());
					request.getSession().setAttribute("subtitle",
							obj[4].toString().trim());
					request.getSession().setAttribute("style",
							obj[5].toString().trim());
					request.getSession().setAttribute("nickname",
							obj[6].toString().trim());
					((PrintWriter) response.getWriter()).print("0");
				}
			}
		} catch (Exception e) {
			e.printStackTrace();
		}
	}

}