springboot 动态websocket

Spring Boot 支持使用 Spring WebSocket 模块来实现 WebSocket 功能。在 Spring WebSocket 中，我们可以通过实现 WebSocketHandler 接口来处理 WebSocket 请求，也可以使用 @Controller 注解和 @MessageMapping 注解来处理 WebSocket 请求。

动态 WebSocket 可以通过在后端使用 WebSocketSession 对象动态向前端推送数据。我们可以在后端创建一个 WebSocketSession 对象，然后将其存储在一个 Map 中，当需要向前端推送数据时，可以根据前端的连接 ID 从 Map 中获取相应的 WebSocketSession 对象，然后使用该对象向前端推送数据。

下面是一个使用 Spring WebSocket 实现动态 WebSocket 的示例代码：

@Controller
public class WebSocketController {
    private static final Map<String, WebSocketSession> sessions = new ConcurrentHashMap<>();

    @Autowired
    private SimpMessagingTemplate messagingTemplate;

    @MessageMapping("/connect")
    public void connect(Principal principal, WebSocketSession session) throws Exception {
        sessions.put(principal.getName(), session);
    }

    @MessageMapping("/push")
    public void push(Principal principal, String message) throws Exception {
        WebSocketSession session = sessions.get(principal.getName());
        if (session != null && session.isOpen()) {
            messagingTemplate.convertAndSendToUser(principal.getName(), "/topic/push", message);
        }
    }

    @MessageMapping("/disconnect")
    public void disconnect(Principal principal) throws Exception {
        sessions.remove(principal.getName());
    }
}

在上述代码中，我们使用了 @Controller 注解来标注 WebSocketController 类，并使用 @MessageMapping 注解来标注处理 WebSocket 请求的方法。在 connect 方法中，我们将 WebSocketSession 对象存储在 sessions Map 中，并在 push 方法中使用 messagingTemplate 对象向前端推送数据。disconnect 方法用于在断开连接时从 sessions Map 中移除 WebSocketSession 对象。

在前端实现时，我们可以使用 Stomp.js 库来实现 WebSocket 连接和消息推送。下面是一个使用 Stomp.js 实现动态 WebSocket 的示例代码：

var stompClient = null;

function connect() {
    var socket = new SockJS('/websocket');
    stompClient = Stomp.over(socket);
    stompClient.connect({}, function(frame) {
        stompClient.subscribe('/user/topic/push', function(message) {
            var data = JSON.parse(message.body);
            // 处理推送的数据
        });
        stompClient.send('/app/connect');
    });
}

function push(data) {
    stompClient.send('/app/push', {}, data);
}

function disconnect() {
    stompClient.send('/app/disconnect');
    stompClient.disconnect();
}

在上述代码中，我们使用 connect 方法来建立 WebSocket 连接，使用 push 方法来向后端推送数据，使用 disconnect 方法来断开 WebSocket 连接。在 connect 方法中，我们使用 Stomp.over 方法创建一个 Stomp 客户端，并使用 subscribe 方法来订阅推送消息。在 push 方法中，我们使用 send 方法来向后端推送数据。在 disconnect 方法中，我们使用 send 方法来通知后端断开 WebSocket 连接，并使用 disconnect 方法来关闭 Stomp 客户端。