快捷導(dǎo)航

WebSocket 中使用 @Autowired 注入對應(yīng)為null的解決方案

更新時(shí)間：2024年10月22日 09:30:59 作者：_江嶼_

SpringBoot集成WebSocket時(shí),會(huì)遇到service對象為null的情況,原因是Spring默認(rèn)為單例模式與WebSocket的多對象模式相沖突,當(dāng)客戶端與服務(wù)器端建立連接時(shí),會(huì)創(chuàng)建新的WebSocket對象,本文給大家介紹WebSocket 中使用 @Autowired 注入對應(yīng)為null的問題,感興趣的朋友一起看看吧

代碼示例：

@Component
public class WebSocketHandlerMessage implements WebSocketHandler {
    @Autowired
    private BarrageMessageService barrageMessageService;
｝

結(jié)果顯示：

SpringBoot項(xiàng)目集成 webSocket,當(dāng)客戶端與服務(wù)器端建立連接的時(shí)候，發(fā)現(xiàn) barrageMessageService 對象并未注入而是為 null。

產(chǎn)生原因：spring管理的都是單例（singleton），和 websocket （多對象）相沖突。
詳細(xì)解釋：項(xiàng)目啟動(dòng)時(shí)初始化，會(huì)初始化 websocket （非用戶連接的），spring 同時(shí)會(huì)為其注入 service，該對象的 service 不是 null，被成功注入。但是，由于 spring 默認(rèn)管理的是單例，所以只會(huì)注入一次 service。當(dāng)客戶端與服務(wù)器端進(jìn)行連接時(shí)，服務(wù)器端又會(huì)創(chuàng)建一個(gè)新的 websocket 對象，這時(shí)問題出現(xiàn)了：spring 管理的都是單例，不會(huì)給第二個(gè) websocket 對象注入 service，所以導(dǎo)致只要是用戶連接創(chuàng)建的 websocket 對象，都不能再注入了。

像 controller 里面有 service， service 里面有 dao。因?yàn)?controller，service ，dao 都有是單例，所以注入時(shí)不會(huì)報(bào) null。但是 websocket 不是單例，所以使用spring注入一次后，后面的對象就不會(huì)再注入了，會(huì)報(bào)NullException。

解決方法：

方案一：使用靜態(tài)，讓 service 屬于類，然后給類的 service 注入。

@Component
public class WebSocketHandlerMessage implements WebSocketHandler {
    /**
     *  這里使用靜態(tài)，讓 service 屬于類
     */
    private static BarrageMessageService barrageMessageService;
    /**
     *  注入的時(shí)候，給類的 service 注入
     */
    @Autowired
    public void setBarrageMessageService(BarrageMessageService barrageMessageService) {
        WebSocketHandlerMessage.barrageMessageService = barrageMessageService;
    }
｝

方案二：在新建立連接的時(shí)候重新從Spring 容器中獲取 BarrageMessageService 對象，這樣就可以正常使用了。

@Component
public class WebSocketHandlerMessage implements WebSocketHandler {
    /**
     * 獲取 barrageMessageService 對象方法
     *
     * @return
     */
    public BarrageMessageService getMessageService() {
        return SpringContext.getBean(BarrageMessageService.class);
    }
    /**
     * 獲取 stringRedisTemplate 對象方法
     *
     * @return
     */
    public StringRedisTemplate getStringRedisTemplate() {
        return SpringContext.getBean(StringRedisTemplate.class);
    }
｝

SpringContext 工具類方法：

/**
 * @Description: SpringContext  獲取 Spring 上下文信息
 * @Author: mingtian
 * @CreateDate: 2020/6/8 14:59
 * @Version: 1.0
 */
@Component
public class SpringContext implements ApplicationContextAware {
    /**
     * 打印日志
     */
    private Logger logger = LoggerFactory.getLogger(getClass());
    /**
     * 獲取上下文對象
     */
    private static ApplicationContext applicationContext;
    @Override
    public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
        SpringContext.applicationContext = applicationContext;
        logger.info("set applicationContext");
    }
    /**
     * 獲取 applicationContext
     *
     * @return
     */
    public static ApplicationContext getApplicationContext() {
        return applicationContext;
    }
    /**
     * 通過 name 獲取 bean 對象
     *
     * @param name
     * @return
     */
    public static Object getBean(String name) {
        return getApplicationContext().getBean(name);
    }
    /**
     * 通過 class 獲取 bean 對象
     *
     * @param clazz
     * @param <T>
     * @return
     */
    public static <T> T getBean(Class<T> clazz) {
        return getApplicationContext().getBean(clazz);
    }
    /**
     * 通過 name，clazz  獲取指定的 bean 對象
     *
     * @param name
     * @param clazz
     * @param <T>
     * @return
     */
    public static <T> T getBean(String name, Class<T> clazz) {
        return getApplicationContext().getBean(name, clazz);
    }
}

以上二種方案都可以解決，webSocket 中 service 注入為 null 的問題。

到此這篇關(guān)于WebSocket 中使用 @Autowired 注入對應(yīng)為null的文章就介紹到這了,更多相關(guān)WebSocket @Autowired 注入為null內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: