WebSocket的使用（消息推送）

weixin_48570662

已于 2022-06-17 11:45:31 修改

阅读量724

点赞数 1

文章标签： websocket 网络协议网络 java javascript

于 2022-06-17 11:44:29 首次发布

本文链接：https://blog.csdn.net/weixin_48570662/article/details/125330745

版权

1.引入依赖

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

2.配置类

@Configuration
public class WebSocketConfig {

    @Bean
    public ServerEndpointExporter getServerEndpointExporter(){
        return new ServerEndpointExporter();
    }
}

3.业务方法

@Component
@ServerEndpoint("/webSocket/{oid}")
public class WebSocketService {
    private static ConcurrentHashMap<String,Session> sessionMap = new ConcurrentHashMap<>();
    /**
     * 前端发送请求建立WebSocket连接就会执行@OnOpen方法
     */
    @OnOpen
    public void open(@PathParam("oid")String orderId, Session session){
        sessionMap.put(orderId,session);
    }
    /**
     * 前端关闭页面或者主动关闭webSocket连接，都会执行close
     */
    @OnClose
    public void close(@PathParam("oid")String orderId){
        sessionMap.remove(orderId);
    }

    //封装的发送消息方法
    public static void  sendMessage(String orderId,String message){
        try {
            Session session = sessionMap.get(orderId);
            session.getBasicRemote().sendText(message);
        } catch (IOException e) {
            e.printStackTrace();
        }

    }
}

当需要向前端推送消息时调用sendMessage方法根据orderId拿到session发送消息

4.前端发送连接请求和前端接收请求

// 前端发送websocket连接请求
					var websocket = null;
					var webSocketUrl =  webSocketBaseUrl +"webSocket/" + this.orderInfo.orderId;
					websocket = new WebSocket(webSocketUrl);
					// 只要后端通过websocket向此连接发送消息就会触发onmessage事件
					websocket.onmessage = function(event){
						console.log(event)
						var msg = event.data;
						if(msg == "1"){
							$("#div1").html("<label style='font-size:20px; color:green'>订单支付完成!</label>")
						}

					}