From 559a1a4f558a52349acd04e04a1ab9f65fa3221f Mon Sep 17 00:00:00 2001
From: wangliwen <wangliwen@hisense.com>
Date: Tue, 26 Jul 2022 19:45:27 +0800
Subject: [PATCH] ...

---
 .../java/io/renren/websocket/WebSocketServer.java    |  4 ++++
 .../java/io/renren/websocket/task/HeartRateTask.java | 12 +++++++++---
 2 files changed, 13 insertions(+), 3 deletions(-)

diff --git a/renren-admin/src/main/java/io/renren/websocket/WebSocketServer.java b/renren-admin/src/main/java/io/renren/websocket/WebSocketServer.java
index 09bd6b49..39148c0d 100644
--- a/renren-admin/src/main/java/io/renren/websocket/WebSocketServer.java
+++ b/renren-admin/src/main/java/io/renren/websocket/WebSocketServer.java
@@ -101,4 +101,8 @@ public class WebSocketServer {
             log.error("send message error，" + e.getMessage(), e);
         }
     }
+
+    public static Map<String, WebSocketData> getServers() {
+        return servers;
+    }
 }
\ No newline at end of file
diff --git a/renren-admin/src/main/java/io/renren/websocket/task/HeartRateTask.java b/renren-admin/src/main/java/io/renren/websocket/task/HeartRateTask.java
index 58e95139..683bc84f 100644
--- a/renren-admin/src/main/java/io/renren/websocket/task/HeartRateTask.java
+++ b/renren-admin/src/main/java/io/renren/websocket/task/HeartRateTask.java
@@ -28,8 +28,14 @@ public class HeartRateTask implements ITask {
      */
     @Override
     public void run(String params) {
-        logger.info("websocket心跳");
-        MessageData<String> message = new MessageData<String>().msg("heartRate:" + LocalDateTime.now());
-        webSocketServer.sendMessageAll(message);
+        try {
+            if (webSocketServer.getServers().isEmpty()) {
+                return;
+            }
+            MessageData<String> message = new MessageData<String>().msg("heartRate:" + LocalDateTime.now());
+            webSocketServer.sendMessageAll(message);
+        } catch (Exception exception) {
+            logger.error("websocket客户端心跳失败", exception);
+        }
     }
 }