From 27ad176af126076e49a757e7d852e8d5ca670b2d Mon Sep 17 00:00:00 2001
From: DJIsean <102007705+DJIsean@users.noreply.github.com>
Date: Thu, 11 Aug 2022 15:13:09 +0800
Subject: [PATCH] Update HmsJsonUtil.java

---
 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