lxw
2023-08-15 160e150009b51a39fa95d9462c3798ba28d51a09
src/main/java/com/whyc/service/PowerInfService.java
@@ -6,6 +6,7 @@
import com.github.pagehelper.PageHelper;
import com.github.pagehelper.PageInfo;
import com.whyc.dto.Response;
import com.whyc.dto.paramter.PowerInfPar;
import com.whyc.mapper.PowerInfMapper;
import com.whyc.pojo.PowerInf;
import org.springframework.stereotype.Service;
@@ -13,6 +14,7 @@
import javax.annotation.Resource;
import java.util.Iterator;
import java.util.List;
import java.util.stream.Collectors;
@Service
public class PowerInfService {
@@ -91,9 +93,9 @@
        return new Response().set(1, list, "查询成功");
    }
    public Response getPowerDevicesPage(int pageNum, int pageSize, Long uId, String stationId, String stationName1, String stationName2, String stationName5, String powerProducer) {
    public Response getPowerDevicesPage(int pageNum, int pageSize, Long uId, PowerInfPar powerInfPar) {
        PageHelper.startPage(pageNum, pageSize);
        List<PowerInf> list = mapper.getPowerDevicesPage(uId, stationId, stationName1, stationName2, stationName5, powerProducer);
        List<PowerInf> list = mapper.getPowerDevicesPage(uId, powerInfPar);
        PageInfo<PowerInf> pageInfo = new PageInfo<>(list);
        return new Response().set(1, pageInfo, "查询成功");
    }
@@ -163,4 +165,25 @@
        List<String> list = mapper.getPwrProducer();
        return new Response().setII(1, list.size() > 0 ? true : false, list, "获取所有的电源品牌");
    }
    public Response getModelList() {
        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);
    }
    //根据品牌查询型号
    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);
    }
}