springboot整合websocket两种方式

最新推荐文章于 2024-07-19 11:10:22 发布

Shemuel_Deng

最新推荐文章于 2024-07-19 11:10:22 发布

阅读量2.6k

点赞数 4

分类专栏： Java 文章标签： websocket springboot stomp

本文链接：https://blog.csdn.net/qq_35249342/article/details/119324967

版权

本文介绍了两种在SpringBoot中整合WebSocket的方法：1) 利用html5原生支持，通过ServerEndpointExporter和serverEndpoint配置；2) 使用STOMP协议，添加websocket配置，重写DefaultHandshakeHandler并实现WebSocketMessageBrokerConfigurer，前端使用Vue实现订阅和发送消息。

摘要由CSDN通过智能技术生成

方式1：html5原生支持方式

向spring容器中注入一个ServerEndpointExporter

package com.tinet.websocket.config;

import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.socket.server.standard.ServerEndpointExporter;

/**
 * @author dsx
 */
@Configuration
public class WebsocketConfig {
   

    /**
     * ServerEndpointExporter 作用
     *
     * 这个Bean会自动注册使用@ServerEndpoint注解声明的websocket endpoint
     *
     * @return
     */
    @Bean
    public ServerEndpointExporter serverEndpointExporter(){
   
        return new ServerEndpointExporter();
    }
}

添加一个serverEndpoint

用于给前端建立连接的server

package com.tinet.websocket.config;

import com.alibaba.fastjson.JSON;
import com.tinet.websocket.pojo.MessageBody;
import org.springframework.stereotype.Component;

import javax.websocket.*;
import javax.websocket.server.PathParam;
import javax.websocket.server.ServerEndpoint;
import java.io.IOException;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.atomic.AtomicInteger;


/**
 * @author DENGSHAOXIANG
 */
@ServerEndpoint("/websocket/test/{userId}")
@Component
public class WebsocketServer {
   

    public WebsocketServer() {
   
        //每当有一个连接，都会执行一次构造方法
        System.out.println("新的连接。。。");
    }

    /**
     * 存放当前连接数
     */
    private static final AtomicInteger COUNT = new AtomicInteger();


    /**
     * 存放所有的连接
     */
    private static final ConcurrentHashMap<String, Session> SESSIONS = new ConcurrentHashMap<>();


    //发送消息
    public void sendMessage(Session toSession, String message) {
   
        if (toSession != null) {
   
            try {
   
                toSession.getBasicRemote().sendText(message);
            } catch (IOException e) {
   
                e.printStackTrace();
            }
        } else {
   
            System.out.println("对方不在线");
        }
    }

    public void sendMessageToUser(String user, String message) {
   
        Session toSession = SESSIONS.get(user);
        sendMessage(toSession, message);
    }


    @OnMessage
    public void onMessage(String message) {
   
        System.out.println("服务器收到消息：" + message);
        MessageBody messageBody = JSON.parseObject(message, MessageBody.class);
        String userId = messageBody.getUserId();
        sendMessageToUser(userId, messageBody.getMessage());


    }

    @OnOpen
    public void onOpen(Session session, @PathParam("userId") String userId) {
   
        if (SESSIONS.get(userId) != null) {
   
            return;
        }
        SESSIONS.put(userId, session);
        COUNT.incrementAndGet();
        System.out.println(userId + "上线了，当前在线人数：" + COUNT);

    }

    @OnClose
    public void onClose(@PathParam("userId") String userId) {
   
        SESSIONS.remove(userId);
        COUNT.decrementAndGet();
        System.out.println(userId + "下线了ÿ