修改

whyclxw

2024-07-01 0ce771b4a4ff5897dc91bb1b84a4561781b00ca5

 src/main/java/com/whyc/service/CKPowerDevAlarmService.java

@@ -1,6 +1,7 @@
package com.whyc.service;

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.github.pagehelper.PageHelper;
import com.github.pagehelper.PageInfo;
@@ -10,7 +11,8 @@
import org.springframework.stereotype.Service;

import javax.annotation.Resource;
import java.util.List;
import java.util.*;
import java.util.stream.Collectors;

@Service
public class CKPowerDevAlarmService {
@@ -18,8 +20,8 @@
    @Resource
    private CKPowerDevAlarmMapper mapper;

    public Response getPage(CKPowerDevAlarm alarm, int pageNum, int pageSize) {
        PageHelper.startPage(pageNum,pageSize);
    public Response getPage(CKPowerDevAlarm alarm) {
        PageHelper.startPage(alarm.getPageNum(),alarm.getPageSize());
        QueryWrapper<CKPowerDevAlarm> query = Wrappers.query();
        if(alarm.getDevType()!=0) {
            query.eq("dev_type", alarm.getDevType());
@@ -34,4 +36,32 @@
        PageInfo<CKPowerDevAlarm> pageInfo = new PageInfo<>(alarmList);
        return new Response().set(1,pageInfo);
    }

    public Response updateForConfirm(int id) {
        UpdateWrapper<CKPowerDevAlarm> update = Wrappers.update();
        update.set("alm_is_confirmed",1);
        update.set("alm_confirmed_time",new Date());
        update.eq("num",id);

        mapper.update(null,update);
        return new Response().set(1,"确认完成");
    }

    public Response getCountByLevel() {
        Map<Integer,Integer> countMap = new HashMap<>();
        countMap.put(0,0);
        countMap.put(1,0);
        countMap.put(2,0);
        countMap.put(3,0);
        QueryWrapper<CKPowerDevAlarm> query = Wrappers.query();
        query.select("alm_level");
        List<CKPowerDevAlarm> alarmList = mapper.selectList(query);
        countMap.put(0,alarmList.size());
        Map<Integer, List<CKPowerDevAlarm>> levelMap = alarmList.stream().collect(Collectors.groupingBy(CKPowerDevAlarm::getAlmLevel));
        Set<Integer> levelKeyMap = levelMap.keySet();
        for (Integer level : levelKeyMap) {
            countMap.put(level,levelMap.get(level).size());
        }
        return new Response().set(1,countMap);
    }
}