HTML5 WebSocket實作點對點聊天的範例程式碼

昨天使用HTML5的websocket與Tomcat實現了多人聊天，那是最簡單也是最基本的，其中註意的就是開發環境，要滿足jdk1.7和tomcat8，當然了tom7 的7.063也行！

今天是國慶日的最後一天，苦逼的加班，繼續搞程式碼！令人欣慰的是，我用google找到了關於websocket的點對點聊天，更好的是可以和大多數系統很好的配合起來看下效果圖

因為是模擬的，這裡給的是兩個JSP頁面A和B，裡面分別往session裡放了兩個名字小明和小化，注意，這裡的session是HttpSession session，之前多人聊天裡的session是javax.websocket.Session;不一樣的。

這裡想一下， 使用HttpSession session控制聊天的用戶，好處怎樣，自己猜~~~

這裡沒有使用註解，傳統的web.xml設定方式，先在系統啟動的時候呼叫InitServlet方法

public class InitServlet extends HttpServlet { private static final long serialVersionUID = -3163557381361759907L; private static HashMap<String,MessageInbound> socketList publicvate static HashMap<String,MessageInbound> s. InitServlet.socketList = new HashMap<String,MessageInbound>(); super.init(config); System.out.println(初始化聊天容器); } public static HashMap<String,MessageInbound> getSocketList() { return InitServlet.socketList; } }

這裡你可以跟自己的系統結合，對應的web設定碼如下：

 <?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>websocket</servlet-name> <servlet-class>socket.MyWebSocketServlet</servlet-class > </servlet> <servlet-mapping> <servlet-name>websocket</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>

這就是最普通的前台像後台發送請求的過程，也是很容易嵌入到自己的系統裡

MyWebSocketServlet：

 public class MyWebSocketServlet extends WebSocketServlet { public String getUser(HttpServletRequest request){ String userName = (String) request.getSession().getAttribute(user); if(userName==null){ return noundull; (String arg0, HttpServletRequest request) { System.out.println(用戶+ request.getSession().getAttribute(user) + 登入); return new MyMessageInbound(this.getUser(request)); }}

MyMessageInbound繼承MessageInbound

 package socket;import java.io.IOException;import java.nio.ByteBuffer;import java.nio.CharBuffer;import java.util.HashMap;import org.apache.catalina.websocket.MessageInboundache;import org. .WsOutbound;import util.MessageUtil;public class MyMessageInbound extends MessageInbound { private String name; public MyMessageInbound() { super(); } public MyMessageInbound(String name) { super(); this.name = name; ) throws IOException { } @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 MessageInbound messageFromInbound = userMsgMap.get(fromName); if(messageInbound!=null && messageFromInbound!=null){ //如果發往人存在進行操作WsOutbound outbound = messageInbound.getWsOutbound(); content = messageMap.get(content); //取得訊息內容String msgContentString = fromName + 說: + content; //建構發送的訊息//發出去內容CharBuffer toMsg = CharBuffer.wrap(msgContentString.toCharArray()); CharBuffer fromMsg = CharBuffer.wrap(msgContentString.toCharArray()); outFromBound.writeTextMessage(fromMsg); outbound.writeTextMessage(toMsg); // outFromBound.flush(); outbound.flush(); } @Override protected void onClose(int status) { InitServlet ; super.onClose(status); } @Override protected void onOpen(WsOutbound outbound) { super.onOpen(outbound); //登入的使用者註冊進去if(name!=null){ InitServlet.getSocketList().put(name, this);//存放客服ID與使用者} } @Override public int getReadTimeout() { return 0; } }

在onTextMessage中處理前台發出的訊息，並封裝訊息傳給目標

還有一個messageutil

 package util;import java.nio.CharBuffer;import java.util.HashMap;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[0]); map.put(toName, m[1]); map.put(content, m[2]); return map; }}

當然了，前台也要依照規定的格式傳遞訊息

<%@ 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-1.7.2.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/WebSocketUser/websocket.do); else if ('MozWebSocket' in window) ws = new MozWebSocket(ws://localhost:8080/WebSocketUser/websocket.WebSocket(ws://localhost:8080/WebSocketUser/websocket.do); support); ws.onmessage = function(evt) { //alert(evt.data); console.log(evt); // $(#xiaoxi).val(evt.data); setMessageInnerHTML(evt.data); }; function setMessageInnerHTML(innerHTML){ document.getElementByunction setMessageInnerHTML(innerHTML){ document.getElementByunction setMessageInnerHTML(innerHTML){ document.getElementByunction( 'message').innerHTML += innerHTML + '<br/>'; } 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>聊天框：<div id=message style=height: 250px;width: 280px;border: 1px solid; overflow: auto;></div><br><input type= button value=send onclick=sendMsg()></input></body></html>

這是A.jsp頁面，B同上

透過以上程式碼，就可以實現一個點對點的聊天功能，如果做的大，可以做成一個web版的聊天系統，包括聊天室和單人聊天，都說websocket不支援二進位的傳輸，但看到個大流說了這樣的話

不過現在做下來感覺使用二元化的意義不是很大。很久以前就一直困混，怎麼都說JS不支援二進制，發現其實只是一堆坑貨對這個沒研究。 。 （用的是filereader）

以上就是本文的全部內容，希望對大家的學習有所幫助，也希望大家多多支持VeVb武林網。