首先给大家声明一点:需要 jdk 7 , tomcat需要支持websocket的版本
1.InitServlet
该类主要是用来初始化构造将来存储用户身份信息的map仓库,利用其初始化方法Init 初始化仓库, 利用其静态方法getSocketList 获得对应的用户身份信息。
webSocket ,我认为MessageInbound 用来识别登录人的信息,用它来找到对应的人,推送消息。每次登录都会产生一个MessageInbound。
这里的 HashMap
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 ListsocketList; 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 HashMapmessageMap = 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 HashMapgetMessage(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实现简易的点对点聊天功能实例代码的相关知识,希望对大家有所帮助,如果大家有任何疑问请给我留言,小编会及时回复大家的。在此也非常感谢大家对脚本之家网站的支持!