简洁明了！Springboot整合WebSocket（两种方法及问题解决）

第一种方法（继承TextWebSocketHandler）

该方法一共两步

首先需要创建一个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的服务已经可以使用了，如需要在创建新的处理器只需要创建好websocket处理器后按同样的方法注入即可

第二种方法

待写~

问题

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) {
        ChatWebSocketHandler.applicationContext = applicationContext;
    }

并修改启动类，使项目在启动时将上下文注入到WebSocket处理器中，通过getBean获取service和mapper对象

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