springboot+websocket+sockjs进行点对点消息发送

￡神魔★判官ぃ 2022-08-28 12:44 174阅读 0赞

#  #

1.添加maven相关依赖

--------------------

<dependency>
                <groupId>org.springframework.boot</groupId>
                <artifactId>spring-boot-starter-websocket</artifactId>
                <version>2.1.6.RELEASE</version>
            </dependency>

2.添加websocket配置类

--------------------

@Configuration
    @EnableWebSocketMessageBroker
    public class WebSocketConfig implements WebSocketMessageBrokerConfigurer {
        @Override
        public void registerStompEndpoints(StompEndpointRegistry registry) {
            registry.addEndpoint("/socket").setAllowedOrigins("*").withSockJS();
        }
        @Override
        public void configureMessageBroker(MessageBrokerRegistry registry) {
            registry.enableSimpleBroker("/user","/message");
        }
    }

继承AbstractWebSocketMessageBrokerConfigurer类实现 registerStompEndpoints方法，其中registry.addEndpoint("/socket").setAllowedOrigins("\*").withSockJS()是指用来注册一个STOMP节点，并指定映射的url为"/socket"，同时指定使用SockJS协议。registry.enableSimpleBroker用于配置消息代理，"/user"配置的是点对点代理，即在线聊天，"/message"配置的是消息发出后的通知的路径。

3.前端实现js部分代码

--------------------

SockJs后的端口号对应了位置类中的"/socket",stompClient.subscribe表示接收消息的路径，stompClient.send("/message"表示发送消息的路径。

4.Controller部分

--------------------

@Controller
    public class ChatController {
        @Autowired
        private SimpMessagingTemplate simpMessagingTemplate;
        @GetMapping("/user/{id}")
        public String userChat(@PathVariable(value = "id") int id, Model model){
            model.addAttribute("id",id);
            return "chat";
        }
    
        @MessageMapping("/message")
        public void sendMessage(Message message){
            simpMessagingTemplate.convertAndSendToUser(message.getToUser(),"/message/get",message.getMsg());
            simpMessagingTemplate.convertAndSendToUser(message.getFromUser(),"/message/get",message.getMsg());
        }
    }

自动注入SimpMessagingTemplate类用来发送消息到前端，"/user"路径通过传进的id不同来区分发送消息的人，服务器接收消息的接口是"/message"，接收到以后在convertAndSendToUser方法调用发送给对应的id的用户.综上所述实现了一个简单的一对一单点发消息功能。