Spring无STOMP WebSocket服务器实现报错：IllegalStateException调度器异常

看起来你在实现无STOMP的Spring WebSocket服务器时碰到了这个常见的调度器配置问题，我来帮你梳理下可能的原因和解决办法：

异常原因分析 #

java.lang.IllegalStateException: Unexpected use of scheduler 通常是因为WebSocket处理流程中尝试使用了未正确配置的任务调度器——要么是你的WebSocketHandler在异步处理消息时依赖了未初始化的调度器，要么是Spring内部的WebSocket组件需要调度器但没有找到合适的Bean。

具体解决步骤 #

1. 显式配置TaskScheduler Bean #

在你的WebSocketConfig类中添加一个TaskScheduler的Bean定义，Spring WebSocket在处理异步操作时会用到这个调度器：

@Configuration
@EnableWebSocket
public class WebSocketConfig implements WebSocketConfigurer {
    @Autowired
    private WebSocketHandler webSocketHandler;

    // 显式配置WebSocket任务调度器
    @Bean
    public TaskScheduler webSocketTaskScheduler() {
        ThreadPoolTaskScheduler scheduler = new ThreadPoolTaskScheduler();
        scheduler.setThreadNamePrefix("ws-task-scheduler-");
        scheduler.setPoolSize(3); // 根据你的并发需求调整大小
        scheduler.initialize();
        return scheduler;
    }

    @Override
    public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
        registry.addHandler(webSocketHandler, "/app");
    }
}

2. 检查WebSocketHandler的实现 #

确保你的WebSocketHandler没有错误地使用异步逻辑或者依赖未配置的调度资源：

• 避免在handleTextMessage等核心方法中直接使用@Async注解（如果要用，需要确保已经配置了@EnableAsync和对应的线程池）
• 如果需要异步处理消息，建议在Handler内部使用自定义的线程池，而不是依赖Spring默认的调度器，示例如下：

@Component
public class CustomWebSocketHandler extends TextWebSocketHandler {
    // 自定义线程池用于异步消息处理
    private final ExecutorService executor = Executors.newFixedThreadPool(4);

    @Override
    protected void handleTextMessage(WebSocketSession session, TextMessage message) throws Exception {
        String payload = message.getPayload();
        // 异步处理消息，避免阻塞WebSocket连接线程
        executor.submit(() -> {
            try {
                session.sendMessage(new TextMessage("Processed: " + payload));
            } catch (IOException e) {
                // 处理异常
                e.printStackTrace();
            }
        });
    }

    // 记得在容器销毁时关闭线程池
    @PreDestroy
    public void shutdownExecutor() {
        executor.shutdown();
    }
}

3. 排查依赖Bean的调度需求 #

如果你的WebSocketHandler注入了其他Bean，检查这些Bean是否带有@Scheduled注解或者异步操作——如果有，需要确保已经在配置类上添加@EnableScheduling，并且配置了对应的调度器。

验证修改 #

完成上述配置后，重启应用，尝试建立WebSocket连接并发送消息，应该就能解决这个异常了。

内容的提问来源于stack exchange，提问作者nagy.zsolt.hun