gaoluyang
4 天以前 8c65f898ed6e5b0c7a94c792c4562f6ec831dccd
yudao-module-mes/src/main/java/cn/iocoder/yudao/module/mes/controller/admin/dv/machinery/MesDvMachineryController.java
@@ -54,6 +54,8 @@
    private MesMdWorkshopService workshopService;
    @Resource
    private cn.iocoder.yudao.module.mes.service.md.workstation.MesMdWorkstationService workstationService;
    @Resource
    private cn.iocoder.yudao.module.mes.service.md.workstation.MesMdWorkstationMachineService workstationMachineService;
    @PostMapping("/create")
    @Operation(summary = "创建设备")
@@ -82,7 +84,6 @@
    @GetMapping("/get")
    @Operation(summary = "获得设备")
    @Parameter(name = "id", description = "编号", required = true, example = "1024")
    @PreAuthorize("@ss.hasPermission('mes:dv-machinery:query')")
    public CommonResult<MesDvMachineryRespVO> getMachinery(@RequestParam("id") Long id) {
        MesDvMachineryDO machinery = machineryService.getMachinery(id);
        if (machinery == null) {
@@ -93,7 +94,6 @@
    @GetMapping("/page")
    @Operation(summary = "获得设备分页")
    @PreAuthorize("@ss.hasPermission('mes:dv-machinery:query')")
    public CommonResult<PageResult<MesDvMachineryRespVO>> getMachineryPage(@Valid MesDvMachineryPageReqVO pageReqVO) {
        PageResult<MesDvMachineryDO> pageResult = machineryService.getMachineryPage(pageReqVO);
        return success(new PageResult<>(buildMachineryRespVOList(pageResult.getList()), pageResult.getTotal()));
@@ -113,10 +113,17 @@
    }
    @GetMapping("/simple-list")
    @Operation(summary = "获得设备精简列表", description = "主要用于前端的下拉选项")
    @PreAuthorize("@ss.hasPermission('mes:dv-machinery:query')")
    public CommonResult<List<MesDvMachineryRespVO>> getMachinerySimpleList() {
    @Operation(summary = "获得设备精简列表", description = "主要用于前端的下拉选项;传 workstationId 时只返回该工作站绑定的设备")
    @Parameter(name = "workstationId", description = "工作站编号,可选;传则只返回该工作站绑定的设备")
    public CommonResult<List<MesDvMachineryRespVO>> getMachinerySimpleList(
            @RequestParam(value = "workstationId", required = false) Long workstationId) {
        List<MesDvMachineryDO> list = machineryService.getMachineryList();
        if (workstationId != null) {
            var boundIds = cn.iocoder.yudao.framework.common.util.collection.CollectionUtils.convertSet(
                    workstationMachineService.getWorkstationMachineListByWorkstationId(workstationId),
                    cn.iocoder.yudao.module.mes.dal.dataobject.md.workstation.MesMdWorkstationMachineDO::getMachineryId);
            list = list.stream().filter(machinery -> boundIds.contains(machinery.getId())).toList();
        }
        return success(BeanUtils.toBean(list, MesDvMachineryRespVO.class));
    }