Code/sms_water/src/main/java/com/huaxu/service/DayReportService.java

package com.huaxu.service;


import com.baomidou.mybatisplus.core.metadata.IPage;
import com.huaxu.dao.DayReportMapper;
import com.huaxu.dto.DeviceDto;
import com.huaxu.dto.ReportAttributeDto;
import com.huaxu.dto.ReportDto;
import com.huaxu.entity.DayReportEntity;
import com.huaxu.model.LoginUser;
import com.huaxu.util.UserUtil;
import org.apache.ibatis.annotations.Param;
import org.apache.xmlbeans.impl.xb.xsdschema.BlockSet;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.mongodb.core.aggregation.DateOperators;
import org.springframework.stereotype.Service;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;

import javax.annotation.Resource;

import org.springframework.transaction.annotation.Transactional;

import java.util.*;

/**
 * 日报Service接口
 *
 * @author: WYY
 * @date 2020-12-03 16:20
 */
@Service
public class DayReportService extends ServiceImpl<DayReportMapper, DayReportEntity> {

    @Resource
    private DayReportMapper dayReportMapper;
    @Autowired
    private SceneService sceneService;
    @Autowired
    private DeviceService deviceService;
    @Autowired
    private DeviceParmService deviceParmService;


    /**
     * 查列表
     */
    public List<DayReportEntity> findList(DayReportEntity dayReportEntity) {
        return dayReportMapper.findList(dayReportEntity);
    }

    /**
     * 保存
     */
    public boolean addDayReport(DayReportEntity dayReport) {
        if (this.save(dayReport)) {
            return true;
        }
        return false;
    }


    /**
     * 根居ID获取对象
     */
    public DayReportEntity findDayReportById(Long id) {
        return dayReportMapper.findDayReportById(id);
    }

    public Page<ReportDto> findPage(IPage<ReportDto> page, Long[] ids, Integer year, Integer month, Integer day) {
        LoginUser currentUser = UserUtil.getCurrentUser();
        Page<ReportDto> reportPage = new Page<>();
        //查询场景下的所有设备信息
        List<DeviceDto> devices = new ArrayList<>();
        DeviceDto deviceDto = new DeviceDto();
        for (Long id : ids) {
            deviceDto.setSceneIds(sceneService.findByParentIdsLike(id));
            if (deviceDto.getSceneIds().size() > 0)
                devices.addAll(deviceService.selectList(deviceDto));
        }
        if (devices.size() > 0) {
            //根据设备ID查询报表测点信息
            DayReportEntity dayReportEntity = new DayReportEntity();
            dayReportEntity.setYear(year);
            dayReportEntity.setMonth(month);
            dayReportEntity.setDay(day);
            dayReportEntity.setTenantId(currentUser.getTenantId());
            dayReportEntity.setParentSceneIds(ids);
            dayReportEntity.setDeviceIds(devices);
            //查询到分页的行数
            reportPage = dayReportMapper.findPage(page, dayReportEntity);
            //查询标题
            ReportDto reportDto = new ReportDto();
            reportDto.setYear(year);
            reportDto.setMonth(month);
            reportDto.setDay(day);
            reportDto.setDeviceIds(devices);
            List<ReportAttributeDto> reportAttributeDtos2 = deviceParmService.findAttributeNameList(reportDto);
            List<String> dataTitle = new ArrayList<>();
            Map<String, Integer> mapsTitle = new LinkedHashMap<>();
            int numCount = 0;
            for (ReportAttributeDto title : reportAttributeDtos2) {
                dataTitle.add(title.getAttributeName());
                if (!mapsTitle.containsKey(title.getAttributeName()))
                    mapsTitle.put(title.getAttributeName(), numCount++);
            }
            //固定参数项
            List<ReportDto> reportDtos = dayReportMapper.findReport(reportDto);
            //单个属性值
            for (ReportDto item : reportPage.getRecords()) {
                item.setCollectDate(item.getYear() + "-" + String.format("%02d", item.getMonth()) + "-" + String.format("%02d", item.getDay()) + " " + String.format("%02d", item.getHour()) + ":00");
                item.setDeviceIds(devices);
                item.setDataTitle(dataTitle);
                //固定参数项
                calcUsage(item, reportDtos);
                //动态参数项
                List<ReportAttributeDto> reportAttributeDtos = dayReportMapper.findAttributeList(item);
                //item.setDataValues(reportAttributeDtos);
                Map<String, ReportAttributeDto> mapData = new LinkedHashMap<>();
                for (ReportAttributeDto itemData : reportAttributeDtos) {
                    if (!mapData.containsKey(itemData.getAttributeName()))
                        mapData.put(itemData.getAttributeName(), itemData);
                }
                List<Map<String, Object>> maps = new ArrayList<>();
                for (String key : mapsTitle.keySet()) {
                    Map<String, Object> map = new LinkedHashMap<>();
                    map.put("attributeName", mapData.get(key) == null ? key : mapData.get(key).getAttributeName());
                    map.put("minValue" + mapsTitle.get(key), mapData.get(key) == null ? "-" : mapData.get(key).getMinValue());
                    map.put("maxValue" + mapsTitle.get(key), mapData.get(key) == null ? "-" : mapData.get(key).getMaxValue());
                    map.put("avgValue" + mapsTitle.get(key), mapData.get(key) == null ? "-" : mapData.get(key).getAvgValue());
                    map.put("sumValue" + mapsTitle.get(key), mapData.get(key) == null ? "-" : mapData.get(key).getSumValue());
                    map.put("latestValue" + mapsTitle.get(key), mapData.get(key) == null ? "-" : mapData.get(key).getLatestValue());
                    maps.add(map);
                }
                item.setDataMapValues(maps);
            }
        }
        return reportPage;
    }
    //计算固定的参数项
    private void calcUsage(ReportDto item, List<ReportDto> reportDtos) {
        //固定参数项
        int index = -1;
        for (int i = 0; i < reportDtos.size(); i++) {
            if (reportDtos.get(i).getHour().equals(item.getHour()))
                index = i;
        }
        if (index != -1) {
            if ((index + 2) <= reportDtos.size()) {
                item.setYieldWaterUsage(reportDtos.get(index).getYieldWaterUsage() - reportDtos.get(index + 1).getYieldWaterUsage());
                item.setIntakeWaterUsage(reportDtos.get(index).getIntakeWaterUsage() - reportDtos.get(index + 1).getIntakeWaterUsage());
                item.setPowerUsage(reportDtos.get(index).getPowerUsage() - reportDtos.get(index + 1).getPowerUsage());
                item.setDrugUsage(reportDtos.get(index).getDrugUsage() - reportDtos.get(index + 1).getDrugUsage());
            } else {
                item.setYieldWaterUsage(0d);
                item.setIntakeWaterUsage(0d);
                item.setPowerUsage(0d);
                item.setDrugUsage(0d);
            }
        }
    }
}