【转载】spring springboot websocket 不能注入( @Autowired ) service bean 报 null 错误

最近项目要实现一个需求，springboot使用websocket定时给前端返回数据，我使用@Autiwired注解注入service服务类,但是运行起来service类为null,在网上查了一下，发现在@ServerEndPoint中是无法使用@Autowired注入Bean的。原理大致如下：

其实在项目启动的时候，这个MyWebSocket已经被注入了这个MyService的Bean，但是WebSocket的特点就是每建立一个连接，会生成一个新的MyWebSocket对象，网上有人分析过源码，也可以理解成new了一个WebSocket，这样当然是不能获得自动注入的对象了。
后来在网上看到有网友按照如下的代码，解决了此问题，

大概就是获取到Spring容器，然后去进行手动注入。

因为新生成的MyWebSocket对象根本不存在与Spring容器中，所以传统的ContextLoader.getCurrentWebApplicationContext()自然也就无法使用。

我用到了ApplicationContextAware来在项目加载时就获得项目的Spring容器。代码如下

@Component
@Lazy(false)
public class MyApplicationContextAware implements ApplicationContextAware {
 
    private static ApplicationContext APPLICATION_CONTEXT;
 
    @Override
    public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
        APPLICATION_CONTEXT = applicationContext;
    }
 
    public static ApplicationContext getApplicationContext() {
        return APPLICATION_CONTEXT;
    }
}
这样我们就可以在任何类中去拿到这个类的ApplicationContext

通过这个ApplicationContext我们就可以在新生成一个WebSocket对象时来手动注入Service层了

@ServerEndpoint(value = "/websocket")
@Component
public class MyWebSocket {
 
    private static ConcurrentHashMap<String, MyWebSocket> map = new ConcurrentHashMap<>();
 
    private Session session;
 
    private MyService myServiceImpl = (MyService) MyApplicationContextAware.getApplicationContext().getBean("myServiceImpl");
但是我按照此代码，springboot启动报错了，后来又看到一个网友有个更简单的方式能解决此问题，代码如下：

解决方法：将要注入的 service 改成 static，就不会为null了。
参考代码：

@Controller
@ServerEndpoint(value="/chatSocket")
public class ChatSocket {
    //  这里使用静态，让 service 属于类
    private static ChatService chatService;

    // 注入的时候，给类的 service 注入
    @Autowired
    public void setChatService(ChatService chatService) {
        ChatSocket.chatService = chatService;
    }
}
1
2
3
4
5
6
7
8
9
10
11
12
本质原因：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。

参考：http://www.fengyunxiao.cn
参考：https://www.fengyunxiao.cn/blogs
--------------------- 
作者：亦寒2017 
来源：CSDN 
原文：https://blog.csdn.net/m0_37202351/article/details/86255132 
版权声明：本文为博主原创文章，转载请附上博文链接！