根据品牌查询型号

lxw

2023-08-07 cf114af3352dcaa9c73f162fd43090d689983d58

根据品牌查询型号

 src/main/java/com/whyc/controller/PowerInfController.java

@@ -19,6 +19,8 @@
import org.springframework.util.StringUtils;
import org.springframework.web.bind.annotation.*;

import java.util.List;

@RequestMapping("powerInf")
@RestController
@Api(tags = "数据管理-电源信息")
@@ -215,4 +217,10 @@
        return service.getModelList();
    }

    @PostMapping("getModelListByPwrProducer")
    @ApiOperation(value = "根据品牌查询型号")
    public Response getModelListByPwrProducer(@RequestBody List<String> producer) {
        return service.getModelListByPwrProducer(producer);
    }

}

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

@@ -170,8 +170,20 @@
        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);
    }
}