Spring WebSocket：手动实例化Handler的优势及Bean注入方法

一、手动new Handler的优势 #

• 规避单例并发风险：Spring默认Bean是单例模式，若将WebSocketHandler注册为单例Bean，所有WebSocket连接会共享同一个实例。如果Handler内部维护了连接相关的状态（比如用户会话信息），很容易引发多线程并发问题。手动实例化可以按需创建Handler实例，比如为每个连接分配独立实例，确保状态隔离。
• 实例创建逻辑更灵活：可以直接通过构造函数传入初始化参数，无需依赖Spring的@Value或@Autowired注入，初始化逻辑更直观可控，适合需要动态配置的场景。
• 降低容器耦合度：Handler无需添加@Component等Spring注解，不需要被Spring容器管理，代码独立性更强，甚至可以在非Spring环境中复用。
• 明确控制作用域：无需额外配置@Scope注解，就能直接控制Handler的实例数量和生命周期，避免因作用域配置错误导致的问题。

二、手动实例化Handler时注入Bean的方案 #

如果需要给手动new出来的Handler注入Spring管理的Bean，有两种常用的简便方案：

方案1：构造函数传递Bean #

在配置类中先注入所需Bean，再通过Handler的构造函数传递进去：

@Configuration
@EnableWebSocket
public class WebSocketConfig implements WebSocketConfigurer {
    // 注入需要的业务Bean
    @Autowired
    private MyService myService;

    @Override
    public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
        // 手动实例化Handler时传入Bean
        registry.addHandler(new MyHandler(myService), "/myHandler")
            .addInterceptors(new HttpSessionHandshakeInterceptor());
    }
}

// 对应的Handler实现类
public class MyHandler extends TextWebSocketHandler {
    private final MyService myService;

    // 通过构造函数接收注入的Bean
    public MyHandler(MyService myService) {
        this.myService = myService;
    }

    @Override
    protected void handleTextMessage(WebSocketSession session, TextMessage message) throws Exception {
        // 直接使用myService处理业务逻辑
        String result = myService.processMessage(message.getPayload());
        session.sendMessage(new TextMessage(result));
    }
}

方案2：通过ApplicationContext获取Bean #

让Handler实现ApplicationContextAware接口，获取Spring容器后手动获取Bean：

public class MyHandler extends TextWebSocketHandler implements ApplicationContextAware {
    private static ApplicationContext appContext;
    private MyService myService;

    @Override
    public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
        appContext = applicationContext;
    }

    @Override
    public void afterConnectionEstablished(WebSocketSession session) throws Exception {
        // 在连接建立时获取Bean（按需获取即可）
        myService = appContext.getBean(MyService.class);
    }

    @Override
    protected void handleTextMessage(WebSocketSession session, TextMessage message) throws Exception {
        // 使用myService处理消息
        String result = myService.processMessage(message.getPayload());
        session.sendMessage(new TextMessage(result));
    }
}

注：Spring 4.2.2版本对WebSocket的构造函数注入支持更完善，相比4.2.1版本减少了配置上的限制，推荐使用构造函数传递Bean的方案，代码更简洁且符合依赖注入的最佳实践。

内容的提问来源于stack exchange，提问作者z.x.99