SpringBoot整合WebSocket两步曲

本文主要是介绍SpringBoot整合WebSocket两步曲，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

• 首先需要创建一个websocket处理器，该类需要继承TextWebSocketHandler并重写里面的方法

import org.springframework.web.socket.CloseStatus;
import org.springframework.web.socket.WebSocketMessage;
import org.springframework.web.socket.WebSocketSession;
import org.springframework.web.socket.handler.TextWebSocketHandler;public class WebSocketHandler extends TextWebSocketHandler {/*** 连接成功调用*/@Overridepublic void afterConnectionEstablished(WebSocketSession session) throws Exception {super.afterConnectionEstablished(session);}/*** 收到消息时调用*/@Overridepublic void handleMessage(WebSocketSession session, WebSocketMessage<?> message) throws Exception {super.handleMessage(session, message);}/*** 关闭连接时调用*/@Overridepublic void afterConnectionClosed(WebSocketSession session, CloseStatus status) throws Exception {super.afterConnectionClosed(session, status);}/*** 发生错误时调用*/@Overridepublic void handleTransportError(WebSocketSession session, Throwable exception) throws Exception {super.handleTransportError(session, exception);}
}

创建好websocket处理器后添加配置类，将websocket处理器注入容器

import org.springframework.context.annotation.Configuration;
import org.springframework.web.socket.config.annotation.*;
import com.piim.handler.ChatWebSocketHandler;@Configuration
@EnableWebSocket
public class WebSocketConfig implements WebSocketConfigurer {@Overridepublic void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {//配置在线聊天处理器registry.addHandler(new WebSocketHandler(), "/chat").setAllowedOrigins("*");}
}

• 问题

  1. WebSocket文本消息长度限制，如图片转的base64无法发送

     原因：Http请求的大小默认大小为0.8kb

     解决方法：在配置文件中添加该配置

     #限制Tomcat接收的HTTP请求的大小，包括WebSocket消息的大小限制，修改为10m
     server:max-http-header-size: 10485760
     

  2. WebSocket中无法使用mapper和service，注入为null
     原因：spring 默认管理的是单例，所以只会注入一次 service。当新用户进入聊天时，系统又会创建一个新的 websocket 对象，spring 管理的都是单例，不会给第二个 websocket 对象注入 service，所以导致只要是用户连接创建的 websocket 对象，都不能再注入，mapper同理
     解决方法：

     在WebSocket处理器添加spring上下文对象并创建对应的set方法

     //解决无法注入service和mapper问题
     private static ApplicationContext applicationContext;public static void setApplicationContext(ApplicationContext applicationContext) {WebSocketHandler.applicationContext = applicationContext;}
     

     并在启动类中将spring上下文对象注入

     //解决websocketServer无法注入mapper问题
     SpringApplication springApplication = new SpringApplication(PiImApplication.class);
     ConfigurableApplicationContext configurableApplicationContext = springApplication.run(args);
     ChatWebSocketHandler.setApplicationContext(configurableApplicationContext);
     

     通过getBean获取service和mapper对象

      MessageService messageService = applicationContext.getBean(MessageService.class);
     

这篇关于SpringBoot整合WebSocket两步曲的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！

---