package com.iailab.module.infra.service.monitordisk;

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.iailab.framework.tenant.core.aop.TenantIgnore;
import com.iailab.module.infra.dal.dataobject.monitormem.MonitorMemDO;
import org.apache.commons.lang3.ObjectUtils;
import org.springframework.stereotype.Service;
import javax.annotation.Resource;
import org.springframework.validation.annotation.Validated;

import com.iailab.module.infra.controller.admin.monitordisk.vo.*;
import com.iailab.module.infra.dal.dataobject.monitordisk.MonitorDiskDO;
import com.iailab.framework.common.pojo.PageResult;
import com.iailab.framework.common.util.object.BeanUtils;

import com.iailab.module.infra.dal.mysql.monitordisk.MonitorDiskMapper;

import java.math.BigDecimal;
import java.time.LocalDateTime;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

import static com.iailab.framework.common.exception.util.ServiceExceptionUtil.exception;
import static com.iailab.module.infra.enums.ErrorCodeConstants.*;

/**
 * 磁盘监控日志 Service 实现类
 *
 * @author 超级管理员
 */
@Service
@Validated
public class MonitorDiskServiceImpl implements MonitorDiskService {

    @Resource
    private MonitorDiskMapper monitorDiskMapper;

    @Override
    public Long createMonitorDisk(MonitorDiskSaveReqVO createReqVO) {
        // 插入
        MonitorDiskDO monitorDisk = BeanUtils.toBean(createReqVO, MonitorDiskDO.class);
        monitorDiskMapper.insert(monitorDisk);
        // 返回
        return monitorDisk.getId();
    }

    @Override
    public void createMonitorDiskBatch(List<MonitorDiskSaveReqVO> createReqVOS) {
        // 插入
        List<MonitorDiskDO> monitorDiskDOS = BeanUtils.toBean(createReqVOS, MonitorDiskDO.class);
        monitorDiskDOS.stream().forEach(monitorDiskDO -> {
            monitorDiskDO.setTenantId(1);
            monitorDiskDO.setCreator("iailab");
        });
        monitorDiskMapper.insertBatch(monitorDiskDOS);
    }

    @Override
    public void updateMonitorDisk(MonitorDiskSaveReqVO updateReqVO) {
        // 校验存在
        validateMonitorDiskExists(updateReqVO.getId());
        // 更新
        MonitorDiskDO updateObj = BeanUtils.toBean(updateReqVO, MonitorDiskDO.class);
        monitorDiskMapper.updateById(updateObj);
    }

    @Override
    public void deleteMonitorDisk(Long id) {
        // 校验存在
        validateMonitorDiskExists(id);
        // 删除
        monitorDiskMapper.deleteById(id);
    }

    private void validateMonitorDiskExists(Long id) {
        if (monitorDiskMapper.selectById(id) == null) {
            throw exception(MONITOR_DISK_NOT_EXISTS);
        }
    }

    @Override
    public MonitorDiskDO getMonitorDisk(Long id) {
        return monitorDiskMapper.selectById(id);
    }

    @Override
    public PageResult<MonitorDiskDO> getMonitorDiskPage(MonitorDiskPageReqVO pageReqVO) {
        return monitorDiskMapper.selectPage(pageReqVO);
    }

    /**
     * 折线图展示
     * @param reqVO 查询
     * @return
     */
    @Override
    public List<Map<String, List<Map<String, Object>>>> getMonitorDiskList(MonitorDiskReqVO reqVO) {
        List<MonitorDiskDO> monitorDiskList = monitorDiskMapper.getMonitorDiskList(reqVO);
        if(ObjectUtils.isNotEmpty(monitorDiskList)) {
            // 根据 hostName 字段进行分组
            Map<String, List<MonitorDiskDO>> hostNameCollect = monitorDiskList.stream()
                    .collect(Collectors.groupingBy(MonitorDiskDO::getHostName));
            List<Map<String, List<Map<String, Object>>>> host = hostNameCollect.entrySet().stream().map(hostEntry -> {
                List<MonitorDiskDO> hostValue = hostEntry.getValue();
                String key = hostEntry.getKey();
                Map<String, List<Map<String, Object>>> hostItem = new HashMap<>();
                // 根据 createTime 字段进行分组
                Map<LocalDateTime, List<MonitorDiskDO>> createTimeCollect = hostValue.stream()
                        .collect(Collectors.groupingBy(MonitorDiskDO::getCreateTime));
                List<Map<String, Object>> collect = createTimeCollect.entrySet().stream().map(entry -> {
                    List<MonitorDiskDO> value = entry.getValue();
                    Map<String, Object> item = new HashMap<>();
                    value.stream().forEach(monitorDiskDO -> {
                        item.put("createTime", monitorDiskDO.getCreateTime());
                        item.put(monitorDiskDO.getDisk(), monitorDiskDO.getSpaceRatio().multiply(new BigDecimal(100)));
                    });
                    return item;
                }).collect(Collectors.toList());
                hostItem.put(key, collect);
                return hostItem;
            }).collect(Collectors.toList());
            return host;
        }
        return null;
    }

    /**
     * 饼图展示
     * @param reqVO 查询
     * @return
     */
    @Override
    @TenantIgnore
    public List<Map<String, Object>> getMonitorDiskInfo(MonitorDiskReqVO reqVO) {
        List<MonitorDiskDO> monitorDiskInfo = monitorDiskMapper.getMonitorDiskInfo(reqVO);
        if(ObjectUtils.isNotEmpty(monitorDiskInfo)) {
            // 根据 hostName 字段进行分组
            Map<String, List<MonitorDiskDO>> hostNameCollect = monitorDiskInfo.stream()
                    .collect(Collectors.groupingBy(MonitorDiskDO::getHostName));
            List<Map<String, Object>> collect = hostNameCollect.entrySet().stream().map(entry -> {
                List<MonitorDiskDO> value = entry.getValue();
                Map<String, Object> item = new HashMap<>();
                List<Map<String, Object>> disks = new ArrayList<>();
                value.stream().forEach(monitorDiskDO -> {
                    Map<String, Object> diskMap = new HashMap<>();
                    item.put("name", monitorDiskDO.getHostName());
                    item.put("ip", monitorDiskDO.getHostIp());
                    diskMap.put("disk", monitorDiskDO.getDisk());
                    diskMap.put("total", monitorDiskDO.getSpaceTotal().divide(new BigDecimal(1024)).setScale(3, BigDecimal.ROUND_HALF_UP).toString());
                    diskMap.put("used", monitorDiskDO.getSpaceUsed().divide(new BigDecimal(1024)).setScale(2, BigDecimal.ROUND_HALF_UP));
                    disks.add(diskMap);
                });
                item.put("disks", disks);
                return item;
            }).collect(Collectors.toList());
            return collect;
        }
        return null;
    }

    @Override
    public List<String> getAllHost() {
        QueryWrapper<MonitorDiskDO> queryWrapper = new QueryWrapper<>();
        queryWrapper.select("DISTINCT host_name"); // 添加 DISTINCT 修饰
        List<String> hosts = monitorDiskMapper.selectObjs(queryWrapper);
        return hosts;
    }

    @Override
    public List<String> getAllIp() {
        QueryWrapper<MonitorDiskDO> queryWrapper = new QueryWrapper<>();
        queryWrapper.select("DISTINCT host_ip"); // 添加 DISTINCT 修饰
        List<String> ips = monitorDiskMapper.selectObjs(queryWrapper);
        return ips;
    }

}