使用JavaWeb webSocket实现简易的点对点聊天功能实例代码

package socket; import java.nio.CharBuffer; import java.util.ArrayList; import java.util.HashMap; import java.util.List; import javax.servlet.ServletConfig; import javax.servlet.ServletException; import javax.servlet.http.HttpServlet; import org.apache.catalina.websocket.MessageInbound; public class InitServlet extends HttpServlet {   private static final long serialVersionUID = -L;    //private static List<MessageInbound> socketList;     private static HashMap<String,MessageInbound> socketList;     public void init(ServletConfig config) throws ServletException {   //    InitServlet.socketList = new ArrayList<MessageInbound>();       InitServlet.socketList = new HashMap<String,MessageInbound>();       super.init(config);       System.out.println("Server start============");     }     public static HashMap<String,MessageInbound> getSocketList() {       return InitServlet.socketList;     }   /*  public static List<MessageInbound> getSocketList() {       return InitServlet.socketList;     }   */}

package socket; import java.io.IOException; import java.io.PrintWriter; import java.nio.CharBuffer; import javax.servlet.ServletException; import javax.servlet.http.HttpServlet; import javax.servlet.http.HttpServletRequest; import javax.servlet.http.HttpServletResponse; import org.apache.catalina.websocket.StreamInbound; import org.apache.catalina.websocket.WebSocketServlet; /** *  * @ClassName: MyWebSocketServlet  * @Description: 建立连接时创立  * @author mangues * @date -- */ public class MyWebSocketServlet extends WebSocketServlet {   public String getUser(HttpServletRequest request){      String userName = (String) request.getSession().getAttribute("user");     if(userName==null){       return null;     }     return userName;     // return (String) request.getAttribute("user");    }    @Override   protected StreamInbound createWebSocketInbound(String arg,       HttpServletRequest request) {     System.out.println("##########");      return new MyMessageInbound(this.getUser(request));    } } 

package socket; import java.io.IOException; import java.nio.ByteBuffer; import java.nio.CharBuffer; import java.util.HashMap; import org.apache.catalina.websocket.MessageInbound; import org.apache.catalina.websocket.WsOutbound; import util.MessageUtil; public class MyMessageInbound extends MessageInbound {   private String name;   public MyMessageInbound() {     super();   }   public MyMessageInbound(String name) {     super();     this.name = name;   }   @Override    protected void onBinaryMessage(ByteBuffer arg) throws IOException {      // TODO Auto-generated method stub    }    @Override    protected void onTextMessage(CharBuffer msg) throws IOException {      //用户所发消息处理后的map     HashMap<String,String> messageMap = MessageUtil.getMessage(msg);  //处理消息类     //上线用户集合类map     HashMap<String, MessageInbound> userMsgMap = InitServlet.getSocketList();     String fromName = messageMap.get("fromName");  //消息来自人 的userId     String toName = messageMap.get("toName");     //消息发往人的 userId     //获取该用户     MessageInbound messageInbound = userMsgMap.get(toName);  //在仓库中取出发往人的MessageInbound     if(messageInbound!=null){   //如果发往人 存在进行操作       WsOutbound outbound = messageInbound.getWsOutbound();        String content = messageMap.get("content"); //获取消息内容       String msgContentString = fromName + "   " + content;  //构造发送的消息       //发出去内容       CharBuffer toMsg = CharBuffer.wrap(msgContentString.toCharArray());       outbound.writeTextMessage(toMsg); //       outbound.flush();     }    /* for (MessageInbound messageInbound : InitServlet.getSocketList()) {        CharBuffer buffer = CharBuffer.wrap(msg);        WsOutbound outbound = messageInbound.getWsOutbound();        outbound.writeTextMessage(buffer);        outbound.flush();      } */   }    @Override    protected void onClose(int status) {      InitServlet.getSocketList().remove(this);      super.onClose(status);    }    @Override    protected void onOpen(WsOutbound outbound) {      super.onOpen(outbound);      //登录的用户注册进去     if(name!=null){       InitServlet.getSocketList().put(name, this);      } //    InitServlet.getSocketList().add(this);    }  }

 package util; import java.nio.CharBuffer; import java.util.HashMap; /**  *   * @ClassName: MessageUtil   * @Description: 消息处理类  * @author mangues * @date -- */ public class MessageUtil {   public static HashMap<String,String> getMessage(CharBuffer msg) {     HashMap<String,String> map = new HashMap<String,String>();     String msgString = msg.toString();     String m[] = msgString.split(",");     map.put("fromName", m[]);     map.put("toName", m[]);     map.put("content", m[]);     return map;   } }

<?xml version="1.0" encoding="UTF-8"?><web-app version="3.0"   xmlns="http://java.sun.com/xml/ns/javaee"   xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"   xsi:schemaLocation="http://java.sun.com/xml/ns/javaee   http://java.sun.com/xml/ns/javaee/web-app_3_0.xsd"> <servlet>   <servlet-name>mywebsocket</servlet-name>   <servlet-class>socket.MyWebSocketServlet</servlet-class>  </servlet>  <servlet-mapping>   <servlet-name>mywebsocket</servlet-name>   <url-pattern>*.do</url-pattern>  </servlet-mapping>  <servlet>   <servlet-name>initServlet</servlet-name>   <servlet-class>socket.InitServlet</servlet-class>   <load-on-startup>1</load-on-startup>  </servlet>  <welcome-file-list>  <welcome-file>index.jsp</welcome-file> </welcome-file-list></web-app>

<%@ page language="java" contentType="text/html; charset=UTF-"   pageEncoding="UTF-"%> <!DOCTYPE html> <html> <head> <meta http-equiv="Content-Type" content="text/html; charset=UTF-"> <title>Index</title> <script type="text/javascript" src="js/jquery ...min.js"></script> <%session.setAttribute("user", "小化");%> <script type="text/javascript"> var ws = null; function startWebSocket() {   if ('WebSocket' in window)     ws = new WebSocket("ws://localhost:/webSocket/mywebsocket.do");   else if ('MozWebSocket' in window)     ws = new MozWebSocket("ws://localhost:/webSocket/mywebsocket.do");   else     alert("not support");   ws.onmessage = function(evt) {     //alert(evt.data);     console.log(evt);     $("#xiaoxi").val(evt.data);   };   ws.onclose = function(evt) {     //alert("close");     document.getElementById('denglu').innerHTML="离线";   };   ws.onopen = function(evt) {     //alert("open");     document.getElementById('denglu').innerHTML="在线";     document.getElementById('userName').innerHTML='小化';   }; } function sendMsg() {   var fromName = "小化";   var toName = document.getElementById('name').value; //发给谁   var content = document.getElementById('writeMsg').value; //发送内容   ws.send(fromName+","+toName+","+content); } </script> </head> <body onload="startWebSocket();"> <p>聊天功能实现</p> 登录状态： <span id="denglu" style="color:red;">正在登录</span> <br> 登录人： <span id="userName"></span> <br> <br> <br> 发送给谁：<input type="text" id="name" value="小明"></input> <br> 发送内容：<input type="text" id="writeMsg"></input> <br> 聊天框：<textarea rows="" cols="" readonly id="xiaoxi"></textarea> <br> <input type="button" value="send" onclick="sendMsg()"></input> </body> </html>

<%@ page language="java" contentType="text/html; charset=UTF-8"  pageEncoding="UTF-8"%><!DOCTYPE html><html><head><meta http-equiv="Content-Type" content="text/html; charset=UTF-8"><title>Index</title><script type="text/javascript" src="js/jquery 2.1.1.min.js"></script><%session.setAttribute("user", "小明");%><script type="text/javascript">var ws = null;function startWebSocket() {  if ('WebSocket' in window)    ws = new WebSocket("ws://localhost:8080/webSocket/mywebsocket.do");  else if ('MozWebSocket' in window)    ws = new MozWebSocket("ws://localhost:8080/webSocket/mywebsocket.do");  else    alert("not support");  ws.onmessage = function(evt) {    console.log(evt);    //alert(evt.data);    $("#xiaoxi").val(evt.data);  };  ws.onclose = function(evt) {    //alert("close");    document.getElementById('denglu').innerHTML="离线";  };  ws.onopen = function(evt) {    //alert("open");    document.getElementById('denglu').innerHTML="在线";    document.getElementById('userName').innerHTML="小明";  };}function sendMsg() {  var fromName = "小明";  var toName = document.getElementById('name').value; //发给谁  var content = document.getElementById('writeMsg').value; //发送内容  ws.send(fromName+","+toName+","+content);}</script></head><body onload="startWebSocket();"><p>聊天功能实现</p>登录状态：<span id="denglu" style="color:red;">正在登录</span><br>登录人：<span id="userName"></span><br><br><br>发送给谁：<input type="text" id="name" value="小化"></input><br>发送内容：<input type="text" id="writeMsg"></input><br>聊天框：<textarea rows="13" cols="100" readonly id="xiaoxi"></textarea><br><input type="button" value="send" onclick="sendMsg()"></input></body></html>