资源模板以及资源策略模块方法和接口。

2025-09-11 17:57:51 +08:00
parent df7ed79fcb
commit 3dc8b2c2c6
59 changed files with 3523 additions and 184 deletions
@@ -0,0 +1,104 @@
+package com.ruoyi.rocketmq.utils;
+
+import com.ruoyi.common.core.constant.SecurityConstants;
+import com.ruoyi.common.core.domain.R;
+import com.ruoyi.rocketmq.domain.RmTemplateLinux;
+import com.ruoyi.rocketmq.domain.RmTemplateSwitch;
+import com.ruoyi.rocketmq.mapper.RmMonitorTemplateMapper;
+import com.ruoyi.rocketmq.mapper.RmTemplateLinuxMapper;
+import com.ruoyi.rocketmq.mapper.RmTemplateSwitchMapper;
+import com.ruoyi.system.api.RemoteRevenueConfigService;
+import com.ruoyi.system.api.domain.RmResourceGroupRemote;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.beans.factory.annotation.Autowired;
+
+import java.lang.reflect.Field;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+import java.util.stream.Collectors;
+@Slf4j
+public class DataProcessUtil {
+    @Autowired
+    private RmMonitorTemplateMapper rmMonitorTemplateMapper;
+    @Autowired
+    private static RmTemplateLinuxMapper rmTemplateLinuxMapper;
+    @Autowired
+    private static RmTemplateSwitchMapper rmTemplateSwitchMapper;
+    @Autowired
+    private static RemoteRevenueConfigService remoteRevenueConfigService;
+
+    /**
+     * 补充资源组信息
+     */
+    public static String getResourceGroupNameById(Long resourceGroupId) {
+        try {
+            R<RmResourceGroupRemote> resourceGroupResponse = remoteRevenueConfigService
+                    .getResourceGroupMsgById(resourceGroupId, SecurityConstants.INNER);
+
+            if (resourceGroupResponse != null && resourceGroupResponse.getData() != null) {
+                return resourceGroupResponse.getData().getGroupName();
+            }
+        } catch (Exception e) {
+            log.error("获取资源组信息失败，resourceGroupId: {}", resourceGroupId, e);
+        }
+        return null;
+    }
+    /**
+     * 处理Linux模板数据
+     */
+    public static void processLinuxTemplateData(Long templateId, Map<String, Object> result) {
+        RmTemplateLinux query = new RmTemplateLinux();
+        query.setTemplateId(templateId);
+        List<RmTemplateLinux> linuxItems = rmTemplateLinuxMapper.selectRmTemplateLinuxList(query);
+
+        if (!linuxItems.isEmpty()) {
+            Map<String, List<RmTemplateLinux>> linuxData = new HashMap<>();
+            linuxData.put("cpu", filterItemsByDataType(linuxItems, "CPU"));
+            linuxData.put("other", filterItemsByDataType(linuxItems, "OTHER"));
+            linuxData.put("point", filterItemsByDataType(linuxItems, "POINT"));
+            linuxData.put("net", filterItemsByDataType(linuxItems, "NET"));
+            linuxData.put("disk", filterItemsByDataType(linuxItems, "DISK"));
+            linuxData.put("docker", filterItemsByDataType(linuxItems, "DOCKER"));
+
+            result.put("linux", linuxData);
+        }
+    }
+
+    /**
+     * 处理Switch模板数据
+     */
+    public static void processSwitchTemplateData(Long templateId, Map<String, Object> result) {
+        RmTemplateSwitch query = new RmTemplateSwitch();
+        query.setTemplateId(templateId);
+        List<RmTemplateSwitch> switchItems = rmTemplateSwitchMapper.selectRmTemplateSwitchList(query);
+
+        if (!switchItems.isEmpty()) {
+            Map<String, List<RmTemplateSwitch>> switchData = new HashMap<>();
+            switchData.put("system", filterItemsByDataType(switchItems, "SYSTEM"));
+            switchData.put("mpu", filterItemsByDataType(switchItems, "MPU"));
+            switchData.put("power", filterItemsByDataType(switchItems, "POWERSOURCE"));
+            switchData.put("netport", filterItemsByDataType(switchItems, "NETPORT"));
+            switchData.put("lightmodule", filterItemsByDataType(switchItems, "LIGHTMODULE"));
+            switchData.put("fan", filterItemsByDataType(switchItems, "FAN"));
+
+            result.put("switch", switchData);
+        }
+    }
+    /**
+     * 通用过滤方法
+     */
+    public static  <T> List<T> filterItemsByDataType(List<T> items, String dataType) {
+        return items.stream()
+                .filter(item -> {
+                    try {
+                        Field dataTypeField = item.getClass().getDeclaredField("dataType");
+                        dataTypeField.setAccessible(true);
+                        return dataType.equals(dataTypeField.get(item));
+                    } catch (Exception e) {
+                        return false;
+                    }
+                })
+                .collect(Collectors.toList());
+    }
+}