Просмотр исходного кода

fix:【Infra 基础设置】支持 yudao.websocket.enable 手动关闭

YunaiV месяцев назад: 11
Родитель
Сommit
d09e4849ba

+ 3 - 3
yudao-module-infra/src/main/java/cn/iocoder/yudao/module/infra/api/websocket/WebSocketSenderApiImpl.java

@@ -1,10 +1,9 @@
 package cn.iocoder.yudao.module.infra.api.websocket;
 
 import cn.iocoder.yudao.framework.websocket.core.sender.WebSocketMessageSender;
+import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Component;
 
-import jakarta.annotation.Resource;
-
 /**
  * WebSocket 发送器的 API 实现类
  *
@@ -13,7 +12,8 @@ import jakarta.annotation.Resource;
 @Component
 public class WebSocketSenderApiImpl implements WebSocketSenderApi {
 
-    @Resource
+    @SuppressWarnings("SpringJavaAutowiredFieldsWarningInspection")
+    @Autowired(required = false) // 由于 yudao.websocket.enable 配置项,可以关闭 WebSocket 的功能,所以这里只能不强制注入
     private WebSocketMessageSender webSocketMessageSender;
 
     @Override

+ 3 - 3
yudao-module-infra/src/main/java/cn/iocoder/yudao/module/infra/websocket/DemoWebSocketMessageListener.java

@@ -6,11 +6,10 @@ import cn.iocoder.yudao.framework.websocket.core.sender.WebSocketMessageSender;
 import cn.iocoder.yudao.framework.websocket.core.util.WebSocketFrameworkUtils;
 import cn.iocoder.yudao.module.infra.websocket.message.DemoReceiveMessage;
 import cn.iocoder.yudao.module.infra.websocket.message.DemoSendMessage;
+import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Component;
 import org.springframework.web.socket.WebSocketSession;
 
-import jakarta.annotation.Resource;
-
 /**
  * WebSocket 示例:单发消息
  *
@@ -19,7 +18,8 @@ import jakarta.annotation.Resource;
 @Component
 public class DemoWebSocketMessageListener implements WebSocketMessageListener<DemoSendMessage> {
 
-    @Resource
+    @SuppressWarnings("SpringJavaAutowiredFieldsWarningInspection")
+    @Autowired(required = false) // 由于 yudao.websocket.enable 配置项,可以关闭 WebSocket 的功能,所以这里只能不强制注入
     private WebSocketMessageSender webSocketMessageSender;
 
     @Override