ansteel-biz/src/main/java/com/iailab/module/ansteel/coking/service/impl/CokingTraceEnergyServiceImpl.java

@@ -1,14 +1,20 @@
package com.iailab.module.ansteel.coking.service.impl;

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.iailab.framework.common.util.date.DateUtils;
import com.iailab.framework.common.util.object.ConvertUtils;
import com.iailab.module.ansteel.coking.dao.CokingTraceEnergyDao;
import com.iailab.module.ansteel.coking.dto.CokingTraceEnergyDTO;
import com.iailab.module.ansteel.coking.entity.CokingProdDayEntity;
import com.iailab.module.ansteel.coking.entity.CokingTraceEnergyEntity;
import com.iailab.module.ansteel.coking.service.CokingTraceEnergyService;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import java.util.List;
import java.util.Map;
import java.util.*;

import static com.iailab.framework.common.pojo.CommonResult.success;

/**
 * 焦化工序能耗整体情况
@@ -23,7 +29,37 @@
    private CokingTraceEnergyDao cokingTraceEnergyDao;

    @Override
    public List<CokingTraceEnergyEntity> list(Map<String, Object> params) {
        return cokingTraceEnergyDao.selectList(new QueryWrapper<>());
    public Map<String,List<CokingTraceEnergyDTO>> list(Map<String, Object> params) {
        Map<String,List<CokingTraceEnergyDTO>> result = new HashMap<>(2);

        String clock = (String) params.get("clock");
        String month;
        if (StringUtils.isBlank(clock)) {
            Calendar calendar = Calendar.getInstance();
            calendar.add(Calendar.DAY_OF_YEAR, -1);
            clock = DateUtils.format(calendar.getTime(), "yyyy-MM-dd");
            month = DateUtils.format(calendar.getTime(), "yyyy-MM");
        }else {
            Date parse = DateUtils.parse(clock, "yyyy-MM-dd");
            month = DateUtils.format(parse, "yyyy-MM");
        }

        // 天数据
        QueryWrapper<CokingTraceEnergyEntity> queryWrapperDay = new QueryWrapper<>();
        queryWrapperDay.eq("clock", clock);
        List<CokingTraceEnergyEntity> dayList = cokingTraceEnergyDao.selectList(queryWrapperDay);
        result.put("day",ConvertUtils.sourceToTarget(dayList,CokingTraceEnergyDTO.class));

        // 月数据
        QueryWrapper<CokingTraceEnergyEntity> queryWrapperMonth = new QueryWrapper<>();
        queryWrapperMonth.like("clock", month);
        queryWrapperMonth.select("type","SUM(steam_p) as steam_p","SUM(steam_c) as steam_c","SUM(gas_p) as gas_p","SUM(gas_c) as gas_c","SUM(power_c) as power_c")
                .groupBy("type");

        List<CokingTraceEnergyEntity> monthList = cokingTraceEnergyDao.selectList(queryWrapperMonth);
        monthList.forEach(e -> e.setClock(month));
        result.put("month",ConvertUtils.sourceToTarget(monthList,CokingTraceEnergyDTO.class));

        return result;
    }
}