whyclxw
2024-04-28 b25b814a949d2065a61fd3167206ffb51a57af65
src/main/java/com/whyc/service/PowerInfService.java
@@ -9,11 +9,14 @@
import com.whyc.dto.paramter.PowerInfPar;
import com.whyc.mapper.PowerInfMapper;
import com.whyc.pojo.PowerInf;
import com.whyc.util.ActionUtil;
import org.springframework.stereotype.Service;
import javax.annotation.Resource;
import java.util.HashMap;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;
@Service
@@ -144,7 +147,7 @@
    public Response updateCancelExceptionCause(int num) {
        UpdateWrapper<PowerInf> update = Wrappers.update();
        update.set("exception_cause", null).set("exception_cause_analysis", null).eq("num", num);
        mapper.update(null, update);
        mapper.update((PowerInf) ActionUtil.objeNull, update);
        return new Response().setII(1, "取消成功");
    }
@@ -170,8 +173,36 @@
        QueryWrapper<PowerInf> query = Wrappers.query();
        query.select("distinct model");
        List<PowerInf> powerInfList = mapper.selectList(query);
        List<String> modelList = powerInfList.stream().filter(model->model!=null).map(PowerInf::getModel).collect(Collectors.toList());
        return new Response().set(1,modelList);
        List<String> modelList = powerInfList.stream().filter(model -> model != null).map(PowerInf::getModel).collect(Collectors.toList());
        return new Response().set(1, modelList);
    }
    //根据品牌查询型号
    public Response getModelListByPwrProducer(List<String> producer) {
        QueryWrapper<PowerInf> query = Wrappers.query();
        query.select("distinct model");
        if (producer != null && producer.size() > 0) {
            query.in("PowerProducer", producer);
        }
        List<PowerInf> powerInfList = mapper.selectList(query);
        List<String> modelList = powerInfList.stream().filter(model -> model != null).map(PowerInf::getModel).collect(Collectors.toList());
        return new Response().set(1, modelList);
    }
    public String getStationIdByPowerDeviceId(Integer powerDeviceId) {
        QueryWrapper<PowerInf> query = Wrappers.query();
        query.eq("PowerDeviceId",powerDeviceId).last(" limit 1");
        return mapper.selectOne(query).getStationId();
    }
    public Response getProducerCount(int userId) {
        List<PowerInf> powerInfList = getPwrdevInfAnalysis(userId);
        Map<String, List<PowerInf>> powerProducerMap = powerInfList.stream().collect(Collectors.groupingBy(PowerInf::getPowerProducer));
        Map<String, Object> AnalysisMap = new HashMap<>();
        for (String producer : powerProducerMap.keySet()) {
            AnalysisMap.put(producer, powerProducerMap.get(producer).size());
        }
        return new Response().set(1, AnalysisMap, "电源信息");
    }
}