bihuisong 1 year ago
parent
commit
400e81a069

+ 6 - 1
ruoyi-system/src/main/java/com/ruoyi/system/service/impl/UnableToAccessWebsiteListTodayImpl.java

@@ -3,6 +3,7 @@ package com.ruoyi.system.service.impl;
 import com.alibaba.fastjson.JSON;
 import com.ruoyi.system.domain.UnableToAccessWebsiteListToday;
 import com.ruoyi.system.domain.WebSiteTenDaysHits;
+import com.ruoyi.system.domain.WebsiteAvailabilityMonitor;
 import com.ruoyi.system.mapper.UnableToAccessWebsiteListTodayMapper;
 import com.ruoyi.system.mapper.WebSiteTenDaysHitsMapper;
 import com.ruoyi.system.service.IUnableToAccessWebsiteListTodayService;
@@ -12,7 +13,9 @@ import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 import org.springframework.util.CollectionUtils;
 
+import java.util.HashMap;
 import java.util.List;
+import java.util.Map;
 
 
 @Service
@@ -30,10 +33,12 @@ public class UnableToAccessWebsiteListTodayImpl implements IUnableToAccessWebsit
 
     @Override
     public void insertBatch(List<UnableToAccessWebsiteListToday> list) {
+        Map<String, List<UnableToAccessWebsiteListToday>> returnMap = new HashMap<>();
         if (!CollectionUtils.isEmpty(list)) {
             unableToAccessWebsiteListTodayMapper.deleteUnableToAccessWebsiteListTodayAll();
         }
         unableToAccessWebsiteListTodayMapper.insertBatch(list);
-        WebSocketUsers.sendMessageToUsersByText(JSON.toJSONString(list));
+        returnMap.put("unableToAccessWebsiteListToday",list);
+        WebSocketUsers.sendMessageToUsersByText(JSON.toJSONString(returnMap));
     }
 }

+ 5 - 1
ruoyi-system/src/main/java/com/ruoyi/system/service/impl/WebsiteAvailabilityMonitorServiceImpl.java

@@ -11,7 +11,9 @@ import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 import org.springframework.util.CollectionUtils;
 
+import java.util.HashMap;
 import java.util.List;
+import java.util.Map;
 
 import static com.ruoyi.common.core.domain.AjaxResult.success;
 
@@ -32,11 +34,13 @@ public class WebsiteAvailabilityMonitorServiceImpl implements IWebsiteAvailabili
      */
     @Override
     public void insertBatch(List<WebsiteAvailabilityMonitor> list) {
+        Map<String, List<WebsiteAvailabilityMonitor>> returnMap = new HashMap<>();
         if (!CollectionUtils.isEmpty(list)) {
             websiteAvailabilityMonitorMapper.deleteWebsiteAvailabilityMonitorAll();
         }
         websiteAvailabilityMonitorMapper.insertBatch(list);
-        WebSocketUsers.sendMessageToUsersByText(JSON.toJSONString(success(list)));
+        returnMap.put("websiteAvailabilityMonitor",list);
+        WebSocketUsers.sendMessageToUsersByText(JSON.toJSONString(success(returnMap)));
     }
 
 }