首先給大家聲明一點：需要 jdk 7 ， tomcat需要支持websocket的版本
1.InitServlet
該類主要是用來初始化構造將來存儲用戶身份信息的map倉庫，利用其初始化方法Init 初始化倉庫， 利用其靜態方法getSocketList 獲得對應的用戶身份信息。
webSocket ，我認為MessageInbound 用來識別登錄人的信息，用它來找到對應的人，推送消息。每次登錄都會產生一個MessageInbound。
這里的 HashMap<String,MessageInbound> ：string 存儲用戶session的登錄id,MessageInbound存儲 推送需要的身份信息。以上屬于個人口頭話理解。
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
|
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; } */} |
2.MyWebSocketServlet
websocket用來建立連接的servlet，建立連接時，首先在session獲取該登錄人的userId，在調用MyMessageInbound構造函數傳入userId
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
|
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)); } } |
3.onOpen方法調用InitServlet的map身份倉庫，
放入用戶userId 和 對應該登錄用戶的websocket身份信息MessageInbound （可以用userId來尋找到推送需要的 身份MessageInbound ）
onTextMessage ：用來獲取消息，并發送消息
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
|
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); } } |
4.消息處理類，處理前端發來的消息
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
|
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; }} |
5.web配置
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
|
<?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> |
6,前端，為方便起見，我直接用了兩個jsp，在其中用<%session.setAttribute("user","小明")%>；來表示登錄。
兩個jsp沒任何本質差別，只是用來表示兩個不同的人登錄，可以同兩個瀏覽器打開不同的jsp，來聊天操作
A.小化
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
|
<%@ 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> |
B.小明
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
|
<%@ 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> |
以上所述是小編給大家介紹的使用JavaWeb webSocket實現簡易的點對點聊天功能實例代碼的相關知識，希望對大家有所幫助，如果大家有任何疑問請給我留言，小編會及時回復大家的。在此也非常感謝大家對服務器之家網站的支持！
