controlservlet.java

一个小公司要求给写的很简单的任务管理系统。

package com.wykj.control;

import javax.servlet.*;
import javax.servlet.http.*;
import java.io.*;
import java.util.*;
import com.wykj.util.*;
import com.wykj.buss.*;
import com.wykj.model.UserModel;
import com.wykj.model.RwbModel;
import java.text.SimpleDateFormat;
import java.text.ParseException;

/**
 * <p>Title: </p>
 * <p>Description: </p>
 * <p>Copyright: Copyright (c) 2007</p>
 * <p>Company: </p>
 * @author not attributable
 * @version 1.0
 */

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

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

  //Process the HTTP Get request
  public void doGet(HttpServletRequest request, HttpServletResponse response) throws
      ServletException, IOException {
    String action = Util.trimNull(request.getParameter("action"));
    String url = "";
    if (action.equals("listtask")) {
      url = doListTask(request, response);
    }
    else if (action.equals("addTask")) {
      url = toAddTask(request, response);
    }
    else if (action.equals("login")) {
      url = doLogin(request, response);
    }
    else if (action.equals("saveTask")) {
      url = doSaveTask(request, response);
    }
    else if (action.equals("delTask")) {
      url = doDelTask(request, response);
    }

    else if (action.equals("updatetask")) {
      url = doUpdateTask(request, response);
    }
    else if (action.equals("logout")) {
      url = doLogout(request, response);
    }
    else {
      this.doListTask(request, response);
      url = "/taskList.jsp";
    }
    RequestDispatcher rd = getServletContext().getRequestDispatcher(url);
    try {
      rd.forward(request, response);
    }
    catch (IOException ex) {
    }
    catch (ServletException ex) {
    }

  }

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

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

  public String doListTask(HttpServletRequest request,
                           HttpServletResponse response) {
    String url = "/taskList.jsp";
    String addUser =  this.getparameter(request,"addName");
    String rq =this.getparameter(request,"adddate");
    String rqE = Util.trimNull(request.getParameter("adddateE"));
    ArrayList taskarray = new BussTaskManager().getTaskListSort(addUser, rq,rqE);
    request.removeAttribute("taskarray");
    request.setAttribute("taskarray", taskarray);
    return url;
  }

  public String doLogin(HttpServletRequest request,
                        HttpServletResponse response) {
    String info = "";
    String userName = this.getparameter(request,"userName");
    String password = Util.trimNull(request.getParameter("password"));
   // userName = Util.dbUnicodeToCh(userName);
    BussUserManager bm = new BussUserManager();
    ArrayList array = bm.getUserList(userName, password);

    if (array.size() > 0) {
      info = " 登录成功 ，欢迎您 " + userName;
      request.getSession().setAttribute("LOGIN_USER", array.get(0));
    }
    else {
      info = " 登录失败 ，请确认用户名和密码正确 ! ";
    }

    ArrayList taskarray = array = new BussTaskManager().getTaskList(userName, Util.getToday(),Util.getToday());
    request.setAttribute("taskarray", taskarray);
    request.setAttribute("INFO", info);
    return "/taskList.jsp";

  }

  public String doUpdateTask(HttpServletRequest request,
                             HttpServletResponse response) {
    String info = "";

    return info;
  }

  public String toAddTask(HttpServletRequest request,
                          HttpServletResponse response) {
    UserModel um = (UserModel) request.getSession().getAttribute("LOGIN_USER");
    if (um == null) {
      request.setAttribute("INFO", "您登录停留时间过长，请重新登录！");
      return "/taskList.jsp";
    }
    String userName = um.getUserName();
    String rq = Util.trimNull(request.getParameter("adddate"));
     String rqE = Util.trimNull(request.getParameter("adddateE"));
    if ("3".equals(um.getFlag())) {
      userName = "";
    }
    ArrayList array = new BussTaskManager().getTaskList(userName, rq,rqE);
    request.setAttribute("taskUpdate", array);
    return "/taskAdd.jsp";
  }

  public String doLogout(HttpServletRequest request,
                         HttpServletResponse response) {
    request.getSession().removeAttribute("LOGIN_USER");
    return "/taskList.jsp";
  }

  public String doSaveTask(HttpServletRequest request,
                           HttpServletResponse response) {
    //request.getSession().removeAttribute("LOGIN_USER");
    ArrayList array = new ArrayList();
    RwbModel rm = new RwbModel();
    rm.setId(getparameter(request, "id"));
    rm.setRqStr(getparameter(request, "rq"));
    rm.setAddUser(getparameter(request, "addUser"));
    rm.setTaskContent(getparameter(request, "taskContent"));
    rm.setTaskEndR(getparameter(request, "taskEndR"));
    rm.setTaskEff(getparameter(request, "taskEff"));
    rm.setTaskEnd(getparameter(request, "taskEnd"));
    rm.setTaskCheck(getparameter(request, "taskCheck"));
    rm.setTaskResult(getparameter(request, "taskResult"));
    rm.setPrize(getparameter(request, "prize"));
    rm.setReportType(getparameter(request, "reportType"));
    rm.setOtherfactor(getparameter(request, "otherfactor"));
    array.add(rm);

    if (rm.getId().length() >0){
      new BussTaskManager().updateTasks(array);
    }else {
      new BussTaskManager().addTasks(array);
    }

    this.toAddTask(request,response);
    return "/taskAdd.jsp";
  }
  public String doDelTask(HttpServletRequest request,
                           HttpServletResponse response) {
    //request.getSession().removeAttribute("LOGIN_USER");
    String id = getparameter(request, "id");
    new BussTaskManager().delTask(id);
    this.toAddTask(request,response);
    return "/taskAdd.jsp";
  }

  private String getparameter(HttpServletRequest request, String param) {
    String text = request.getParameter(param);
    text = Util.trimNull(text);
    text = Util.dbUnicodeToCh(text);
    return text;
  }

}