saas-houduan/ruoyi-rocketmq/src/main/java/com/ruoyi/rocketmq/utils/DataProcessUtil.java

package com.ruoyi.rocketmq.utils;

import com.ruoyi.common.core.constant.SecurityConstants;
import com.ruoyi.common.core.domain.R;
import com.ruoyi.common.core.utils.StringUtils;
import com.ruoyi.rocketmq.domain.RmTemplateLinux;
import com.ruoyi.rocketmq.domain.RmTemplateSwitch;
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 com.ruoyi.system.api.domain.RmResourceRegistrationRemote;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import java.lang.reflect.Field;
import java.util.Collections;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;
@Slf4j
@Component
public class DataProcessUtil {
    @Autowired
    private  RmTemplateLinuxMapper rmTemplateLinuxMapper;
    @Autowired
    private  RmTemplateSwitchMapper rmTemplateSwitchMapper;
    @Autowired
    private  RemoteRevenueConfigService remoteRevenueConfigService;

    /**
     * 补充资源组信息
     */
    public  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;
    }
    /**
     * 资源组信息
     */
    public  RmResourceGroupRemote getResourceMsgById(Long resourceGroupId) {
        try {
            R<RmResourceGroupRemote> resourceGroupResponse = remoteRevenueConfigService
                    .getResourceGroupMsgById(resourceGroupId, SecurityConstants.INNER);

            if (resourceGroupResponse != null && resourceGroupResponse.getData() != null) {
                return resourceGroupResponse.getData();
            }
        } catch (Exception e) {
            log.error("获取资源信息失败，resourceGroupId: {}", resourceGroupId, e);
        }
        return null;
    }
    /**
     * 处理Linux模板数据
     */
    public  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("vfs", 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  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("switchOther", filterItemsByDataType(switchItems, "SYSTEM"));
            switchData.put("switchMpu", filterItemsByDataType(switchItems, "MPU"));
            switchData.put("switchPwr", filterItemsByDataType(switchItems, "POWERSOURCE"));
            switchData.put("switchNet", filterItemsByDataType(switchItems, "NETPORT"));
            switchData.put("switchModule", filterItemsByDataType(switchItems, "LIGHTMODULE"));
            switchData.put("switchFan", 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());
    }
    /**
     * 获取资源设备列表
     */
    public  List<RmResourceRegistrationRemote> getResourceDevices(Long resourceGroupId) {
        R<RmResourceGroupRemote> groupResponse = remoteRevenueConfigService
                .getResourceGroupMsgById(resourceGroupId, SecurityConstants.INNER);

        if (groupResponse == null || groupResponse.getData() == null
                || StringUtils.isEmpty(groupResponse.getData().getIncludedDevicesId())) {
            log.error("资源组信息不完整，resourceGroupId: {}", resourceGroupId);
            return Collections.emptyList();
        }

        String[] deviceIds = groupResponse.getData().getIncludedDevicesId().split(",");
        R<List<RmResourceRegistrationRemote>> devicesResponse = remoteRevenueConfigService
                .getRegistrationByIds(deviceIds, SecurityConstants.INNER);

        return devicesResponse != null ? devicesResponse.getData() : Collections.emptyList();
    }
}