package com.ibcio; 
 
import javax.servlet.annotation.WebServlet; 
import javax.servlet.http.HttpServletRequest; 
 
import org.apache.catalina.websocket.StreamInbound; 
 
@WebServlet(urlPatterns = { "/message"}) 
//如果要接收瀏覽器的ws://協(xié)議的請(qǐng)求就必須實(shí)現(xiàn)WebSocketServlet這個(gè)類 
public class WebSocketMessageServlet extends org.apache.catalina.websocket.WebSocketServlet { 
 
  private static final long serialVersionUID = 1L; 
   
  public static int ONLINE_USER_COUNT = 1; 
   
  public String getUser(HttpServletRequest request){ 
    return (String) request.getSession().getAttribute("user"); 
  } 
 
  //跟平常Servlet不同的是，需要實(shí)現(xiàn)createWebSocketInbound，在這里初始化自定義的WebSocket連接對(duì)象 
  @Override 
  protected StreamInbound createWebSocketInbound(String subProtocol,HttpServletRequest request) { 
    return new WebSocketMessageInbound(this.getUser(request)); 
  } 
} 

 package com.ibcio; 
 
import java.io.IOException; 
import java.nio.ByteBuffer; 
import java.nio.CharBuffer; 
 
import net.sf.json.JSONObject; 
 
import org.apache.catalina.websocket.MessageInbound; 
import org.apache.catalina.websocket.WsOutbound; 
 
public class WebSocketMessageInbound extends MessageInbound { 
 
  //當(dāng)前連接的用戶名稱 
  private final String user; 
 
  public WebSocketMessageInbound(String user) { 
    this.user = user; 
  } 
 
  public String getUser() { 
    return this.user; 
  } 
 
  //建立連接的觸發(fā)的事件 
  @Override 
  protected void onOpen(WsOutbound outbound) { 
    // 觸發(fā)連接事件，在連接池中添加連接 
    JSONObject result = new JSONObject(); 
    result.element("type", "user_join"); 
    result.element("user", this.user); 
    //向所有在線用戶推送當(dāng)前用戶上線的消息 
    WebSocketMessageInboundPool.sendMessage(result.toString()); 
     
    result = new JSONObject(); 
    result.element("type", "get_online_user"); 
    result.element("list", WebSocketMessageInboundPool.getOnlineUser()); 
    //向連接池添加當(dāng)前的連接對(duì)象 
    WebSocketMessageInboundPool.addMessageInbound(this); 
    //向當(dāng)前連接發(fā)送當(dāng)前在線用戶的列表 
    WebSocketMessageInboundPool.sendMessageToUser(this.user, result.toString()); 
  } 
 
  @Override 
  protected void onClose(int status) { 
    // 觸發(fā)關(guān)閉事件，在連接池中移除連接 
    WebSocketMessageInboundPool.removeMessageInbound(this); 
    JSONObject result = new JSONObject(); 
    result.element("type", "user_leave"); 
    result.element("user", this.user); 
    //向在線用戶發(fā)送當(dāng)前用戶退出的消息 
    WebSocketMessageInboundPool.sendMessage(result.toString()); 
  } 
 
  @Override 
  protected void onBinaryMessage(ByteBuffer message) throws IOException { 
    throw new UnsupportedOperationException("Binary message not supported."); 
  } 
 
  //客戶端發(fā)送消息到服務(wù)器時(shí)觸發(fā)事件 
  @Override 
  protected void onTextMessage(CharBuffer message) throws IOException { 
    //向所有在線用戶發(fā)送消息 
    WebSocketMessageInboundPool.sendMessage(message.toString()); 
  } 
} 

package com.ibcio; 
 
import java.io.IOException; 
import java.nio.CharBuffer; 
import java.util.HashMap; 
import java.util.Map; 
import java.util.Set; 
 
public class WebSocketMessageInboundPool { 
 
  //保存連接的MAP容器 
  private static final Map<String,WebSocketMessageInbound > connections = new HashMap<String,WebSocketMessageInbound>(); 
   
  //向連接池中添加連接 
  public static void addMessageInbound(WebSocketMessageInbound inbound){ 
    //添加連接 
    System.out.println("user : " + inbound.getUser() + " join.."); 
    connections.put(inbound.getUser(), inbound); 
  } 
   
  //獲取所有的在線用戶 
  public static Set<String> getOnlineUser(){ 
    return connections.keySet(); 
  } 
   
  public static void removeMessageInbound(WebSocketMessageInbound inbound){ 
    //移除連接 
    System.out.println("user : " + inbound.getUser() + " exit.."); 
    connections.remove(inbound.getUser()); 
  } 
   
