From d6910a568c44a33c576b330a53bb7aee6dbfa632 Mon Sep 17 00:00:00 2001
From: shenyijian <1178253012@qq.com>
Date: Wed, 15 Nov 2023 10:26:56 +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