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

谁借莪１个温暖的怀抱￠ 2022-11-09 12:49 312阅读 0赞

# 场景 #

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

[https://blog.csdn.net/BADAO\_LIUMANG\_QIZHI/article/details/114392573][https_blog.csdn.net_BADAO_LIUMANG_QIZHI_article_details_114392573]

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

[https://blog.csdn.net/BADAO\_LIUMANG\_QIZHI/article/details/114480731][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][https_blog.csdn.net_badao_liumang_qizhi]  
关注公众号  
霸道的程序猿  
获取编程相关电子书、教程推送与免费下载。

# 实现 #

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

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

添加位置

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L0JBREFPX0xJVU1BTkdfUUlaSEk_size_16_color_FFFFFF_t_70][]

然后在项目的启动类中

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

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

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L0JBREFPX0xJVU1BTkdfUUlaSEk_size_16_color_FFFFFF_t_70 1][]

然后就可以通过

applicationContext.getBean(IBusAreaControlService.class);

来调用其他service

[https_blog.csdn.net_BADAO_LIUMANG_QIZHI_article_details_114392573]: https://blog.csdn.net/BADAO_LIUMANG_QIZHI/article/details/114392573
[https_blog.csdn.net_BADAO_LIUMANG_QIZHI_article_details_114480731]: https://blog.csdn.net/BADAO_LIUMANG_QIZHI/article/details/114480731
[https_blog.csdn.net_badao_liumang_qizhi]: https://blog.csdn.net/badao_liumang_qizhi
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L0JBREFPX0xJVU1BTkdfUUlaSEk_size_16_color_FFFFFF_t_70]: /images/20221023/c6f8821c474a49d7bd7f94adf7e6822b.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L0JBREFPX0xJVU1BTkdfUUlaSEk_size_16_color_FFFFFF_t_70 1]: /images/20221023/a16924cbcfcf404287bbb90914f94385.png