webSocket使用说明

前言：集成之前请确认ActiveMQ 能够正常启动发送消息
1.引入JAR包

先引入这两个jar包
2.创建文件

这里userId是从前端传来的当前登录者，如果发送的消息不用指定人接受 这里可以不用传这个值，并把方法中的这个参数给删除掉就好了
下面是创建连接的全文：
package com.hotent.activeMQ.controller.webSocket;

import java.io.IOException;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;

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;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.hotent.base.core.web.GenericController;

//websocket连接URL地址和可被调用配置
@ServerEndpoint(value="/websocketDemo/{userId}")
public class WebsocketDemo extends GenericController{
//日志记录
private Logger logger = LoggerFactory.getLogger(WebsocketDemo.class);
//静态变量，用来记录当前在线连接数。应该把它设计成线程安全的。
private static int onlineCount = 0;

//记录每个用户下多个终端的连接
private static Map<String, Set<WebsocketDemo>> userSocket = new HashMap<>();

//需要session来对用户发送数据, 获取连接特征userId
private Session session;
private String userId;

/**
 * @Title: onOpen
 * @Description: websocket连接建立时的操作
 * @param @param userId 用户id
 * @param @param session websocket连接的session属性
 * @param @throws IOException
 */
@OnOpen
public void onOpen(@PathParam("userId") String userId, Session session) throws IOException {
    this.session = session;
   // User userDetails = (User) SecurityContextHolder.getContext().getAuthentication().getPrincipal();
   
    this.userId = userId;
    onlineCount++;
    //根据该用户当前是否已经在别的终端登录进行添加操作
    if (userSocket.containsKey(this.userId)) {
        logger.debug("当前用户id:{}已有其他终端登录",this.userId);
        userSocket.get(this.userId).add(this); //增加该用户set中的连接实例
    }else {
        logger.debug("当前用户id:{}第一个终端登录",this.userId);
        Set<WebsocketDemo> addUserSet = new HashSet<>();
        addUserSet.add(this);
        userSocket.put(this.userId, addUserSet);
    }
    logger.debug("用户{}登录的终端个数是为{}",userId,userSocket.get(this.userId).size());
    logger.debug("当前在线用户数为：{},所有终端个数为：{}",userSocket.size(),onlineCount);
}

/**
 * @Title: onClose
 * @Description: 连接关闭的操作
 */
@OnClose
public void onClose(){
    //移除当前用户终端登录的websocket信息,如果该用户的所有终端都下线了，则删除该用户的记录
    if (userSocket.get(this.userId).size() == 0) {
        userSocket.remove(this.userId);
    }else{
        userSocket.get(this.userId).remove(this);
    }
    logger.debug("用户{}登录的终端个数是为{}",this.userId,userSocket.get(this.userId).size());
    logger.debug("当前在线用户数为：{},所有终端个数为：{}",userSocket.size(),onlineCount);
}

/**
 * @Title: onMessage
 * @Description: 收到消息后的操作
 * @param @param message 收到的消息
 * @param @param session 该连接的session属性
 */
@OnMessage
public void onMessage(String message, Session session) {
    logger.debug("收到来自用户id为：{}的消息：{}",this.userId,message);
    if(session ==null)  logger.debug("session null");
}

/**
 * @Title: onError
 * @Description: 连接发生错误时候的操作
 * @param @param session 该连接的session
 * @param @param error 发生的错误
 */
@OnError
public void onError(Session session, Throwable error){
    logger.debug("用户id为：{}的连接发送错误",this.userId);
    error.printStackTrace();
}

/**
 * @Title: sendMessageToUser
 * @Description: 发送消息给用户下的所有终端
 * @param @param userId 用户id
 * @param @param message 发送的消息
 * @param @return 发送成功返回true，反则返回false
 */
public Boolean sendMessageToUser(String userId,String message){
    if (userSocket.containsKey(userId)) {
        logger.debug(" 给用户id为：{}的所有终端发送消息：{}",userId,message);
        for (WebsocketDemo WS : userSocket.get(userId)) {
            logger.debug("sessionId为:{}",WS.session.getId());
            try {
                WS.session.getBasicRemote().sendText(message);
            } catch (IOException e) {
                e.printStackTrace();
                logger.debug(" 给用户id为：{}发送消息失败",userId);
                return false;
            }
        }
        return true;
    }
    logger.debug("发送错误：当前连接不包含id为：{}的用户",userId);
    return false;
}

}

3.发送消息
这里 发送消息是从ActiveMQ来发送的 所用到的是wsMessageService里面的方法


以下是service 全文：
package com.hotent.activeMQ.persistence.websocket;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.stereotype.Service;

import com.hotent.activeMQ.controller.webSocket.WebsocketDemo;

@Service
public class WSMessageService {

private Logger logger = LoggerFactory.getLogger(WSMessageService.class);
//声明websocket连接类
private WebsocketDemo websocketDemo = new WebsocketDemo();

/**
 * @Title: sendToAllTerminal
 * @Description: 调用websocket类给用户下的所有终端发送消息
 * @param @param userId 用户id
 * @param @param message 消息
 * @param @return 发送成功返回true，否则返回false
 */
public Boolean sendToAllTerminal(String userId,String message){
    logger.debug("向用户{}的消息：{}",userId,message);
    if(websocketDemo.sendMessageToUser(userId,message)){
        return true;
    }else{
        return false;
    }
}

}
4.前端页面

这里的是前端访问的方法，其中
websocket = new WebSocket(“ws://localhost:8080/ayrdp/websocketDemo/”+id);
这是连接websocket的路径，，这里你访问项目的路径是localhost的话就要用localhost
如果是IP地址的话 这几就好改成IP地址

以下为jsp页面的全文
<%@page import=“com.hotent.sys.util.ContextUtil”%>
<%@ page language=“java” contentType=“text/html; charset=UTF-8” pageEncoding=“UTF-8”%>
<!doctype html>

<%@include file="/commons/include/list.jsp"%>

webSocket多终端聊天测试•

发送

这里有引用的一个js

Js全文太长了 这里就只有一个图片

测试的时候请先启动ActiveMQ ，不然会报错的