diff --git a/src/main/java/tech/riemann/ims/controller/platform/material/MaterialController.java b/src/main/java/tech/riemann/ims/controller/platform/material/MaterialController.java index c5fa92e..0f3ad5c 100644 --- a/src/main/java/tech/riemann/ims/controller/platform/material/MaterialController.java +++ b/src/main/java/tech/riemann/ims/controller/platform/material/MaterialController.java @@ -80,10 +80,16 @@ public class MaterialController { @PostMapping("material/list") @Operation(summary = "查询所有物料列表") - public List all(@Parameter(description = "类型") @RequestParam(required = false, defaultValue = "") String type) { + public List all(@Parameter(description = "类型") @RequestParam(required = false, defaultValue = "") String type, + @Parameter(description = "是否赋码") @RequestParam(required = false) Boolean assignRule) { String[] types = StringUtils.isNotBlank(type) ? type.split(",") : new String[0]; - List materials = materialService.queryLikeRight(List.of(types)); - materials.forEach(item -> item.setTypeName(typeService.getTypeName(item.getType()))); + List materials = materialService.queryLikeRight(List.of(types)); + if (Boolean.FALSE.equals(assignRule)) { + materials.stream().filter(item -> Boolean.TRUE.equals(item.getAssignRule())) + .forEach(item -> item.setTypeName(typeService.getTypeName(item.getType()))); + } else { + materials.forEach(item -> item.setTypeName(typeService.getTypeName(item.getType()))); + } return materials; }