使用Ajax轮询模拟简单的站内信箱(消息管理)功能

前一段时间项目需要写一个类似于站内信箱的消息管理的功能，由于对前端不是很熟悉，刚开始不知道怎么做，后来看了网上的方案，现模拟一个非常简单的消息管理。

我们首先看一下最终效果的样式，就是非常简单的一个样子，如下所示：

我这边暂时只写了三个类型的消息，订单、上下架、审核。不同的消息推送过来会显示在不同类型上，消息管理上显示的是三种类型的消息总数量。

好，接下来介绍一下代码。首先是数据库的设计，共有七个字段，分别是一个自增字段id，消息类型type，消息标题title，消息具体内容detail，消息状态status，创建时间time以及用户userId。如下所示：

然后我们看推送消息的后台代码PullMsgServlet，分别有四个方法，分别是获取所有消息的历史记录，获取某人的未读消息，获取未读消息的总数，还有获取某种类型的未读消息总数。

 protected void doPost(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
         String operateType = Base64Util.decode(request.getParameter("operateType"));
         msgDao = new MessageDao();
         switch (operateType) {
             case "showHistory":
                 showHistory(request, response);
                 break;
             case "showPending":
                 showPending(request, response);
                 break;
             case "showCount":
                 showCount(request,response);
                 break;
             case "showCountByType":
                 showCountByType(request, response);
                 break;
             default:
                 break;
         }
         response.setCharacterEncoding("utf-8");
         response.setContentType("text/html;charset=UTF-8");
         response.getWriter().print(Base64Util.encode(result.toString()));
     }

     /**
      * 所有消息的历史记录
      * @param request
      * @param response
      * @throws ServletException
      * @throws IOException
      */
     private void showHistory(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException{
         HttpSession session = request.getSession();
         userId = session.getAttribute("userId").toString();
         List<Object> historyList = msgDao.getMsgByUserId(userId);
         HttpResultList result = new HttpResultList();
         result.setResult(historyList.size()>0);
         result.setDatas(historyList);
     }

     /**
      * 某人的未读消息
      * @param request
      * @param response
      * @throws ServletException
      * @throws IOException
      */
     private void showPending(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException{
         HttpSession session = request.getSession();
         userId = session.getAttribute("userId").toString();
         List<Object> pendingList = msgDao.getUnreadMsg();
         HttpResultList result = new HttpResultList();
         result.setResult(pendingList.size()>0);
         result.setDatas(pendingList);
     }

     /**
      * 某人未读消息的总数
      * @param request
      * @param response
      * @throws ServletException
      * @throws IOException
      */
     private void showCount(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException{
         String number = msgDao.getMsgNum()+"";
         result = new JSONObject();
         result.put("result", number);
     }

     /**
      * 某人某种类型的未读消息总数
      * @param request
      * @param response
      * @throws ServletException
      * @throws IOException
      */
     private void showCountByType(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException{
         Map<String, Integer> map = msgDao.getTypeNum();
         //获得返回的type
         List<String> keyList = new ArrayList<>();
         for (String key : map.keySet()) {
             keyList.add(key);
         }
         //获得返回type的个数
         List<Integer> valueList = new ArrayList<>();
         for (Integer value : map.values()) {
             valueList.add(value);
         }
         result = new JSONObject();
         JSONObject content = new JSONObject();
         content.put("type", JSONArray.fromObject(keyList));
         content.put("typeNum", JSONArray.fromObject(valueList));
         result.put("result", content);
     }

接下来我们看前端代码，这边使用了setInterval定时器，设置的js代码每５秒轮询后台一次，请求后台servlet从数据库里面获取的数据，如下所示：

 setInterval(function(){
     getType();
 },5000);

 /**
  * 得到某个类型的消息数量
  */
 function getType(){
     var lis = document.getElementById("type").getElementsByTagName("li");
     var msgType = "";
     for(var i=0; i<lis.length; i++){
         msgType = lis[i].innerText;
     }
     $.ajax({
         type:"POST",
         url:"PullMsgServlet",
         dataType:'json',
         processData:false,
         data: {
             operateType: "showCountByType"
         },
         success:function(data){
             var result = data.result;
             var type = "";
             if(result){
                 for(var i=0; i<result.type.length; i++){
                     type = result.type[i];
                     if(type == "订单"){
                         $("#order").html(result.typeNum[i]);
                     }else if(type == "上下架"){
                         $("#shelf").html(result.typeNum[i]);
                     }else if(type == "审核"){
                         $("#review").html(result.typeNum[i]);
                     }
                 }
             }
         },
         error: function () {
             // alert("error...");
         }
     });
 }

这样前端每隔５秒就轮询一次PullMsgServlet，查询数据库里是否有未读的消息，如果有，就显示在前端页面上。这样，一个最简单的消息推送以及显示功能就有了，只是我还没有写前端历史列表的查看，等以后写了再更新。。。。