javaweb webSocket 實現簡單的點對點聊天功能

ydx3 9年前發布 | 186K 次閱讀 WebSocket 開發 WebSocket

本文依據 http://redstarofsleep.iteye.com/blog/1488639?page=4  內容修改完成，實現點對點聊天

需要 jdk 7 ， tomcat需要支持websocket的版本 

 

1.InitServlet

   該類主要是用來初始化構造將來存儲用戶身份信息的map倉庫，利用其初始化方法Init 初始化倉庫， 利用其靜態方法getSocketList 獲得對應的用戶身份信息。

   webSocket ，我認為MessageInbound 用來識別登錄人的信息，用它來找到對應的人，推送消息。每次登錄都會產生一個MessageInbound。

  這里的 HashMap<String,MessageInbound>    ：string 存儲用戶session的登錄id,MessageInbound存儲 推送需要的身份信息。以上屬于個人口頭話理解。 

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 = -3163557381361759907L;  

//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;

    }    /}</pre>


2.MyWebSocketServlet  


  websocket用來建立連接的servlet，建立連接時，首先在session獲取該登錄人的userId，在調用MyMessageInbound構造函數傳入userId 


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 2015-7-19
*/
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 arg0,
    


         HttpServletRequest request) {
 System.out.println("##########");  
 return new MyMessageInbound(this.getUser(request)); 

    

    }
    

    • 



}</pre>


3.onOpen方法調用InitServlet的map身份倉庫， 


放入用戶userId 和 對應該登錄用戶的websocket身份信息MessageInbound （可以用userId來尋找到推送需要的 身份MessageInbound ） 


onTextMessage ：用來獲取消息，并發送消息
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 arg0) 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);

    }


}</pre>


4.消息處理類，處理前端發來的消息 


package util;


import java.nio.CharBuffer;
import java.util.HashMap;
/**




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="13" cols="100" readonly id="xiaoxi"></textarea>
<br>
<input type="button" value="send" onclick="sendMsg()"></input>
</body>
</html></pre>

 B.小明 
<%@ 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></pre>

來自：http://www.cnblogs.com/Mangues/p/4658635.html
 本文由用戶 ydx3 自行上傳分享，僅供網友學習交流。所有權歸原作者，若您的權利被侵害，請聯系管理員。
 轉載本站原創文章，請注明出處，并保留原始鏈接、圖片水印。
 本站是一個以用戶分享為主的開源技術平臺，歡迎各類分享！