  public static void sendMessageToUser(String user,String message){ 
    try { 
      //向特定的用戶發(fā)送數(shù)據(jù) 
      System.out.println("send message to user : " + user + " ,message content : " + message); 
      WebSocketMessageInbound inbound = connections.get(user); 
      if(inbound != null){ 
        inbound.getWsOutbound().writeTextMessage(CharBuffer.wrap(message)); 
      } 
    } catch (IOException e) { 
      e.printStackTrace(); 
    } 
  } 
   
  //向所有的用戶發(fā)送消息 
  public static void sendMessage(String message){ 
    try { 
      Set<String> keySet = connections.keySet(); 
      for (String key : keySet) { 
        WebSocketMessageInbound inbound = connections.get(key); 
        if(inbound != null){ 
          System.out.println("send message to user : " + key + " ,message content : " + message); 
          inbound.getWsOutbound().writeTextMessage(CharBuffer.wrap(message)); 
        } 
      } 
    } catch (IOException e) { 
      e.printStackTrace(); 
    } 
  } 
} 

<%@ page language="java" pageEncoding="UTF-8" import="com.ibcio.WebSocketMessageServlet"%> 
<% 
  String user = (String)session.getAttribute("user"); 
  if(user == null){ 
    //為用戶生成昵稱 
    user = "游客" + WebSocketMessageServlet.ONLINE_USER_COUNT; 
    WebSocketMessageServlet.ONLINE_USER_COUNT ++; 
    session.setAttribute("user", user); 
  } 
  pageContext.setAttribute("user", user); 
%> 
<html> 
<head> 
  <title>WebSocket 聊天室</title> 
  <!-- 引入CSS文件 --> 
  <link rel="stylesheet" type="text/css" href="ext4/resources/css/ext-all.css"> 
  <link rel="stylesheet" type="text/css" href="ext4/shared/example.css" /> 
  <link rel="stylesheet" type="text/css" href="css/websocket.css" /> 
   
  <!-- 映入Ext的JS開發(fā)包，及自己實(shí)現(xiàn)的webscoket. --> 
  <script type="text/javascript" src="ext4/ext-all-debug.js"></script> 
  <script type="text/javascript" src="websocket.js"></script> 
  <script type="text/javascript"> 
    var user = "${user}"; 
  </script> 
</head> 
 
<body> 
  <h1>WebSocket聊天室</h1> 
  <p>通過HTML5標(biāo)準(zhǔn)提供的API與Ext富客戶端框架相結(jié)合起來，實(shí)現(xiàn)聊天室，有以下特點(diǎn)：</p> 
  <ul class="feature-list" style="padding-left: 10px;"> 
    <li>實(shí)時(shí)獲取數(shù)據(jù)，由服務(wù)器推送，實(shí)現(xiàn)即時(shí)通訊</li> 
    <li>利用WebSocket完成數(shù)據(jù)通訊，區(qū)別于輪詢，長連接等技術(shù)，節(jié)省服務(wù)器資源</li> 
    <li>結(jié)合Ext進(jìn)行頁面展示</li> 
    <li>用戶上線下線通知</li> 
  </ul> 
  <div id="websocket_button"></div> 
</body> 
</html> 

