java-webSocket
例:
JS:
var proName = '${pageContext.request.contextPath}'; var ws = new WebSocket('ws://'+ window.location.host + proName +'/confservice/confWebSocketService/huiyifuwu');//new webSocket表示已连接 webSocket(); //建立消息连接 function webSocket() { ws.onopen = function(evt){ //心跳检测重置 heartCheck.reset().start(); //ws.send("send123"); }; ws.onmessage = function(evt) { var data = evt.data; if (data != "ping!") { splitData(data); } //心跳检测重置 heartCheck.reset().start(); } ws.onclose = function(evt) { ws.close(); } } function splitData(data) { } //心跳检测 var heartCheck = { timeout: 180000,//180秒 timeoutObj: null, reset: function(){ clearTimeout(this.timeoutObj); return this; }, start: function(){ this.timeoutObj = setTimeout(function(){ //这里发送一个心跳,后端收到后,返回一个心跳消息, //onmessage拿到返回的心跳就说明连接正常 ws.send("ping!"); }, this.timeout) } } // 监听窗口关闭事件,当窗口关闭时,主动去关闭websocket连接,防止连接还没断开就关闭窗口,server端会抛异常。 window.onbeforeunload = function() { ws.close(); }
java:
@Component//申明为spring组件 @ServerEndpoint("/confservice/confWebSocketService/{id}")//webSocket连接地址 public class ConfWebSocketService { private static final Logger logger = Logger.getLogger(ConfWebSocketService.class); private static Map<Session, String> sessionMaps = new HashMap<>(); private static Session sessionf; private static List<User> allUserMS = new ArrayList<>(); private static List<User> allUserWY = new ArrayList<>(); @Autowired private IUserService userService; // 静态初使化当前类 public static ConfWebSocketService confWebSocketService; //注解@PostConstruct,这样方法就会在Bean初始化之后被Spring容器执行 @PostConstruct public void init() { confWebSocketService = this; } /** * 连接成功调用 * @param session */ @OnOpen public void onOpen(Session session, @PathParam("id") String id) { Thread thread = new Thread(new Runnable() { @Override public void run() { try { sessionf = session; sessionMaps.put(session, id); } catch (Exception e) { logger.debug("webSocket连接异常:"+e.getMessage()); e.printStackTrace(); } } }); thread.start(); } /** * 关闭连接调用 */ @OnClose public void onClose(Session session) { Thread thread = new Thread(new Runnable() { @Override public void run() { sessionMaps.remove(session); } }); thread.start(); } /** * 收到客户端消息调用 * @param message * @param session */ @OnMessage public void onMessage(String message,Session session) { if (message.equals("ping!")) { //保持和会议服务长连接 sendMessage(message,"0"); } else if(message.equals("yuyin!")) { //保持和语音识别长连接 sendMessage(message,"2"); } } /** * 出现错误调用 * @param session * @param error */ @OnError public void onError(Session session, Throwable error) { error.printStackTrace(); } /** * 发送消息(自定义的方法) */ public static void sendMessage(String message,String flag) { try { //session.getBasicRemote().sendText(message);//同步发送消息 if (flag.equals("1") || flag.equals("2")) { Role role =new Role(); if (flag.equals("1")) { role.setType("2"); if (allUserMS.size() == 0) { allUserMS = confWebSocketService.userService.getUserByRole(role); } } else { role.setType("0"); if (allUserWY.size() == 0) { allUserWY = confWebSocketService.userService.getUserByRole(role); } } } Set<Map.Entry<Session,String>> entrySet = sessionMaps.entrySet(); for (Entry<Session, String> map : entrySet) { String checkId = map.getValue(); if (flag.equals("0") && checkId.equals("huiyifuwu")) { sessionf = map.getKey(); sessionf.getAsyncRemote().sendText(message);//异步发送消息 break; } else if (flag.equals("1") && allUserMS.size() > 0) { for (User user : allUserMS) { if (checkId.equals(user.getId())) { sessionf = map.getKey(); sessionf.getAsyncRemote().sendText(message);//异步发送消息 break; } } } else if (flag.equals("2") && allUserWY.size() > 0) { for (User user : allUserWY) { if (checkId.equals(user.getId())) { sessionf = map.getKey(); sessionf.getAsyncRemote().sendText(message);//异步发送消息 break; } } } } } catch (Exception e) { e.printStackTrace(); } } }