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。