Websocket集群解决方案( 三 ) _生活百科

获取服务器IP和端口可以具体查看Github源码,这里就不做详细描述了。

修改服务端点 ServerEndpoint在WebSocket添加消息的接收方法， @RabbitListener 接收消息，队列名称使用常量命名，动态队列名称使用 #{name}，其中的name是Queue的bean 名称：

@RabbitListener(queues= "#{psQueue.name}")public void pubsubQueueFirst(String message) {System.out.println(message);sendMessage(message);}

然后再调用sendMessage方法发送给所在连接的客户端。
修改消息发送在WebSocket类的onMessage方法将消息发送改成RabbitMQ方式发送:

@OnMessagepublic void onMessage(String message){if (!message.equals("ping")){log.info("【wesocket】收到客户端发送的消息,message={}",message);//sendMessage(message);if (rabbitTemplate == null) {rabbitTemplate = (RabbitTemplate) SpringContextUtil.getBean("rabbitTemplate");}rabbitTemplate.convertAndSend("PUBLISH_SUBSCRIBE_EXCHANGE", null, message);}}

消息通知流程如下所示：