Spring boot3简单实现websocket(拦截器,跨域,服务端发送)

Spring boot3简单实现websocket

效果

在这里插入图片描述

导入WebSocket依赖

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

WebSocketHandler

public class MyWebSocketHandler1 implements WebSocketHandler {

    //保存用户会话信息,用于服务端群发
    private static final ConcurrentLinkedDeque<WebSocketSession> concurrentLinkedDeque = new ConcurrentLinkedDeque<>();

    //保存当前会话用户名称
    private String username;
    
    @Override
    public void afterConnectionEstablished(WebSocketSession session) {
        this.username = (String) session.getAttributes().get("username");
        concurrentLinkedDeque.add(session);
        System.out.println("用户连接:" + username);
        sendMessage(new TextMessage(username + "加入聊天室"));
    }

    @Override
    public void handleMessage(WebSocketSession session, WebSocketMessage<?> message) {
        sendMessage(message);
    }

    public void sendMessage(WebSocketMessage<?> message) {
        concurrentLinkedDeque.forEach(item -> {
            try {
                if (item.isOpen()) {
                    item.sendMessage(message);
                }
            } catch (Exception e) {
                throw new RuntimeException(e);
            }
        });
    }

    @Override
    public void handleTransportError(WebSocketSession session, Throwable exception) throws Exception {

    }

    @Override
    public void afterConnectionClosed(WebSocketSession session, CloseStatus closeStatus) throws Exception {
        sendMessage(new TextMessage(username + "退出聊天室"));
        concurrentLinkedDeque.remove(session);
    }

    @Override
    public boolean supportsPartialMessages() {
        return false;
    }
}

拦截器

public class MyWebSocketHandshakeInterceptor extends HttpSessionHandshakeInterceptor {

    @Override
    public boolean beforeHandshake(ServerHttpRequest request, ServerHttpResponse response, WebSocketHandler wsHandler, Map<String, Object> attributes) {
        if (request instanceof ServletServerHttpRequest servletRequest) {
            String token = servletRequest.getServletRequest().getParameter("token");
            String username = servletRequest.getServletRequest().getParameter("username");
            if (StringUtils.hasLength(token) && token.equals("321123") && StringUtils.hasLength(username)) {
                attributes.put("username", username);
                return true;
            }
        }
        return false;
    }
} 

配置

@Configuration
@EnableWebSocket
public class WebSocketConfig implements WebSocketConfigurer {
    @Override
    public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
        //配置handler,拦截器和跨域
        registry.addHandler(myHandler(), "/ws/{roomId}").addInterceptors(new MyWebSocketHandshakeInterceptor()).setAllowedOrigins("*");
    }

    @Bean
    public WebSocketHandler myHandler() {
        return new MyWebSocketHandler1();
    }

    @Bean
    public ServletServerContainerFactoryBean createWebSocketContainer() {
        ServletServerContainerFactoryBean container = new ServletServerContainerFactoryBean();
        container.setMaxTextMessageBufferSize(8192);  //文本消息最大缓存
        container.setMaxBinaryMessageBufferSize(8192);  //二进制消息大战缓存
        container.setMaxSessionIdleTimeout(3L * 60 * 1000); // 最大闲置时间,3分钟没动自动关闭连接
        container.setAsyncSendTimeout(10L * 1000); //异步发送超时时间
        return container;
    }
}  

Controller

@RestController
public class WebSocketController {

    @GetMapping("/send/{msg}")
    public HttpStatus sendMessage(@PathVariable String msg) {
        MyWebSocketHandler1 myWebSocketHandler1 = new MyWebSocketHandler1();
        myWebSocketHandler1.sendMessage(new TextMessage(msg));
        return HttpStatus.OK;
    }
}
  • 5
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
实现自定义请求头的鉴权可以通过拦截器或者过滤器来实现。 下面是一个基于拦截器实现自定义请求头的鉴权的示例: 1. 创建一个拦截器类,实现HandlerInterceptor接口,并实现preHandle方法,在该方法中进行鉴权操作。 ```java public class WebSocketInterceptor implements HandlerInterceptor { @Override public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception { String token = request.getHeader("Authorization"); // 获取请求头中的Authorization字段 // 进行鉴权操作 if (token == null || !token.equals("your_token")) { response.setStatus(401); return false; } return true; } } ``` 2. 在WebSocket配置类中添加拦截器。 ```java @Configuration @EnableWebSocket public class WebSocketConfig implements WebSocketConfigurer { @Autowired private WebSocketHandler webSocketHandler; @Autowired private WebSocketInterceptor webSocketInterceptor; @Override public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) { registry.addHandler(webSocketHandler, "/ws").addInterceptors(webSocketInterceptor); // 添加拦截器 } } ``` 在上述代码中,我们通过调用addInterceptors方法来添加拦截器,该方法接收一个实现HandlerInterceptor接口的拦截器对象。 在这个示例中,我们创建了一个WebSocketInterceptor拦截器类,该类会在preHandle方法中进行鉴权操作。我们还将该拦截器对象传递给了registerWebSocketHandlers方法,来注册WebSocket处理器。 当客户端连接WebSocket时,WebSocketInterceptor的preHandle方法会被调用,我们可以在该方法中获取请求头中的Authorization字段进行鉴权操作。如果鉴权失败,则返回401状态码,否则返回true,WebSocket连接将会正常建立。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值