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

时间:2021-11-26 13:22:40

首先给大家声明一点:需要 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实现简易的点对点聊天功能实例代码的相关知识,希望对大家有所帮助,如果大家有任何疑问请给我留言,小编会及时回复大家的。在此也非常感谢大家对服务器之家网站的支持!