导出优化、告警日志方法增加

ruoyi-rocketmq/src/main/java/com/ruoyi/rocketmq/handler/MessageHandler.java

@@ -87,6 +87,8 @@ public class MessageHandler {
     private IRmDeploymentPolicyService rmDeploymentPolicyService;
     @Autowired
     private IInitialBandwidthTrafficTempService initialBandwidthTrafficTempService;
+    @Autowired
+    private IRmAlarmLogService rmAlarmLogService;
 
 
     /**
@@ -1079,6 +1081,37 @@ public class MessageHandler {
             log.error("插入心跳日志失败", e);
         }
     }
+    private void insertAlarmRecords(String clientId){
+        RmAlarmLog rmAlarmLog = new RmAlarmLog();
+        // 查询管理网公网ip
+        RmNetworkInterface rmNetworkInterface = new RmNetworkInterface();
+        rmNetworkInterface.setClientId(clientId);
+        rmNetworkInterface.setNewFlag(1);
+        List<RmNetworkInterface> interfaceList = rmNetworkInterfaceService.selectRmNetworkInterfaceList(rmNetworkInterface);
+        if(interfaceList != null && !interfaceList.isEmpty()){
+            interfaceList.stream()
+                    .filter(info -> "2".equals(info.getBindIp()) || "3".equals(info.getBindIp()))
+                    .findFirst()
+                    .ifPresent(networkInterface -> {
+                        rmAlarmLog.setMgmPublicIp(networkInterface.getPublicIp());
+                    });
+        }
+        rmAlarmLog.setClientId(clientId);
+        rmAlarmLog.setAlarmContent(clientId + "下线");
+        rmAlarmLog.setAlarmTime(DateUtils.getNowDate());
+        rmAlarmLog.setAlarmType("1");
+        // 查询此告警是否存在
+        RmAlarmLog query = new RmAlarmLog();
+        query.setClientId(clientId);
+        query.setAlarmType("1");
+        List<RmAlarmLog> alarmLog = rmAlarmLogService.selectRmAlarmLogList(query);
+        if(alarmLog != null && !alarmLog.isEmpty()){
+            rmAlarmLogService.updateRmAlarmLog(rmAlarmLog);
+        }else{
+            rmAlarmLogService.insertRmAlarmLog(rmAlarmLog);
+        }
+
+    }
 
     /**
      * 应答信息