whyclxw
2025-05-28 e16302f9d475c7cc4dd18c5abf1a23cb5502e362
src/main/java/com/whyc/service/PeakStatisticsService.java
@@ -2,6 +2,7 @@
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.whyc.dto.Response;
import com.whyc.dto.StaticEleprice;
import com.whyc.mapper.BattInfMapper;
@@ -9,13 +10,14 @@
import com.whyc.pojo.Battinf;
import com.whyc.pojo.PeakStatistics;
import com.whyc.util.ActionUtil;
import com.whyc.util.DateUtil;
import com.whyc.util.ThreadLocalUtil;
import org.springframework.stereotype.Service;
import javax.annotation.Resource;
import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;
import java.util.*;
import java.util.stream.Collectors;
@Service
public class PeakStatisticsService {
@@ -133,4 +135,21 @@
        updateWrapper.eq("battGroupId",battGroupId);
        battInfMapper.update(battinf,updateWrapper);
    }
    public Map<Integer, List<PeakStatistics>> getTodayMapByBattGroupId(int battGroupId) {
        QueryWrapper<PeakStatistics> query = Wrappers.query();
        Date now = new Date();
        //String today = DateUtil.YYYY_MM_DD.format(now);
        String today = ThreadLocalUtil.format(now,3);
        query.eq("record_time",today).eq("BattGroupId",battGroupId).in("test_type",2,3);
        List<PeakStatistics> peakStatistics = mapper.selectList(query);
        Map<Integer, List<PeakStatistics>> collect = peakStatistics.stream().collect(Collectors.groupingBy(PeakStatistics::getTestType));
        if(!collect.containsKey(2)){
            collect.put(2,null);
        };
        if(!collect.containsKey(3)){
            collect.put(3,null);
        };
        return collect;
    }
}