//用于展示用戶的聊天信息 
Ext.define('MessageContainer', { 
 
  extend : 'Ext.view.View', 
 
  trackOver : true, 
 
  multiSelect : false, 
 
  itemCls : 'l-im-message', 
 
  itemSelector : 'div.l-im-message', 
 
  overItemCls : 'l-im-message-over', 
 
  selectedItemCls : 'l-im-message-selected', 
 
  style : { 
    overflow : 'auto', 
    backgroundColor : '#fff' 
  }, 
 
  tpl : [ 
      '<div class="l-im-message-warn">​交談中請(qǐng)勿輕信匯款、中獎(jiǎng)信息、陌生電話。 請(qǐng)遵守相關(guān)法律法規(guī)。</div>', 
      '<tpl for=".">', 
      '<div class="l-im-message">', 
      '<div class="l-im-message-header l-im-message-header-{source}">{from} {timestamp}</div>', 
      '<div class="l-im-message-body">{content}</div>', '</div>', 
      '</tpl>'], 
 
  messages : [], 
 
  initComponent : function() { 
    var me = this; 
    me.messageModel = Ext.define('Leetop.im.MessageModel', { 
          extend : 'Ext.data.Model', 
          fields : ['from', 'timestamp', 'content', 'source'] 
        }); 
    me.store = Ext.create('Ext.data.Store', { 
          model : 'Leetop.im.MessageModel', 
          data : me.messages 
        }); 
    me.callParent(); 
  }, 
 
  //將服務(wù)器推送的信息展示到頁面中 
  receive : function(message) { 
    var me = this; 
    message['timestamp'] = Ext.Date.format(new Date(message['timestamp']), 
        'H:i:s'); 
    if(message.from == user){ 
      message.source = 'self'; 
    }else{ 
      message.source = 'remote'; 
    } 
    me.store.add(message); 
    if (me.el.dom) { 
      me.el.dom.scrollTop = me.el.dom.scrollHeight; 
    } 
  } 
}); 

  Ext.onReady(function() { 
      //創(chuàng)建用戶輸入框 
      var input = Ext.create('Ext.form.field.HtmlEditor', { 
            region : 'south', 
            height : 120, 
            enableFont : false, 
            enableSourceEdit : false, 
            enableAlignments : false, 
            listeners : { 
              initialize : function() { 
                Ext.EventManager.on(me.input.getDoc(), { 
                      keyup : function(e) { 
                        if (e.ctrlKey === true 
                            && e.keyCode == 13) { 
                          e.preventDefault(); 
                          e.stopPropagation(); 
                          send(); 
                        } 
                      } 
                    }); 
              } 
            } 
          }); 
      //創(chuàng)建消息展示容器 
      var output = Ext.create('MessageContainer', { 
            region : 'center' 
          }); 
 
      var dialog = Ext.create('Ext.panel.Panel', { 
            region : 'center', 
            layout : 'border', 
            items : [input, output], 
            buttons : [{ 
                  text : '發(fā)送', 
                  handler : send 
                }] 
          }); 
      var websocket; 
 
      //初始話WebSocket 
      function initWebSocket() { 
        if (window.WebSocket) { 
          websocket = new WebSocket(encodeURI('ws://localhost:8080/WebSocket/message')); 
          websocket.onopen = function() { 
            //連接成功 
            win.setTitle(title + ' (已連接)'); 
          } 
          websocket.onerror = function() { 
            //連接失敗 
            win.setTitle(title + ' (連接發(fā)生錯(cuò)誤)'); 
          } 
          websocket.onclose = function() { 
            //連接斷開 
            win.setTitle(title + ' (已經(jīng)斷開連接)'); 
          } 
          //消息接收 
          websocket.onmessage = function(message) { 
            var message = JSON.parse(message.data); 
            //接收用戶發(fā)送的消息 
            if (message.type == 'message') { 
              output.receive(message); 
            } else if (message.type == 'get_online_user') { 
              //獲取在線用戶列表 
              var root = onlineUser.getRootNode(); 
              Ext.each(message.list,function(user){ 
                var node = root.createNode({ 
                  id : user, 
                  text : user, 
                  iconCls : 'user', 
                  leaf : true 
                }); 
                root.appendChild(node); 
              }); 
            } else if (message.type == 'user_join') { 
              //用戶上線 
                var root = onlineUser.getRootNode(); 
                var user = message.user; 
                var node = root.createNode({ 
                  id : user, 
                  text : user, 
                  iconCls : 'user', 
                  leaf : true 
                }); 
                root.appendChild(node); 
            } else if (message.type == 'user_leave') { 
                //用戶下線 
                var root = onlineUser.getRootNode(); 
                var user = message.user; 
                var node = root.findChild('id',user); 
                root.removeChild(node); 
            } 
          } 
        } 
      }; 
 
      //在線用戶樹 
      var onlineUser = Ext.create('Ext.tree.Panel', { 
            title : '在線用戶', 
            rootVisible : false, 
            region : 'east', 
            width : 150, 
            lines : false, 
            useArrows : true, 
            autoScroll : true, 
            split : true, 
            iconCls : 'user-online', 
            store : Ext.create('Ext.data.TreeStore', { 
                  root : { 
                    text : '在線用戶', 
                    expanded : true, 
                    children : [] 
                  } 
                }) 
          }); 
      var title = '歡迎您：' + user; 
      //展示窗口 
      var win = Ext.create('Ext.window.Window', { 
            title : title + ' (未連接)', 
            layout : 'border', 
            iconCls : 'user-win', 
            minWidth : 650, 
            minHeight : 460, 
            width : 650, 
            animateTarget : 'websocket_button', 
            height : 460, 
            items : [dialog,onlineUser], 
            border : false, 
            listeners : { 
              render : function() { 
                initWebSocket(); 
              } 
            } 
          }); 
 
      win.show(); 
 
      //發(fā)送消息 
      function send() { 
        var message = {}; 
        if (websocket != null) { 
          if (input.getValue()) { 
            Ext.apply(message, { 
                  from : user, 
                  content : input.getValue(), 
                  timestamp : new Date().getTime(), 
                  type : 'message' 
                }); 
            websocket.send(JSON.stringify(message)); 
            //output.receive(message); 
            input.setValue(''); 
          } 
        } else { 
          Ext.Msg.alert('提示', '您已經(jīng)掉線，無法發(fā)送消息!'); 
        } 
      } 
    });