WebSocket获取bean

### 如何在 WebSocket 中获取 Spring Bean

为了在 WebSocket 组件中使用 Spring 容器管理的 Bean，可以采用多种方法来确保这些组件能够访问到所需的资源和服务。

#### 方法一：通过 `@Autowired` 注解自动装配

当定义了一个 WebSocket 处理类时，可以直接利用 Spring 的依赖注入机制。对于基于注解配置的应用程序来说，这通常是最简单的方法之一：

@Component
public class MyWebSocketHandler extends TextWebSocketHandler {

    private final TestService testService;

    @Autowired
    public MyWebSocketHandler(TestService testService) {
        this.testService = testService;
    }

    @Override
    protected void handleTextMessage(WebSocketSession session, TextMessage message) throws Exception {
        // 使用已注入的服务对象执行业务逻辑
        String result = testService.someMethod();
        System.out.println(result);
        
        super.handleTextMessage(session, message);
    }
}

此方式下，只要 `TestService` 是由 Spring 管理的一个 Bean，则它会按照正常的 IoC 原则被正确地注入给 `MyWebSocketHandler` 实例[^2]。

#### 方法二：静态持有 Application Context 来手动查找 Bean

另一种情况是在某些特定场景下可能无法直接应用上述依赖注入的方式（比如自定义 Endpoint），此时可以通过保存 ApplicationContext 并从中检索所需 Bean:

首先创建一个工具类用于存储上下文引用：

public class AppContextUtil implements ApplicationContextAware {

    private static ApplicationContext context;

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

    public static Object getBean(String beanName){
        return context.getBean(beanName);
    }
    
    public static <T> T getBean(Class<T> clazz){
        return context.getBean(clazz);
    }
}

接着可以在任何地方调用该工具类来获得指定名称或类型的 Bean 对象：

// 在 Web Socket 连接建立后的处理函数内
@Override
public void afterConnectionEstablished(WebSocketSession session) throws Exception {
    TestService serviceInstance = (TestService)AppContextUtil.getBean("testService");
    // 或者更简洁的形式
    TestService anotherWayToGetService = AppContextUtil.getBean(TestService.class);

    // 接下来就可以正常使用这个服务实例了...
}

这种方法虽然绕过了标准的 DI 流程，但在灵活性方面提供了额外的选择[^3]。

#### 关于作用域特别说明

值得注意的是，如果希望某个 Service 类型的对象在整个 HTTP 会话期间保持一致的行为模式，那么应该将其声明为具有 `session` 范围内的单件形式。这样做的好处是可以让同一用户的多次请求之间共享相同的状态数据而不会相互干扰[^4]:

@Service
@SessionScope
public class TestService {
    // ...
}