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

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的引用。

参考:https://www.cnblogs.com/zxb1996/p/websocket_question1.html

 

网友解答(推荐):

  本质原因: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。

posted @ 2020-12-21 23:40  10114  阅读(834)  评论(0编辑  收藏  举报