From d2108ee735894d73bf655bda1651ecaefbc4c09d Mon Sep 17 00:00:00 2001
From: rain <167982779@qq.com>
Date: Sat, 23 Mar 2024 09:35:02 +0800
Subject: [PATCH] 图斑

---
 src/main/java/com/dji/sample/component/websocket/config/WebSocketDefaultFactory.java |    7 ++++++-
 1 files changed, 6 insertions(+), 1 deletions(-)

diff --git a/src/main/java/com/dji/sample/component/websocket/config/WebSocketDefaultFactory.java b/src/main/java/com/dji/sample/component/websocket/config/WebSocketDefaultFactory.java
index 4fdd774..f6ad8fb 100644
--- a/src/main/java/com/dji/sample/component/websocket/config/WebSocketDefaultFactory.java
+++ b/src/main/java/com/dji/sample/component/websocket/config/WebSocketDefaultFactory.java
@@ -1,5 +1,7 @@
 package com.dji.sample.component.websocket.config;
 
+import com.dji.sample.component.websocket.service.IWebSocketManageService;
+import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Component;
 import org.springframework.web.socket.WebSocketHandler;
 import org.springframework.web.socket.handler.WebSocketHandlerDecoratorFactory;
@@ -13,8 +15,11 @@
 @Component
 public class WebSocketDefaultFactory implements WebSocketHandlerDecoratorFactory {
 
+    @Autowired
+    private IWebSocketManageService webSocketManageService;
+
     @Override
     public WebSocketHandler decorate(WebSocketHandler handler) {
-        return new WebSocketDefaultHandler(handler);
+        return new WebSocketDefaultHandler(handler, webSocketManageService);
     }
 }
\ No newline at end of file

--
Gitblit v1.9.3