SpringBoot集成WebSocket实现

@Configuration
public class WebSocketConfig {

    @Bean
    public ServerEndpointExporter serverEndpointExporter() {
        return new ServerEndpointExporter();
    }
}

@Component
@ServerEndpoint("/webSocket")
public class WebSocketServer {

    // concurrent包的线程安全set, 用来存放每个客户端对应的DCHWebSocket对象
    private static CopyOnWriteArraySet webSocketSet = new CopyOnWriteArraySet<>();

    // 与客户端的连接会话, 需要通过session给客户端发送数据
    private Session session=null;

    private Logger logger = LoggerFactory.getLogger(WebSocketServer.class);

    @OnOpen
    public void onOpen(Session session) {
        System.out.println("新建了一个websocket连接");
        this.session = session;
        webSocketSet.add(this);
    }

    @OnClose
    public void onClose() {
        System.out.println("关闭了一个websocket连接");
        webSocketSet.remove(this);
    }

    @OnMessage
    public void onMessage(String message) {
        try {
            for (WebSocketServer webSocket : webSocketSet) {
                Session socketSession = webSocket.session;
                if (socketSession.isOpen()) {
                    socketSession.getBasicRemote().sendText(message);
                }
            }
        }catch (Exception e) {
            logger.error("socket发送消息异常",  e);
        }
    }

    @OnError
    public void onError(Throwable error) {
        System.out.println("webSocket发生错误");
        logger.error("webSocket发生错误", error);
    }
}

//实现自己的逻辑
@Autowired
    private WebSocketServer webSocketServer;

    @GetMapping("/xxx")
    public void receiveImgCountMsg(String xxx) {
        JSONObject msgObj = new JSONObject();
        msgObj.put("xxx", xxxx);
        webSocketServer.onMessage(msgObj.toJSONString());
    }

    	org.springframework.boot
    	spring-boot-starter-websocket

5.参考;SpringBoot快速搭建简易WebSocket系统（springboot整合thymeleaf模板引擎）_一位不爱熬夜但又经常熬夜的程序员的博客-CSDN博客