JeecgBoot关于websocket的改进意见

1,环境描述

Jeecgboot3.0

2,websocket权限认证

在shiroConfig.java代码中注释掉

filterChainDefinitionMap.put("/websocket/**", "anon");//系统通知和公告

配置拦截器

package org.jeecg.config.websocket;


import org.apache.commons.lang.StringUtils;
import org.springframework.http.server.ServerHttpRequest;
import org.springframework.http.server.ServerHttpResponse;
import org.springframework.http.server.ServletServerHttpRequest;
import org.springframework.http.server.ServletServerHttpResponse;
import org.springframework.stereotype.Component;
import org.springframework.web.socket.WebSocketHandler;
import org.springframework.web.socket.server.support.HttpSessionHandshakeInterceptor;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.util.Map;

@Component
public class HandshakeInterceptor extends HttpSessionHandshakeInterceptor {

    @Override
    public boolean beforeHandshake(ServerHttpRequest request,
                                   ServerHttpResponse response, WebSocketHandler wsHandler,
                                   Map<String, Object> attributes) throws Exception {
        return super.beforeHandshake(request, response, wsHandler, attributes);
    }

    @Override
    public void afterHandshake(ServerHttpRequest request,
                               ServerHttpResponse response, WebSocketHandler wsHandler,
                               Exception ex) {

        HttpServletRequest httpServletRequest = ((ServletServerHttpRequest)request).getServletRequest();
        HttpServletResponse httpServletResponse = ((ServletServerHttpResponse)response).getServletResponse();
        if(StringUtils.isNotEmpty(httpServletRequest.getHeader("sec-websocket-protocol")))
            httpServletResponse.addHeader("sec-websocket-protocol", httpServletRequest.getHeader("sec-websocket-protocol"));
        super.afterHandshake(request, response, wsHandler, ex);
    }
}

修改前端的请求,在HeaderNotice.vue中,

let token = Vue.ls.get(ACCESS_TOKEN)

var url = window._CONFIG['domianURL'].replace("https://","wss://").replace("http://","ws://")+"/websocket/"+userId +"?token="+token

3, HashMap不是线程安全的的,可以改为线程安全的map,即是

    private static Map<String, Session> sessionPool = new ConcurrentHashMap<>();

4,重复发送问题

原有的代码为

 /**
     * 服务器端推送消息
     */
    public void pushMessage(String message) {
        try {
            webSockets.forEach(ws -> ws.session.getAsyncRemote().sendText(message));
        } catch (Exception e) {
            e.printStackTrace();
        }
    }


    @OnMessage
    public void onMessage(String message) {
        //todo 现在有个定时任务刷,应该去掉
        log.debug("【websocket消息】收到客户端消息:" + message);
        JSONObject obj = new JSONObject();
        //业务类型
        obj.put(WebsocketConst.MSG_CMD, WebsocketConst.CMD_CHECK);
        //消息内容
        obj.put(WebsocketConst.MSG_TXT, "心跳响应");
        for (WebSocket webSocket : webSockets) {
            webSocket.pushMessage(message);
        }
    }

可将OnMessage函数改为



    @OnMessage
    public void onMessage(String message) {
        //todo 现在有个定时任务刷,应该去掉
        log.debug("【websocket消息】收到客户端消息:" + message);
        JSONObject obj = new JSONObject();
        //业务类型
        obj.put(WebsocketConst.MSG_CMD, WebsocketConst.CMD_CHECK);
        //消息内容
        obj.put(WebsocketConst.MSG_TXT, "心跳响应");
        for (WebSocket webSocket : webSockets) {
//            webSocket.pushMessage(message);
            webSocket.session.getAsyncRemote().sendText(message);
        }
    }

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
Spring Boot 配置 WebSocket 的过程通常包括以下几个步骤: 1. 添加依赖:在你的 `pom.xml` 或者 `build.gradle` 文件中添加 Spring Websocket 和相关的库,如 Stomp 或者 WebSocket 支持。 ```xml <!-- Maven --> <dependency> <groupId>org.springframework.boot</groupId> <artifactId>spring-boot-starter-websocket</artifactId> </dependency> <!-- Gradle --> implementation 'org.springframework.boot:spring-boot-starter-websocket' ``` 2. 配置WebSocket:在 `application.yml` 或 `application.properties` 中,启用 WebSocket 并指定一个默认的端口(如果需要的话)。 ```yaml server: port: 8080 web: socket: port: 8080 ``` 3. 创建WebSocket处理器:创建一个实现了 `TextMessageHandler` 或 `BinaryMessageHandler` 接口的类,并实现消息处理逻辑。 ```java import org.springframework.messaging.handler.annotation.MessageMapping; import org.springframework.messaging.simp.SimpMessageSendingOperations; @Service public class ChatWebSocketHandler { private final SimpMessageSendingOperations messageSendingOperations; @Autowired public ChatWebSocketHandler(SimpMessageSendingOperations messageSendingOperations) { this.messageSendingOperations = messageSendingOperations; } @MessageMapping("/chat") public void handleMessage(String message) { // 处理接收到的消息,可能需要广播或发送给特定用户 messageSendingOperations.convertAndSendToUser("user1", "/topic/chat", message); } } ``` 4. 注册WebSocket处理器:使用 `@EnableWebSocketMessageBroker` 注解和 `messageBroker` 配置来启用 WebSocket 消息代理。 ```java import org.springframework.context.annotation.Configuration; import org.springframework.messaging.simp.config.MessageBrokerRegistry; import org.springframework.web.socket.config.annotation.AbstractWebSocketMessageBrokerConfigurer; import org.springframework.web.socket.config.annotation.EnableWebSocketMessageBroker; import org.springframework.web.socket.config.annotation.StompEndpointRegistry; @Configuration @EnableWebSocketMessageBroker public class WebSocketConfig extends AbstractWebSocketMessageBrokerConfigurer { @Override public void configureMessageBroker(MessageBrokerRegistry config) { config.enableSimpleBroker("/topic"); config.setApplicationDestinationPrefixes("/app"); } @Override public void registerStompEndpoints(StompEndpointRegistry registry) { registry.addEndpoint("/ws").withSockJS(); } } ``` 5. 访问WebSocket:在前端,你可以使用 JavaScript 的 WebSocket API 或者第三方库(如 Socket.io)来连接到服务器。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值