SpringBoot实现WebSocket发送接收消息 + Vue实现SocketJs接收发送消息

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

package com.cc.ws.config;

import org.springframework.context.annotation.Configuration;
import org.springframework.messaging.simp.config.MessageBrokerRegistry;
import org.springframework.web.socket.config.annotation.EnableWebSocketMessageBroker;
import org.springframework.web.socket.config.annotation.StompEndpointRegistry;
import org.springframework.web.socket.config.annotation.WebSocketMessageBrokerConfigurer;

/**
 * <p>@EnableWebSocketMessageBroker 的作用</p>
 * <li>注解开启使用STOMP协议来传输基于代理(message broker)的消息,</li>
 * <li>这时控制器支持使用 @MessageMapping,就像使用 @RequestMapping一样</li>
 * @author cc
 *
 */
@Configuration
@EnableWebSocketMessageBroker
public class WebSocketConfig implements WebSocketMessageBrokerConfigurer {

    /** <p>启动简单Broker</p>
     * <p>表示客户端订阅地址的前缀信息，也就是客户端接收服务端消息的地址的前缀信息</p>
     * <p>代理的名字：都是自定义的</p>
     *
     * /user     点对点（默认也是/user，可以自定义，但是必须和setUserDestinationPrefix中的设置一致）
     * /topic1   广播模式1
     * /topic2   广播模式2
     *
     * /mass     广播模式：群发
     */
    @Override
    public void configureMessageBroker(MessageBrokerRegistry registry) {
        registry.enableSimpleBroker(
                "/user", "/topic1", "/topic2", "/mass"
        );
        // 点对点使用的订阅前缀（客户端订阅路径上会体现出来），不设置的话，默认也是 /user/
        // 注意，这里必须和上面设置的Broker：/user 一致（两个都可以自定义，但必须一致）。否则连接不上
        registry.setUserDestinationPrefix("/user/");
        // 指服务端接收地址的前缀，意思就是说客户端给服务端发消息的地址的前缀
//        registry.setApplicationDestinationPrefixes("/socket");
    }

    /**
     * 这个方法的作用是添加一个服务端点，来接收客户端的连接。
     * registry.addEndpoint("/socket")表示添加了一个/socket端点，客户端（前端）就可以通过这个端点来进行连接。
     * withSockJS()的作用是开启SockJS支持。
     * @param registry registry
     */
    @Override
    public void registerStompEndpoints(StompEndpointRegistry registry) {
        // 注册一个STOMP的endpoint端点,并指定使用SockJS协议
        // 前端使用这个地址连接后端 WebSocket接口
        registry.addEndpoint("/broadcast", "/point")
                // 允许所有源跨域。还可以指定ip配置：http://ip:*
                // 低版本的SpringBoot（2.1.5.RELEASE 就不行）不行
                .setAllowedOriginPatterns("*")
                .withSockJS();
    }

}

