SpringBoot整合WebSocket时调用service和mapper的方法

场景

SpringBoot+Vue整合WebSocket实现前后端消息推送：

https://blog.csdn.net/BADAO_LIUMANG_QIZHI/article/details/114392573

SpringCloud(若依微服务版为例)集成WebSocket实现前后端的消息推送：

https://blog.csdn.net/BADAO_LIUMANG_QIZHI/article/details/114480731

在上面集成websocket的时候，如果需要在websocket建立连接的回调方法中去调用其他业务的service和mapper的方法时，却注入失败，service为空。

原因：

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

注：

博客：
https://blog.csdn.net/badao_liumang_qizhi
关注公众号
霸道的程序猿
获取编程相关电子书、教程推送与免费下载。

实现

首先在websocket的服务类中添加applicationContext属性以及set方法

    private static ApplicationContext applicationContext;

    public static void setApplicationContext(ApplicationContext context) {
        applicationContext = context;
    }

添加位置

然后在项目的启动类中

        ConfigurableApplicationContext applicationContext =   SpringApplication.run(RuoYiFzysControlApplication.class, args);
        WebSocketService.setApplicationContext(applicationContext);

在项目启动后对其进行赋值，这里的WebSocketService对应上面自己的websocket服务类

然后就可以通过

applicationContext.getBean(IBusAreaControlService.class);

来调用其他service