《java websocket》之 实现

上一篇《java websocket》之 描述篇介绍了一下websocket，那么这一次就自己动手搭建一个属于自己的websocket。

真的挺简单的

环境描述：java jdk1.7+tomcat7.0x

第一步。创建一个干净的web project工程：

第二步。然后导入我们所用到的包：

第三步。创建一个websocket类用来实现接口，在创建一个POOL类用来扩展业务

废话不说直接上代码，相信我，我提供的代码直接复制进去就可以用！

WebSocket 类：

package websocket; import java.net.InetSocketAddress; import java.net.UnknownHostException;import org.java_websocket.handshake.ClientHandshake; import org.java_websocket.server.WebSocketServer;public class WebSocket extends WebSocketServer {public WebSocket(InetSocketAddress address) {super(address);System.out.println("地址"+address);}public WebSocket(int port) throws UnknownHostException {super(new InetSocketAddress(port));System.out.println("端口"+port);}/*** 触发关闭事件*/@Overridepublic void onClose(org.java_websocket.WebSocket conn, int message,String reason, boolean remote) {userLeave(conn);}/*** 触发异常事件*/@Overridepublic void onError(org.java_websocket.WebSocket conn, Exception message) {System.out.println("Socket异常:"+message.toString());e++;}/*** 客户端发送消息到服务器时触发事件*/int j=0;int h=0;int e=0;int l=0;@Overridepublic void onMessage(org.java_websocket.WebSocket conn, String message) {message = message.toString();if(message!=null){//将用户加入this.userjoin(message.toString(), conn);}}/*** 触发连接事件*/@Overridepublic void onOpen(org.java_websocket.WebSocket conn, ClientHandshake handshake) {System.out.println("有人连接Socket conn:"+conn);l++;}/*** 用户加入处理* @param user*/public void userjoin(String user, org.java_websocket.WebSocket conn) {WebSocketPool.sendMessage(user); // 把当前用户加入到所有在线用户列表中String joinMsg = "[系统]"+user+"上线了！";WebSocketPool.sendMessage(joinMsg); // 向所有在线用户推送当前用户上线的消息WebSocketPool.addUser(user, conn); // 向连接池添加当前的连接对象WebSocketPool.sendMessageToUser(conn, WebSocketPool.getOnlineUser().toString()); // 向当前连接发送当前在线用户的列表}/*** 用户下线处理* @param user*/public void userLeave(org.java_websocket.WebSocket conn) {String user = WebSocketPool.getUserByKey(conn);boolean b = WebSocketPool.removeUser(conn); // 在连接池中移除连接if (b) {WebSocketPool.sendMessage(user.toString()); // 把当前用户从所有在线用户列表中删除String joinMsg = "[系统]"+user+"下线了";WebSocketPool.sendMessage(joinMsg); // 向在线用户发送当前用户退出的消息}} }

WebSocketPool 类：

package websocket; import java.util.ArrayList; import java.util.Collection; import java.util.HashMap; import java.util.List; import java.util.Map; import java.util.Set; import org.java_websocket.WebSocket;public class WebSocketPool {private static final Map<WebSocket, String> userconnections = new HashMap<WebSocket, String>();/*** 获取用户名* @param session*/public static String getUserByKey(WebSocket conn) {return userconnections.get(conn);}/*** 获取在线总数* @param*/public static int getUserCount() {return userconnections.size();}/*** 获取WebSocket* @param user*/public static WebSocket getWebSocketByUser(String user) {Set<WebSocket> keySet = userconnections.keySet();synchronized (keySet) {for (WebSocket conn : keySet) {String cuser = userconnections.get(conn);if (cuser.equals(user)) {return conn;}}}return null;}/*** 向连接池中添加连接* @param inbound*/public static void addUser(String user, WebSocket conn) {userconnections.put(conn, user); // 添加连接}/*** 获取所有的在线用户* @return*/public static Collection<String> getOnlineUser() {List<String> setUsers = new ArrayList<String>();Collection<String> setUser = userconnections.values();for (String u : setUser) {setUsers.add(u);}return setUsers;}/*** 移除连接池中的连接* @param inbound*/public static boolean removeUser(WebSocket conn) {if (userconnections.containsKey(conn)) {userconnections.remove(conn); // 移除连接return true;} else {return false;}}/*** 向特定的用户发送数据* @param user* @param message*/public static void sendMessageToUser(WebSocket conn, String message) {if (null != conn) {conn.send(message);}}/*** 向所有的用户发送消息* @param message*/public static void sendMessage(String message) {Set<WebSocket> keySet = userconnections.keySet();synchronized (keySet) {for (WebSocket conn : keySet) {String user = userconnections.get(conn);if (user != null) {conn.send(message);}}}} }

然后让我们先测试一下是否行得通：

所以我在WebSocket 类里面写一个main方法来启动websocket服务:

public static void main(String[] args) throws InterruptedException {System.out.println("开始启动websocket");WebSocketImpl.DEBUG = false;int port = 8888; // 端口随便设置，只要不跟现有端口重复就可以WebSocket s = null;try {s = new WebSocket(port);} catch (UnknownHostException e) {System.out.println("启动websocket失败！");e.printStackTrace();}s.start();System.out.println("启动websocket成功！");}

启动成功后的样子

然后去http://www.blue-zero.com/websocket/ WebSocket在线测试
输入地址跟端口后，点击链接


那么到了这一步就说明你的websocket 已经搭建好了哦！恭喜你。

下面跟着作者把配置好的websocket实际应用到项目中。

第一步：技术点为：项目启动的时候启动配置好的服务。

因为启动项目就要开启websocket服务保持长连接状态。所以我们找到了web.xml，注意必须是3.0以上的

<?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"><display-name></display-name> <filter><filter-name>startFilter</filter-name><filter-class>filter.startFilter</filter-class></filter><servlet> <servlet-name>TestServlet</servlet-name> <servlet-class>servlet.TestServlet</servlet-class> </servlet><servlet-mapping><servlet-name>TestServlet</servlet-name><url-pattern>/TestServlet</url-pattern></servlet-mapping><welcome-file-list><welcome-file>index.jsp</welcome-file></welcome-file-list> </web-app>

创建startFilter (原谅我忘记大写开头，自己以打自己脸3次)，以及TestServlet(不懂Servlet出门百度)

package filter;import java.io.IOException; import java.net.UnknownHostException;import javax.servlet.Filter; import javax.servlet.FilterChain; import javax.servlet.FilterConfig; import javax.servlet.ServletException; import javax.servlet.ServletRequest; import javax.servlet.ServletResponse;import org.java_websocket.WebSocketImpl;import websocket.WebSocket;/*** 创建人：晓枫 创建时间：2014年8月18日* @version*/ public class startFilter implements Filter {/*** 初始化*/public void init(FilterConfig fc) throws ServletException {this.startWebsocketOnline();}/*** 启动socket服务*/public void startWebsocketOnline() {System.out.println("开始启动websocket");WebSocketImpl.DEBUG = false;int port = 8888; // 端口随便设置，只要不跟现有端口重复就可以WebSocket s = null;try {s = new WebSocket(port);s.start();} catch (UnknownHostException e) {System.out.println("启动websocket失败！");e.printStackTrace();}System.out.println("启动websocket成功！");}public void destroy() {// TODO Auto-generated method stub}public void doFilter(ServletRequest arg0, ServletResponse arg1, FilterChain arg2) throws IOException, ServletException {// TODO Auto-generated method stub}}

细心地学者已经发现，实际上就是把main方法里面的代码拿过来了，霍哇啊哈哈哈哈哈哈！

TestServlet 类

package servlet;import java.io.IOException;import javax.servlet.ServletException; import javax.servlet.http.HttpServlet; import javax.servlet.http.HttpServletRequest; import javax.servlet.http.HttpServletResponse;public class TestServlet extends HttpServlet {@Overrideprotected void doGet(HttpServletRequest req, HttpServletResponse resp)throws ServletException, IOException {// TODO Auto-generated method stubsuper.doPost(req, resp);}@Overrideprotected void doPost(HttpServletRequest req, HttpServletResponse resp)throws ServletException, IOException {System.out.println("content"+req.getParameter("content"));req.getSession().setAttribute("content", req.getParameter("content"));req.getRequestDispatcher("/index.jsp").forward(req, resp);} }

到了这一步就基本服务端已经基本搭建好了。那么开始前台。技术为jsp页面+js

index.jsp

<%@ page language="java" import="java.util.*" pageEncoding="UTF-8"%> <% String path = request.getContextPath(); String basePath = request.getScheme()+"://"+request.getServerName()+":"+request.getServerPort()+path+"/"; %><!DOCTYPE HTML PUBLIC "-//W3C//DTD HTML 4.01 Transitional//EN"> <html><head><base href="<%=basePath%>"><title>晓枫的WebSocket</title><meta http-equiv="pragma" content="no-cache"><meta http-equiv="cache-control" content="no-cache"><meta http-equiv="expires" content="0"> <meta http-equiv="keywords" content="keyword1,keyword2,keyword3"><meta http-equiv="description" content="This is my page"><script type="text/javascript" src="js/websocket.js"></script></head><body><input type="hidden" value="${content }" id="content"><form action="TestServlet" method="post">content : <input type="text" name="content"><button type="submit">登陆</button></form><span id="span"></span></body> </html>

就简单的画了几个框框

（废话： 大部分人都有个毛病，就是功能做一半就会被别的事情所打扰，导致事情拖延，然后加班。就比如一个简单的页面就可以测试，非要写的那么华丽！作者我不管在哪个公司，很少加班，因为是后台，界面的好看与否，本身不算到工作量里面。达到目的就可以。如果嫌弃丑会再给你时间改的，不会加班）

websocket.js

/*** 申请一个WebSocket对象，参数是需要连接的服务器端的地址，* 同http协议使用http://开头一样，* WebSocket协议的URL使用ws://开头，* 另外安全的WebSocket协议使用wss://开头* 创建人：晓枫 创建时间 2016-8-18*/ var ws = new WebSocket("ws://127.0.0.1:8888"); ws.onopen = function()//当websocket创建成功时，即会触发onopen事件 { var content=document.getElementById("content").value;if(content==nullcontent==""){ws.send("XiaoFeng");//用于叫消息发送到服务端 注：此处为用户名}else{ws.send(content);} }; ws.onmessage = function(evt)//当客户端收到服务端发来的消息时，会触发onmessage事件，参数evt.data中包含server传输过来的数据 {document.getElementById("span").innerHTML="当前在线人数为："+evt.data; }; ws.onclose = function(evt)//当客户端收到服务端发送的关闭连接的请求时，触发onclose事件 {alert("WebSocketClosed!"); }; ws.onerror = function(evt)//如果出现连接，处理，接收，发送数据失败的时候就会触发onerror事件 {alert("WebSocketError!"); };

到这里就可以结束了。
下面正式展示一下

我们打开多个页面来模仿多人连接.

我们输入一个内容

点击登录

已经可以看到这个用户信息添加到Map了
同时,别的页面也收到了提示消息！

那么到这里就基本结束了。
强制下线就是把连接移除掉，并且通知你的logout方法就可以了。
只能一个人登陆的话，就判断一下Map里面是否有用户就OK了！
实例就在这里，注释也都有，自己开脑洞吧。

支持单聊哦！方法已经写好了在Pool.自己发现吧

到这里开始要开始研究高并发了，可能会穿插一些spring实现之类的，看情况吧。

（下面看不看吧，压力测试一下）

在线测试的网站可以下载压力测试工具

然后稍微的修改一下代码。

然后启动服务器，清空控制台，设置好5000个客户端同时连接，连接成功后发送“1”，项目接收到后回复“2”，也就是说发送次数到为 连接数X2 的时候为一个不丢失， 越接近越好。

开始战斗吧！

额。。。。。。。。。。。。。。。。。。。。

呵呵。

作者： 手插口袋_ 
链接：http://www.imooc.com/article/12401
来源：慕课网

总结

以上是生活随笔为你收集整理的《java websocket》之 实现的全部内容，希望文章能够帮你解决所遇到的问题。

如果觉得生活随笔网站内容还不错，欢迎将生活随笔推荐给好友。