SpringBoot - 整合WebSocket时@ServerEndpoint修饰的类属性注入为null问题
- 前言
- 一. 问题复现
- 1.1 原因分析
- 二. 问题解决
前言
最近在做一个直播弹幕系统,前期准备先用WebSocket
来试试水。我们都知道,使用WebSocket
只需要给对应的类加上注解@ServerEndpoint
即可。这个类中我还引用了一个MQ
的服务类,结果调用的时候发现NPE
。因此写下本篇文章做个记录。
一. 问题复现
1.我们建立一个WebSocket
链接,后端感知到了链接建立(@OnOpen
注解)之后,通过originMessageSender
发送一条消息出去。
相关代码如下:
但是代码跑到这里却发现originMessageSender
这个对象为null
,那么这段代码执行的时候自然而然的就是NPE
异常了。
如图:
为什么会发生这种情况呢?我们来分析下。
1.1 原因分析
首先我们来从@Autowired
和Bean
的属性注入角度来看。
- 容器启动的时候,会将每个单例的
Bean
加载到容器当中。 - 单例
Bean
的加载有三个步骤。实例化、依赖注入、初始化。而@Autowired
的作用就体现于依赖注入当中。 - 我们
BulletScreenService
类中,通过@Autowired
对OriginMessageSender
进行自动装配。其实他是成功的,假设我们这时候装配的对象是sender1
。
接着我们再来从WebSocket
的角度来看:
- 我们每建立一个
WebSocket
链接,都会产生一个新的对象,也就是被@ServerEndpoint
修饰的对象。 - 也就是说
WebSocket
是一个多例的对象而非单例。 也就是说,当建立链接之后创建的BulletScreenService
实例对象和容器启动的时候,加载到容器当中的BulletScreenServiceBean
已经不是一个东西了。 - 那么为什么通过
@Autowired
注入的对象是null
呢?因为@Autowired
注解注入对象是在启动的时候就把对象注入,而不是在使用A对象时才把A需要的B对象注入到A中的。
因此我们每建立一个WebSocket
链接,按照上面的写法,我们是无法注入OriginMessageSender
的。
二. 问题解决
既然给对象注入属性不可行,那么我们换位思考,给类属性注入不就好了嘛?修改如下。
原本代码:
@Autowired
private OriginMessageSender originMessageSender;
修改后代码:
private static OriginMessageSender originMessageSender;
@Autowired
private void setOriginMessageSender(OriginMessageSender originMessageSender) {
BulletScreenService.originMessageSender = originMessageSender;
}
那么在容器启动的时候,就会加载这段代码,然后将originMessageSender
实例赋值给静态变量。(我们知道静态变量属于类,而不是实例对象)
那么同样的,当我们客户端建立起和服务端的WebSocket
链接,发送消息的时候:这里就不再是Null
了。