From e54bc01ed3a571abc9de4dd4c32893e8741bb0f0 Mon Sep 17 00:00:00 2001
From: jpy <812110275@qq.com>
Date: Sat, 27 May 2023 15:13:28 +0800
Subject: [PATCH] test

---
 screen-api/src/main/java/com/moral/api/config/websocket/WebSocketConfig.java |   11 +++++++++++
 1 files changed, 11 insertions(+), 0 deletions(-)

diff --git a/screen-api/src/main/java/com/moral/api/config/websocket/WebSocketConfig.java b/screen-api/src/main/java/com/moral/api/config/websocket/WebSocketConfig.java
index c4f6e0a..9189321 100644
--- a/screen-api/src/main/java/com/moral/api/config/websocket/WebSocketConfig.java
+++ b/screen-api/src/main/java/com/moral/api/config/websocket/WebSocketConfig.java
@@ -1,7 +1,11 @@
 package com.moral.api.config.websocket;
 
+import com.moral.api.websocket.CruiserWebSocketServer;
+import com.moral.api.websocket.SingleDeviceServer;
+import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.context.annotation.Bean;
 import org.springframework.context.annotation.Configuration;
+import org.springframework.data.redis.core.RedisTemplate;
 import org.springframework.web.socket.server.standard.ServerEndpointExporter;
 
 /**
@@ -18,4 +22,11 @@
     public ServerEndpointExporter serverEndpointExporter(){
         return new ServerEndpointExporter();
     }
+
+    @Autowired
+    public void setMessageService(RedisTemplate redisTemplate){
+        SingleDeviceServer.redisTemplate = redisTemplate;
+        CruiserWebSocketServer.redisTemplate = redisTemplate;
+    }
+
 }

--
Gitblit v1.8.0