From 9b2eedb85d53ca32610c32c6e50b5230ab3b16cf Mon Sep 17 00:00:00 2001
From: sean.zhou <sean.zhou@dji.com>
Date: Fri, 22 Jul 2022 20:16:03 +0800
Subject: [PATCH] V1.1.0 for dock
---
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