Spring项目中使用webservice实现h5的websocket通信

一、在项目中建立一个webservice来做后台操作。

package org.calonlan.soulpower.websocket;import java.text.SimpleDateFormat;
import java.util.Date;import javax.websocket.OnClose;
import javax.websocket.OnError;
import javax.websocket.OnMessage;
import javax.websocket.OnOpen;
import javax.websocket.Session;
import javax.websocket.server.PathParam;
import javax.websocket.server.ServerEndpoint;/*** 这里使用注释的方式来向系统指明，我的这个WebSocketTest是一个webservice。 同时指定了路径为/websocket/{uCode}* 其中的{uCode}是一个变化的参数，在网页端动态的输入。这样我就可以通过@PathParam("uCode")这个注释在参数中获得用户的信息等等了。* * */
@ServerEndpoint("/websocket/{uCode}")
public class WebSocketTest {/*** @param message*            这里是客户端传来的消息，我这里只是简单的测试文本消息，消息的种类可以有很多种。* @param uCode*            这就是{uCode}中传来的路径参数，可以用来传递用户的信息。例如帐号。* @throws Exception*             偷懒的人总是抛出一个EXCEPTION*/@OnMessage// 当有消息传来的时候进行处理的方法public void onMessage(String message, @PathParam("uCode") String uCode)throws Exception {System.out.println("revived:" + message);// 输出一下接收到的消息String tem[] = message.split("##to##");// 消息的格式是// tousername##to##message，分隔以后第一个就是要发送的用户名，第二个就是消息了if (SessionUtils.hasConnection(tem[0])) {// 从sessionUtils中判断一下是否要发送的用户名是否已经登录，登录的话做以下操作/* 弄个时间开始 */SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");String x = sdf.format(new Date());/* 弄个时间结束 *//** 消息的发送，可以选择AsyncRemote也可以选择* BasicRemote,区别在于AsyncRemote是不会阻塞的，系统不会去管消息是否发送完成直接占用通道去发* 。使用它的话要用户自己去实现控制，所以我选择BasicRemote，*/SessionUtils.get(tem[0]).getBasicRemote().sendText(uCode + "##to##" + x + ":" + tem[1]);} else {// 没有登录的话做以下操作SessionUtils.get(uCode).getBasicRemote().sendText("用户不在线");}}/*** @param uCode*            同上* @param session*            这个是用户建立的session信息，用来唯一标识这个用户的通信* @throws Exception*             你懂得*/@OnOpen// 用户建立链接的时候执行的方法public void onOpen(@PathParam("uCode") String uCode, Session session)throws Exception {if (SessionUtils.hasConnection(uCode)) {// 判断缓存中是否有uCode，如果有执行括号内的方法SessionUtils.get(uCode).close();// 因为已经登录了，那么将已经登录的下线SessionUtils.remove(uCode);// 移除掉缓存中的<uCode,Session>SessionUtils.put(uCode, session);// 添加新的<uCode,Session>System.out.println(uCode + "has join server");} else {System.out.println(uCode + "has join server");/* 如果没有缓存相关的<uCode,Session>,那么直接添加 */SessionUtils.put(uCode, session);}}@OnClose//客户端断开链接时执行的方法public void onClose(@PathParam("uCode") String uCode) {System.out.println(uCode + "has left server");SessionUtils.remove(uCode);//直接移除就好了，session已经关闭了System.out.println("left deal was finished");}@OnError//客户端出错误的时候执行的方法public void onError(Throwable e, Session session) {if (SessionUtils.clients.containsValue(session)) {//移除出错的sessionSessionUtils.remove(session);}}}

二、实现用来管理session的uti类。

package org.calonlan.soulpower.websocket;import java.util.Iterator;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;import javax.websocket.Session;public class SessionUtils {/*<uCode,Session>的缓存*/public static Map<String, Session> clients = new ConcurrentHashMap<String, Session>();public static void put(String uCode, Session session) {//添加clients.put(uCode, session);}public static Session get(String uCode) {//根据uCode来获取return clients.get(uCode);}public static void remove(String uCode) {//根据uCode来移除clients.remove(uCode);}public static void remove(Session session) {//根据session来移除Iterator<java.util.Map.Entry<String, Session>> ito = clients.entrySet().iterator();while (ito.hasNext()) {java.util.Map.Entry<String, Session> entry = ito.next();if (entry.getKey().equals(session))remove(entry.getKey());}}public static boolean hasConnection(String uCode) {//根据uCode来判断是否包含对应用户名的<uCode,Session>return clients.containsKey(uCode);}
}

三、html页面实现

<%@ page language="java" contentType="text/html; charset=utf-8"pageEncoding="utf-8"%>
<%@ taglib prefix="c" uri="http://java.sun.com/jsp/jstl/core"%>
<!DOCTYPE html>
<html> <head> <title>WebSoket Demo</title> <script type="text/JavaScript"> //验证浏览器是否支持WebSocket协议if (!window.WebSocket) { alert("WebSocket not supported by this browser!"); } var ws;function display() { //创建webSocket，后面的${id}与{uCode}相对应ws=new WebSocket("ws://localhost:8080/newback/websocket/${id}"); //监听消息ws.onmessage = function(event) { log(event.data);}; //绑定关闭事件 ws.onclose = function(event) {/*这段代码的作用是，如果同一个id在其他的页面登录，那么就强制当前的页面下线，关闭。在服务器端我们在同一个id登录时关闭了以前的socket，所以这里在关闭事件中进行操作*/var opened=window.open('about:blank','_self'); opened.opener=null; opened.close();}; //建立websocket的事件，可以用来做一些初始化操作；比如如果用户不在线其他人发送了消息我可以放在数据库里，用户一上线就调用查询方法ws.onopen = function(event) { }; //出现错误的时候的方法ws.onerror =function(event){};} var log = function(s) {//打印消息的方法if (document.readyState !== "complete") {  log.buffer.push(s);  } else {  document.getElementById("contentId").innerHTML += (s + "\n");  }  }function sendMsg(){//这里是发送消息，我制做了c1、c2之间的通话var msg=document.getElementById("messageId");var client='${id}';if(client=='c1')ws.send("c2##to##"+msg.value);//发送消息elsews.send("c1##to##"+msg.value);}</script> </head> <body οnlοad="display();"> <div id="valueLabel"></div> <textarea rows="20" cols="30" id="contentId"></textarea><br/><input name="message" id="messageId"/><button id="sendButton" onClick="javascript:sendMsg()" >Send</button></body> 
</html>

四、测试。

1.登录c1

服务器端显示

2.登录c2

服务器端显示

登录完成，就可以互相发送消息了。

c1发送给c2

c2发送给c1同样的道理。

这里还看到了很多不是发送的消息，这是我在后台设置的定时任务。服务器可以定时的给客户端发送一些消息。

五、定时任务

这里我遍历sessionutils中的map，给所有用户都发送

package org.calonlan.soulpower.timertask;import java.util.Iterator;
import java.util.Map;
import java.util.Map.Entry;import javax.websocket.Session;import org.calonlan.soulpower.util.ComputerUtil;
import org.calonlan.soulpower.websocket.SessionUtils;
import org.hyperic.sigar.SigarException;public class TimerTask {public void printTimeStamp() throws SigarException {Map<String, Session> map = SessionUtils.clients;Iterator<Entry<String, Session>> ito = map.entrySet().iterator();while (ito.hasNext()) {Entry<String, Session> entry = ito.next();try {if (entry.getValue().isOpen()) {entry.getValue().getBasicRemote().sendText(ComputerUtil.getMemory());entry.getValue().getBasicRemote().sendText(ComputerUtil.getCPU());entry.getValue().getBasicRemote().sendText(ComputerUtil.getDisk());} else {map.remove(entry.getKey());}} catch (Exception e) {System.out.println("错误");e.printStackTrace();map.remove(entry.getKey());}}}public TimerTask() {}public void doTask() throws SigarException {this.printTimeStamp();}}