WebSocketConfig.java
package com.ruoyi.framework.config;
import org.springframework.context.annotation.Bean;
import org.springframework.stereotype.Component;
import org.springframework.web.socket.server.standard.ServerEndpointExporter;
@Component
public class WebSocketConfig {
? ? /**
? ? ?* ServerEndpointExporter 作用
? ? ?*
? ? ?* 这个Bean会自动注册使用@ServerEndpoint注解声明的websocket endpoint
? ? ?*
? ? ?* @return
? ? ?*/
? ? @Bean
? ? public ServerEndpointExporter serverEndpointExporter() {
? ? ? ? return new ServerEndpointExporter();
? ? }
}
WebSocketTest.java
package com.ruoyi.framework.web.service;
import org.springframework.stereotype.Component;
import java.io.IOException;
import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.CopyOnWriteArraySet;
import javax.websocket.*;
import javax.websocket.server.PathParam;
import javax.websocket.server.ServerEndpoint;
/**
?* @ServerEndpoint 注解是一个类层次的注解，它的功能主要是将目前的类定义成一个websocket服务器端,
?* 注解的值将被用于监听用户连接的终端访问URL地址,客户端可以通过这个URL来连接到WebSocket服务器端
?*/
@Component
@ServerEndpoint("/websocket")
public class WebSocketTest {
? ? private Session session;
? ? private static CopyOnWriteArraySet<WebSocketTest> webSockets =new CopyOnWriteArraySet<>();
? ? private static Map<String,Session> sessionPool = new HashMap<String,Session>();
? ? @OnOpen
? ? public void onOpen(Session session, @PathParam(value="shopId")String shopId) {
? ? ? ? this.session = session;
? ? ? ? webSockets.add(this);
? ? ? ? sessionPool.put(shopId, session);
? ? ? ? System.out.println("【websocket消息】有新的连接，总数为:"+webSockets.size());
? ? }
? ? @OnClose
? ? public void onClose() {
? ? ? ? webSockets.remove(this);
? ? ? ? System.out.println("【websocket消息】连接断开，总数为:"+webSockets.size());
? ? }
? ? @OnMessage
? ? public void onMessage(String message) {
? ? ? ? System.out.println("【websocket消息】收到客户端消息:"+message);
? ? }
? ? // 此为广播消息
? ? public void sendAllMessage(String message) {
? ? ? ? for(WebSocketTest webSocket : webSockets) {
? ? ? ? ? ? System.out.println("【websocket消息】广播消息:"+message);
? ? ? ? ? ? try {
? ? ? ? ? ? ? ? webSocket.session.getAsyncRemote().sendText(message);
? ? ? ? ? ? } catch (Exception e) {
? ? ? ? ? ? ? ? e.printStackTrace();
? ? ? ? ? ? }
? ? ? ? }
? ? }
? ? // 此为单点消息 (发送文本)
? ? public void sendTextMessage(String shopId, String message) {
? ? ? ? Session session = sessionPool.get(shopId);
? ? ? ? if (session != null) {
? ? ? ? ? ? try {
? ? ? ? ? ? ? ? session.getBasicRemote().sendText(message);
? ? ? ? ? ? } catch (Exception e) {
? ? ? ? ? ? ? ? e.printStackTrace();
? ? ? ? ? ? }
? ? ? ? }
? ? }
? ? // 此为单点消息 (发送对象)
? ? public void sendObjMessage(String shopId, Object message) {
? ? ? ? Session session = sessionPool.get(shopId);
? ? ? ? if (session != null) {
? ? ? ? ? ? try {
? ? ? ? ? ? ? ? session.getAsyncRemote().sendObject(message);
? ? ? ? ? ? } catch (Exception e) {
? ? ? ? ? ? ? ? e.printStackTrace();
? ? ? ? ? ? }
? ? ? ? }
? ? }
};
?
?
