/*** Intended to be overridden by sub-classes that wish to be notified* when the outbound connection is established. The default implementation* is a NO-OP.** @param outbound    The outbound WebSocket connection.*/protected void onOpen(WsOutbound outbound) {// NO-OP}/*** Intended to be overridden by sub-classes that wish to be notified* when the outbound connection is closed. The default implementation* is a NO-OP.** @param status    The status code of the close reason.*/protected void onClose(int status) {// NO-OP}/*** This method is called when there is a binary WebSocket message available* to process. The message is presented via a stream and may be formed from* one or more frames. The number of frames used to transmit the message is* not made visible to the application.** @param is    The WebSocket message** @throws IOException  If a problem occurs processing the message. Any*                      exception will trigger the closing of the WebSocket*                      connection.*/protected abstract void onBinaryData(InputStream is) throws IOException;/*** This method is called when there is a textual WebSocket message available* to process. The message is presented via a reader and may be formed from* one or more frames. The number of frames used to transmit the message is* not made visible to the application.** @param r     The WebSocket message** @throws IOException  If a problem occurs processing the message. Any*                      exception will trigger the closing of the WebSocket*                      connection.*/protected abstract void onTextData(Reader r) throws IOException;

@WebServlet(urlPatterns = { "/chatWebSocket" })
public class ChatWebSocketServlet extends WebSocketServlet {private static final long serialVersionUID = 1L;OnLineUser theUser;@Overrideprotected void doGet(HttpServletRequest req, HttpServletResponse resp)throws ServletException, IOException {theUser = (OnLineUser) req.getSession().getAttribute("loginUser");super.doGet(req, resp);}@Overrideprotected StreamInbound createWebSocketInbound(String subProtocol) {return new ChatMessageInbound(theUser);}}

@Overrideprotected void doGet(HttpServletRequest req, HttpServletResponse resp)throws ServletException, IOException {// Information required to send the server handshake messageString key;String subProtocol = null;List<String> extensions = Collections.emptyList();if (!headerContainsToken(req, "upgrade", "websocket")) {resp.sendError(HttpServletResponse.SC_BAD_REQUEST);return;}if (!headerContainsToken(req, "connection", "upgrade")) {resp.sendError(HttpServletResponse.SC_BAD_REQUEST);return;}if (!headerContainsToken(req, "sec-websocket-version", "13")) {resp.setStatus(426);resp.setHeader("Sec-WebSocket-Version", "13");return;}key = req.getHeader("Sec-WebSocket-Key");if (key == null) {resp.sendError(HttpServletResponse.SC_BAD_REQUEST);return;}String origin = req.getHeader("Origin");if (!verifyOrigin(origin)) {resp.sendError(HttpServletResponse.SC_FORBIDDEN);return;}List<String> subProtocols = getTokensFromHeader(req,"Sec-WebSocket-Protocol-Client");if (!subProtocols.isEmpty()) {subProtocol = selectSubProtocol(subProtocols);}// TODO Read client handshake - Sec-WebSocket-Extensions// TODO Extensions require the ability to specify something (API TBD)//      that can be passed to the Tomcat internals and process extension//      data present when the frame is fragmented.// If we got this far, all is good. Accept the connection.resp.setHeader("upgrade", "websocket");resp.setHeader("connection", "upgrade");resp.setHeader("Sec-WebSocket-Accept", getWebSocketAccept(key));if (subProtocol != null) {resp.setHeader("Sec-WebSocket-Protocol", subProtocol);}if (!extensions.isEmpty()) {// TODO}// Small hack until the Servlet API provides a way to do this.StreamInbound inbound = createWebSocketInbound(subProtocol);((RequestFacade) req).doUpgrade(inbound);}

@Overrideprotected StreamInbound createWebSocketInbound(String subProtocol) {return new ChatMessageInbound(theUser);}

public final class ChatMessageInbound extends MessageInbound {public ChatMessageInbound(OnLineUser theUser) {this.theUser = theUser;}@Overrideprotected void onOpen(WsOutbound outbound) {// 添加链接到容器ChatMessageInbound theBound = this;ChatContainer.addInbound(theBound.theUser, theBound);// 向每个在线用户发送消息ChatContainer.eachAllBound(new ContainerCallBack() {@Overridepublic void eachCallBack(ChatMessageInbound theBound, OnLineUser theUser) {ListUserMsg listUserMsg = new ListUserMsg(ChatContainer.getUserList());WriteTookit.writeToBound(theBound, listUserMsg.toMsg());}});}@Overrideprotected void onClose(int status) {ChatContainer.removeInbound(theUser);}@Overrideprotected void onBinaryMessage(ByteBuffer message) throws IOException {}@Overrideprotected void onTextMessage(CharBuffer message) throws IOException {
//      CHAT_MODEL.setMessage(message.toString());
//      ChatContainer.eachAllBound(new ContainerCallBack() {
//          @Override
//          public void eachCallBack(ChatMessageInbound theBound, OnLineUser theUser) {
//              WriteTookit.writeToBound(theBound, CHAT_MODEL.getSayMsg());
//          }
//      });}// 变量区域private OnLineUser theUser;
}

public final class ChatContainer {/*** 保存服务器连接的用户的容器*/private static final Map<OnLineUser, ChatMessageInbound> CHAT_MAP = new HashMap<OnLineUser, ChatMessageInbound>();/*** 取出用户的连接*/public static ChatMessageInbound getInbound(OnLineUser theUser) {return CHAT_MAP.get(theUser);}/*** 放入一个连接*/public static void addInbound(OnLineUser theUser,ChatMessageInbound outbound) {CHAT_MAP.put(theUser, outbound);System.out.println(CHAT_MAP.size());}/*** 移除一个连接* * @param theUser* @return*/public static ChatMessageInbound removeInbound(OnLineUser theUser) {return CHAT_MAP.remove(theUser);}/*** 遍历所有连接*/public static void eachAllBound(ContainerCallBack callBackInter) {Iterator<OnLineUser> keyIter = CHAT_MAP.keySet().iterator();while (keyIter.hasNext()) {OnLineUser theUser = keyIter.next();callBackInter.eachCallBack(CHAT_MAP.get(theUser), theUser);}}/*** 回调函数的接口* * @author WangZhenChong*/public interface ContainerCallBack {void eachCallBack(ChatMessageInbound theBound, OnLineUser theUser);}}

/*** 前台和后台交互的信息类型常量* * @author WangZhenChong* */
public final class MsgTypeConstants {public static short GET_USER_LIST = 1;// 在线所有用户信息交互public static short SEND_ONE_TO_ONE = 2;// 对一个用户发送消息public static short SEND_ONE_TO_ALL = 3;// 对所有用户发送消息public static short SEND_SERVER_MSG = 4;// 发送系统消息
}

$(document).ready(function() {$("#connBtn").bind('click', function() {$.ajax({url : "/tomcatWebSocket/Login#?asdasdasd",type : "POST",processData : false,data : $.param({username : document.getElementById("usernameField").value}),success : function(msg, status) {initChat();initUserList();$("#sendBtn").removeAttr("disabled");$("#connBtn").attr("disabled", "disabled");$("#usernameField").attr("disabled", "disabled");},error : function(jqXHR, textStatus, errorThrown) {alert("服务器内部错误");}});});var Chat = {};
Chat.socket = null;
function initChat() {var wsURL = 'ws://' + window.location.host+ '/tomcatWebSocket/chatWebSocket';if ('WebSocket' in window) {Chat.socket = new WebSocket(wsURL);} else if ('MozWebSocket' in window) {Chat.socket = new MozWebSocket(wsURL);} else {alert("浏览器不支持");return false;}Chat.socket.onopen = function() {};Chat.socket.onclose = function() {Chat.writeToConsole("断开连接了 ");initChat();};Chat.socket.onmessage = function(message) {if (typeof message.data == "string") {// 如果发送的是字符串信息.var msgObj = eval("(" + message.data + ")");switch (msgObj.MsgType) {case MsgTypeConstants.GET_USER_LIST :// 所有用户信息Chat.preUserList(msgObj.userList);break;case MsgTypeConstants.SEND_ONE_TO_ALL :Chat.writeToConsole(msgObj.msgContext);break;default :alert("未知错误，请刷新页面");}}};Chat.sendMessage = function() {Chat.socket.send(ueditor.getContentTxt());};
}Chat.writeToConsole = function(message) {
//往控制台打印得到的聊天消息
};/*** 处理刷新用户信息的方法。*/
Chat.preUserList = function(userList) {//用户信息列表};