webSocket无法注入bean问题解决方案

ゝ一世哀愁。 2023-09-29 11:51 4阅读 0赞

websocket服务端往往需要和服务层打交道,因此需要将服务层的一些bean注入到websocket实现类中使用，但是呢，websocket实现类虽然顶部加上了@Component注解，依然无法通过@Resource和@Autowire注入spring容器管理下的bean。后来就想用ApplicationContext获取spring容器管理下的bean。但是无法获取ApplicationContext的实例，因为该实例也是在spring下管理的，有点小奔溃。找到解决办法，那就是在初始化ApplicationContext实例的时候将该引用保存到websocket类里。如下：

@Component
    @ServerEndpoint(value = "/messageSocket/{userId}")
    public class MessageWebSocket {
            /**
         * 此处是解决无法注入的关键
         */
        private static ApplicationContext applicationContext;
    
    
        public static void setApplicationContext(ApplicationContext applicationContext) {
            MessageWebSocket.applicationContext = applicationContext;
        }
    
    　　@OnOpen
        public void onOpen(Session session, @PathParam("userId") Integer userId) {
        }
    
    　　@OnClose
        public void onClose() {
    　　}
    
    　　@OnMessage
        public void onMessage(String messageStr, Session session, @PathParam("userId") Integer userId) throws IOException {
            //applicationContext使用
            ThreadPoolTaskExecutor threadPoolTaskExecutor = (ThreadPoolTaskExecutor)applicationContext.getBean("defaultThreadPool");
    　　}
    
    　　@OnError
        public void onError(Session session, Throwable error) {
    　　}
    }

然后在初始化ApplicationContext时（在springboot启动类中）对该类的MessageWebSocket进行赋值。

public class WebApplication {
        public static void main(String[] args) {
            SpringApplication springApplication = new SpringApplication(WebApplication.class);
            ConfigurableApplicationContext configurableApplicationContext = springApplication.run(args);
            //解决WebSocket不能注入的问题
            MessageWebSocket.setApplicationContext(configurableApplicationContext);
        }
    }

思考：

为什么websocket注入不进bean呢（已经将其加入了spring管理了，而且其他类是能够注入websocket的实现bean）？

网上也没有这个问题的具体解释，我的感觉是websocket连接是需要创建多个线程的，其与spring自动注入有某种冲突，因此@Resource和@Autowire无法注入bean的引用。

参考：[websocket无法注入bean问题解决方案 - 会飞的joy - 博客园][websocket_bean_ - _joy -]

网友解答（推荐）：

本质原因：spring管理的都是单例（singleton），和websocket（多对象）想冲突。

详细解释：项目启动时初始化，会初始化websocket（非用户连接的），spring同时会为其注入service，该对象的service不是null，被成功的注入。但是，由于spring默认管理的是单例，所以只会注入一次service。当新用户进入聊天时，系统又会创建一个新的websocket对象，这是矛盾出现了：spring管理的都是单例，不会给第二个websocket对象注入service，所以导致只要是用户连接创建的websocket对象，都不能再注入了。

像controller里面有service，service里面有dao。因为controller，service，dao都是单例，所以注入时不会报null。但是websocket不是单例，所以使用spring注入一次后，后面的对象就不会再注入了，会报null。

[websocket_bean_ - _joy -]: https://www.cnblogs.com/zxb1996/p/websocket_question1.html