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 {
 
    /**
     * 连接成功调用
     */
    @Override
    public void afterConnectionEstablished(WebSocketSession session) throws Exception {
        super.afterConnectionEstablished(session);
    }
 
    /**
     * 收到消息时调用
     */
    @Override
    public void handleMessage(WebSocketSession session, WebSocketMessage<?> message) throws Exception {
        super.handleMessage(session, message);
    }
 
    /**
     * 关闭连接时调用
     */
    @Override
    public void afterConnectionClosed(WebSocketSession session, CloseStatus status) throws Exception {
        super.afterConnectionClosed(session, status);
    }
 
    /**
     * 发生错误时调用
     */
    @Override
    public 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 {
 
    @Override
    public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
        //配置在线聊天处理器
        registry.addHandler(new WebSocketHandler(), "/chat")
                .setAllowedOrigins("*");
    }
}

问题

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

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

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

java 代码解读复制代码#限制Tomcat接收的HTTP请求的大小，包括WebSocket消息的大小限制，修改为10m

server:

max-http-header-size: 10485760

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);

+友情链接