@SpringBootApplication
@EnableScheduling
public class WebSocketDemoApplication {
    public static void main(String[] args) {
        SpringApplication.run(WebSocketDemoApplication.class, args);
    }
}

    let socket1 = new SockJS('http://服务器ip:服务器端口/broadcast');
   
    let stompClient1 = Stomp.over(socket1);//广播模式

    stompClient1.connect({}, (frame) => {
      stompClient1.subscribe('/topic1/', (message) => {
        console.log(message.body);
      });
    });

    @Resource
    private SimpMessagingTemplate simpMsgTemp;

    /** 广播推送消息1：会发送给所有连接了 topic1 这个通道的客户端。
     * topic1：在Broker中配置
     **/
    @Scheduled(cron = "0/1 * * * * ?")
    public void getSocket(){
        String msg = String.format("%s 的第 %s 个消息", "topic1", LocalDateTime.now().getSecond());
        log.info("{}",msg);
        simpMsgTemp.convertAndSend("/topic1/", msg);
    }

    /** 广播推送消息2：多指定一个uri。相当于另一条通道（推荐使用）
     * <li>自定义url后缀，还可以实现用户和用户单点发送。</li>
     * topic2：在Broker中配置
     * custom：是自定义的
     */
    @Scheduled(cron = "0/1 * * * * ?")
    public void getSocketUser(){
        String msg = String.format("topic2 的第 %s 个消息", LocalDateTime.now().getSecond());
        log.info("{}",msg);
        simpMsgTemp.convertAndSend("/topic2/custom" ,msg);
    }

    /**点对点发送 convertAndSendToUser（第一个参数：一般是用户id）
     *  -> 假如用户id是1。用用户id是1的在两个地方登陆了客户端（比如不同的浏览器登陆同一个用户），
     *  -> convertAndSendToUser会把消息发送到用户1登陆的两个客户端中
     * 发送到：/user/{userId}/cs 下。cs是自定义的,且必须自定义一个。
     */
    @Scheduled(cron = "0/1 * * * * ?")
    public void pointToPoint(){
        //这个用户id是后端获取的当前登陆的用户id
        String userId = "123";
        String msg = String.format("点对点：第 %s 个消息。用户id：%s", LocalDateTime.now().getSecond(), userId);
        log.info("{}",msg);
        //发送
        simpMsgTemp.convertAndSendToUser(userId,"/cs/" ,msg);
    }

    //这样才能同时接收后端来的三套不同通道的消息。
    // broadcast 和后端：registerStompEndpoints中的配置必须一致
    // point     和后端：registerStompEndpoints中的配置必须一致
    // broadcast、point 也可以只用一个，这里只是为了好区分。
    let socket1 = new SockJS('http://172.16.8.1:8099/broadcast');
    let socket2 = new SockJS('http://172.16.8.1:8099/broadcast');
    let socket3 = new SockJS('http://172.16.8.1:8099/point');
    // console.log("wb：" + socket)

    let stompClient1 = Stomp.over(socket1);
    let stompClient2 = Stomp.over(socket2);
    let stompClient3 = Stomp.over(socket3);

    // ----------------广播模式1--------------------
    stompClient1.connect({}, (frame) => {
      console.log('-----------frame1', frame)
      stompClient1.subscribe('/topic1/', (message) => {
        console.log(message.body);
        this.msg = message.body;
        // console.log(JSON.parse(message.body));
      });
    });

    // ----------------广播模式2--------------------
    stompClient2.connect({}, (frame) => {
      console.log('-----------frame2', frame)
      stompClient2.subscribe('/topic2/custom', (message) => {
        console.log(message.body);
        this.user = message.body;
        // console.log(JSON.parse(message.body));
      });
    });

    // ----------------点对点模式--------------------
    //前端获取的 userId
    let userId = '123';
    //连接WebSocket服务端
    stompClient3.connect({},(frame) => {
      console.log('Connected:' + frame);
      stompClient3.subscribe('/user/' + userId + '/cs/',
          (response) => {
            this.peer = response.body;
      });
    });

    @Resource
    private SimpMessagingTemplate simpMsgTemp;

    /** <p>广播模式一、接收前端的消息，处理后给前端返回一个消息。</p>
     * <li>后端 把消息处理后 发送到 /mass/getResponse 路径下</li>
     * <ol>
     *     <li>@MessageMapping("/massRequest1") ：作用：接收前端来的消息。类似于@RestController</li>
     *     <li>@SendTo("/mass/getResponse1")：作用跟convertAndSend类似，广播发给与该通道相连的客户端。SendTo 发送至 Broker 下的指定订阅路径 </li>
     *     <li>@SendToUser("/mass/getResponse1")：作用跟convertAndSendToUser类似，定点发送。SendTo 发送至 Broker 下的指定订阅路径 </li>
     *     <li>/mass 必须在配置文件配置</li>
     *     <li>/getResponse1 自定义的后缀</li>
     * </ol>
     */
    @MessageMapping("/massRequest1")
    @SendTo("/mass/getResponse1")
    public String mass1(String chatRoomRequest){
        //处理前端消息……
        log.info("前端消息：{}",chatRoomRequest);
        //返回消息
        return "@SendTo 广播一（单次） 后端处理完成！";
    }

    /** 广播模式二、接收前端的消息，可以多次给前端发消息
     * <li>/mass 必须在配置文件配置</li>
     * <li>/getResponse2 自定义的后缀</li>
     */
    @MessageMapping("/massRequest2")
    public void mass2(String chatRoomRequest){
        log.info("前端的消息：{}",chatRoomRequest);

        for (int i = 0; i < 5; i++) {
            String msg = "后端处理后的 广播二（多次）：" + i;
            simpMsgTemp.convertAndSend("/mass/getResponse2", msg);
        }
        simpMsgTemp.convertAndSend("/mass/getResponse2",
                "后端处理后的 广播二（多次），后端处理完成！");
    }

    /** <p>点对点一、接收前端消息，只能返回一次消息（必须登陆系统才能使用。）</p>
     * <li>只有发送原始消息的客户端才会收到响应消息，而不是所有连接的客户端都会收到响应消息。</li>
     * <li>/alone/getResponse1：自定义的，不需要在配置文件配置。</li>
     *
     * <p>@SendToUser</p>
     * <li>默认该注解前缀为 /user</li>
     * <li>broadcast属性，表明是否广播。就是当有同一个用户登录多个session时，是否都能收到。取值true/false.</li>
     *
     * @param principal Principal ：登陆用户的信息，需要使用spring s安全框架写入信息？
     */
    @MessageMapping("/aloneRequest1")
    @SendToUser("/alone/getResponse1")
    public String alone1(String chatRoomRequest){
        //处理前端消息……
        log.info("前端消息：{}",chatRoomRequest);
        //返回消息
        return "@SendToUser 点对点一（单次） 后端处理完成！";
    }

    /** 点对点二、接收前端消息，可以多次给前端发消息
     * <li>convertAndSendToUser —— 发送消息给指定用户id的</li>
     * <li>如果用户1在两个地方（A/B）登陆可以客户端，并且连接了该通道，其中一个如A给后端发消息，后端返回消息，A/B两个地方都会收到消息</li>
     * <ol>
     *     <li>@MessageMapping("/aloneRequest2") 接收前端指定用户消息，</li>
     *     <li>/alone/getResponse2 不用在配置文件中配置</li>
     *     <li>返回消息 发送到 user/{userId}/alone/getResponse2 下 （定点发送）</li>
     * </ol>
     */
    @MessageMapping("/aloneRequest2")
    public void alone2(String chatRoomRequest){
        //后端获取的当前登陆的用户的id（和前端一致）
        String userId = "456";
        log.info("前端的消息：{}",chatRoomRequest);

        for (int i = 0; i < 5; i++) {
            String msg = "后端处理后的 点对点二（多次）：" + i;
            simpMsgTemp.convertAndSendToUser(userId,"/alone/getResponse2", msg);
        }
        simpMsgTemp.convertAndSendToUser(userId,"/alone/getResponse2",
                "后端处理后的 点对点二（多次），后端处理完成！");
    }

    //连接SockJS的 broadcast
    let socket1 = new SockJS('http://172.16.8.7:8099/broadcast');
    let socket2 = new SockJS('http://172.16.8.7:8099/broadcast');
    let socket3 = new SockJS('http://172.16.8.7:8099/point');
    let socket4 = new SockJS('http://172.16.8.7:8099/point');
    //使用STMOP子协议的WebSocket客户端
    let stompClient1 = Stomp.over(socket1);
    let stompClient2 = Stomp.over(socket2);
    let stompClient3 = Stomp.over(socket3);
    let stompClient4 = Stomp.over(socket4);

    //1广播模式一
    stompClient1.connect({},(frame) => {
      console.log('广播模式一:' + frame);
      //1发送消息
      stompClient1.send("/massRequest1",{},"我是前端来 广播模式一 的消息！");

      //2接收消息
      stompClient1.subscribe('/mass/getResponse1',(response) => {
        this.broadcast1 = response.body
      });
    });

    //2广播模式二
    stompClient2.connect({},(frame) => {
      console.log('广播模式二:' + frame);
      //1发送消息
      stompClient2.send("/massRequest2",{},"我是前端来 广播模式二 的消息");

      //2接收消息
      stompClient2.subscribe('/mass/getResponse2',(response) => {
        this.broadcast2 = response.body
      });
    });

    //3点对点一 ：必须登陆系统才能实现。要往：Principal设置用户登陆信息才行
    //1发送消息
    // stompClient3.send("/aloneRequest1",{},"我是前端来 点对点一 的消息");
    stompClient3.connect({},(frame) => {
      console.log('点对点一1:' + frame);
      stompClient3.send("/aloneRequest1",{},"我是前端来 点对点一 的消息");
      //2接收消息
      stompClient3.subscribe('/user/alone/getResponse1' ,(response) => {
        console.log('-------response.body', response.body)
        this.point1 = response.body
      });
    });

    //4点对点二：必须获取现在登陆了的用户id，且必须和后端一致才行。
    stompClient4.connect({},(frame) => {
      console.log('点对点二:' + frame);
      //1发送消息
      stompClient4.send("/aloneRequest2",{},"我是前端来 点对点二 的消息");

      //2接收消息
      //前端获取的当前登陆的用户userId（和后端一致）
      let userId = '456';
      stompClient4.subscribe('/user/'+userId+'/alone/getResponse2',(response) => {
        this.point2 = response.body
      });
    });