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

首先给大家声明一点:需要 jdk 7 , tomcat需要支持websocket的版本 

1.InitServlet

   该类主要是用来初始化构造将来存储用户身份信息的map仓库,利用其初始化方法Init 初始化仓库, 利用其静态方法getSocketList 获得对应的用户身份信息。

   webSocket ,我认为MessageInbound 用来识别登录人的信息,用它来找到对应的人,推送消息。每次登录都会产生一个MessageInbound。

  这里的 HashMap    :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 = -L; 
   //private static List socketList;  
   private static HashMap socketList;  
   public void init(ServletConfig config) throws ServletException {  
 //    InitServlet.socketList = new ArrayList();  
     InitServlet.socketList = new HashMap();  
     super.init(config);  
     System.out.println("Server start============");  
   }  
   public static HashMap getSocketList() {  
     return InitServlet.socketList;  
   }  
 /*  public static List getSocketList() {  
     return InitServlet.socketList;  
   }  
 */}

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 --
 */
 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 :用来获取消息,并发送消息

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 messageMap = MessageUtil.getMessage(msg);  //处理消息类
     //上线用户集合类map
     HashMap 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.消息处理类,处理前端发来的消息

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

5.web配置



  
  mywebsocket 
  socket.MyWebSocketServlet 
  
  
  mywebsocket 
  *.do 
  
  
  initServlet 
  socket.InitServlet 
  1 
  
 
  index.jsp
 

6,前端,为方便起见,我直接用了两个jsp,在其中用<%session.setAttribute("user","小明")%>;来表示登录。

      两个jsp没任何本质差别,只是用来表示两个不同的人登录,可以同两个浏览器打开不同的jsp,来聊天操作

   A.小化

<%@ page language="java" contentType="text/html; charset=UTF-"
   pageEncoding="UTF-"%>
 
 
 
 
 Index
 
 <%session.setAttribute("user", "小化");%>
 
 
 
 

聊天功能实现

登录状态: 正在登录
登录人:


发送给谁:
发送内容:
聊天框:

 B.小明

<%@ page language="java" contentType="text/html; charset=UTF-8"
  pageEncoding="UTF-8"%>




Index

<%session.setAttribute("user", "小明");%>



聊天功能实现

登录状态: 正在登录
登录人:


发送给谁:
发送内容:
聊天框:

以上所述是小编给大家介绍的使用JavaWeb webSocket实现简易的点对点聊天功能实例代码的相关知识,希望对大家有所帮助,如果大家有任何疑问请给我留言,小编会及时回复大家的。在此也非常感谢大家对脚本之家网站的支持!

你可能感兴趣的:(使用JavaWeb webSocket实现简易的点对点聊天功能实例代码)