webSocket無法注入bean問題解決方案

websocket服務端往往需要和服務層打交道,因此需要將服務層的一些bean注入到websocket實現類中使用，但是呢，websocket實現類雖然頂部加上了@Component注解，依然無法通過@Resource和@Autowire注入spring容器管理下的bean。后來就想用ApplicationContext獲取spring容器管理下的bean。但是無法獲取ApplicationContext的實例，因為該實例也是在spring下管理的，有點小奔潰。找到解決辦法，那就是在初始化ApplicationContext實例的時候將該引用保存到websocket類里。如下

@Component
@ServerEndpoint(value = "/messageSocket/{userId}")
public class MessageWebSocket {
        /**
     * 此處是解決無法注入的關鍵
     */
    private static ApplicationContext applicationContext;


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

　　@OnOpen
    public void onOpen(Session session, @PathParam("userId") Integer userId) {
    }

　　@OnClose
    public void onClose() {
　　}

　　@OnMessage
    public void onMessage(String messageStr, Session session, @PathParam("userId") Integer userId) throws IOException {
        //applicationContext使用
        ThreadPoolTaskExecutor threadPoolTaskExecutor = (ThreadPoolTaskExecutor)applicationContext.getBean("defaultThreadPool");
　　}

　　@OnError
    public void onError(Session session, Throwable error) {
　　}
}

然后在初始化ApplicationContext時（在springboot啟動類中）對該類的MessageWebSocket進行賦值

public class WebApplication {
    public static void main(String[] args) {
        SpringApplication springApplication = new SpringApplication(WebApplication.class);
        ConfigurableApplicationContext configurableApplicationContext = springApplication.run(args);
        //解決WebSocket不能注入的問題
        MessageWebSocket.setApplicationContext(configurableApplicationContext);
    }
}

思考：

　　為什么websocket注入不進bean呢（已經將其加入了spring管理了，而且其他類是能夠注入websocket的實現bean）？

　　　　網上也沒有這個問題的具體解釋，我的感覺是websocket連接是需要創建多個線程的，其與spring自動注入有某種沖突，因此@Resource和@Autowire無法注入bean的引用。

參考：https://www.cnblogs.com/zxb1996/p/websocket_question1.html

 

網友解答（推薦）：

　　本質原因：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。