package cn.iocoder.yudao.module.mes.controller.admin.pd.processparamtemplate;
|
|
import cn.hutool.core.collection.CollUtil;
|
import cn.iocoder.yudao.framework.apilog.core.annotation.ApiAccessLog;
|
import cn.iocoder.yudao.framework.common.pojo.CommonResult;
|
import cn.iocoder.yudao.framework.common.pojo.PageParam;
|
import cn.iocoder.yudao.framework.common.pojo.PageResult;
|
import cn.iocoder.yudao.framework.common.util.collection.MapUtils;
|
import cn.iocoder.yudao.framework.common.util.object.BeanUtils;
|
import cn.iocoder.yudao.framework.excel.core.util.ExcelUtils;
|
import cn.iocoder.yudao.module.mes.controller.admin.pd.processparamtemplate.vo.MesPdProcessParamTemplateImportExcelVO;
|
import cn.iocoder.yudao.module.mes.controller.admin.pd.processparamtemplate.vo.MesPdProcessParamTemplateImportRespVO;
|
import cn.iocoder.yudao.module.mes.controller.admin.pd.processparamtemplate.vo.MesPdProcessParamTemplatePageReqVO;
|
import cn.iocoder.yudao.module.mes.controller.admin.pd.processparamtemplate.vo.MesPdProcessParamTemplateRespVO;
|
import cn.iocoder.yudao.module.mes.controller.admin.pd.processparamtemplate.vo.MesPdProcessParamTemplateSaveReqVO;
|
import cn.iocoder.yudao.module.mes.dal.dataobject.md.unitmeasure.MesMdUnitMeasureDO;
|
import cn.iocoder.yudao.module.mes.dal.dataobject.pd.processparamtemplate.MesPdProcessParamTemplateDO;
|
import cn.iocoder.yudao.module.mes.dal.dataobject.pd.processparamtemplate.MesPdProcessParamTemplateDetailDO;
|
import cn.iocoder.yudao.module.mes.service.md.unitmeasure.MesMdUnitMeasureService;
|
import cn.iocoder.yudao.module.mes.service.pd.processparamtemplate.MesPdProcessParamTemplateService;
|
import io.swagger.v3.oas.annotations.Operation;
|
import io.swagger.v3.oas.annotations.Parameter;
|
import io.swagger.v3.oas.annotations.Parameters;
|
import io.swagger.v3.oas.annotations.tags.Tag;
|
import jakarta.annotation.Resource;
|
import jakarta.servlet.http.HttpServletResponse;
|
import jakarta.validation.Valid;
|
import org.springframework.security.access.prepost.PreAuthorize;
|
import org.springframework.validation.annotation.Validated;
|
import org.springframework.web.bind.annotation.*;
|
import org.springframework.web.multipart.MultipartFile;
|
|
import java.io.IOException;
|
import java.util.*;
|
import java.util.stream.Collectors;
|
|
import static cn.iocoder.yudao.framework.apilog.core.enums.OperateTypeEnum.EXPORT;
|
import static cn.iocoder.yudao.framework.common.pojo.CommonResult.success;
|
import static cn.iocoder.yudao.framework.common.util.collection.CollectionUtils.convertSet;
|
|
@Tag(name = "管理后台 - MES 工艺参数模板")
|
@RestController
|
@RequestMapping("/mes/pd/process-param-template")
|
@Validated
|
public class MesPdProcessParamTemplateController {
|
|
@Resource
|
private MesPdProcessParamTemplateService templateService;
|
|
@Resource
|
private MesMdUnitMeasureService unitMeasureService;
|
|
@PostMapping("/create")
|
@Operation(summary = "创建工艺参数模板")
|
@PreAuthorize("@ss.hasPermission('mes:pd-process-param-template:create')")
|
public CommonResult<Long> createProcessParamTemplate(@Valid @RequestBody MesPdProcessParamTemplateSaveReqVO createReqVO) {
|
return success(templateService.createProcessParamTemplate(createReqVO));
|
}
|
|
@PutMapping("/update")
|
@Operation(summary = "更新工艺参数模板")
|
@PreAuthorize("@ss.hasPermission('mes:pd-process-param-template:update')")
|
public CommonResult<Boolean> updateProcessParamTemplate(@Valid @RequestBody MesPdProcessParamTemplateSaveReqVO updateReqVO) {
|
templateService.updateProcessParamTemplate(updateReqVO);
|
return success(true);
|
}
|
|
@DeleteMapping("/delete")
|
@Operation(summary = "删除工艺参数模板")
|
@Parameter(name = "id", description = "编号", required = true)
|
@PreAuthorize("@ss.hasPermission('mes:pd-process-param-template:delete')")
|
public CommonResult<Boolean> deleteProcessParamTemplate(@RequestParam("id") Long id) {
|
templateService.deleteProcessParamTemplate(id);
|
return success(true);
|
}
|
|
@GetMapping("/get")
|
@Operation(summary = "获得工艺参数模板")
|
@Parameter(name = "id", description = "编号", required = true, example = "1024")
|
public CommonResult<MesPdProcessParamTemplateRespVO> getProcessParamTemplate(@RequestParam("id") Long id) {
|
MesPdProcessParamTemplateDO template = templateService.getProcessParamTemplate(id);
|
if (template == null) {
|
return success(null);
|
}
|
return success(buildTemplateRespVO(template));
|
}
|
|
@GetMapping("/page")
|
@Operation(summary = "获得工艺参数模板分页")
|
public CommonResult<PageResult<MesPdProcessParamTemplateRespVO>> getProcessParamTemplatePage(@Valid MesPdProcessParamTemplatePageReqVO pageReqVO) {
|
PageResult<MesPdProcessParamTemplateDO> pageResult = templateService.getProcessParamTemplatePage(pageReqVO);
|
List<MesPdProcessParamTemplateRespVO> voList = buildTemplateRespVOList(pageResult.getList());
|
return success(new PageResult<>(voList, pageResult.getTotal()));
|
}
|
|
@GetMapping("/export-excel")
|
@Operation(summary = "导出工艺参数模板 Excel")
|
@PreAuthorize("@ss.hasPermission('mes:pd-process-param-template:export')")
|
@ApiAccessLog(operateType = EXPORT)
|
public void exportProcessParamTemplateExcel(@Valid MesPdProcessParamTemplatePageReqVO pageReqVO,
|
HttpServletResponse response) throws IOException {
|
pageReqVO.setPageSize(PageParam.PAGE_SIZE_NONE);
|
List<MesPdProcessParamTemplateDO> list = templateService.getProcessParamTemplatePage(pageReqVO).getList();
|
// 导出 Excel
|
ExcelUtils.write(response, "工艺参数模板.xls", "数据", MesPdProcessParamTemplateRespVO.class,
|
buildTemplateRespVOList(list));
|
}
|
|
@GetMapping("/get-import-template")
|
@Operation(summary = "获得工艺参数模板导入模板")
|
public void importTemplate(HttpServletResponse response) throws IOException {
|
// 手动创建导出 demo
|
List<MesPdProcessParamTemplateImportExcelVO> list = Collections.singletonList(
|
MesPdProcessParamTemplateImportExcelVO.builder()
|
.templateName("示例模板").templateType(1).remark("").build()
|
);
|
// 输出
|
ExcelUtils.write(response, "工艺参数模板导入模板.xls", "工艺参数模板列表",
|
MesPdProcessParamTemplateImportExcelVO.class, list);
|
}
|
|
@PostMapping("/import")
|
@Operation(summary = "导入工艺参数模板")
|
@Parameters({
|
@Parameter(name = "file", description = "Excel 文件", required = true),
|
@Parameter(name = "updateSupport", description = "是否支持更新，默认为 false", example = "true")
|
})
|
@PreAuthorize("@ss.hasPermission('mes:pd-process-param-template:import')")
|
public CommonResult<MesPdProcessParamTemplateImportRespVO> importExcel(@RequestParam("file") MultipartFile file,
|
@RequestParam(value = "updateSupport", required = false,
|
defaultValue = "false") Boolean updateSupport) throws Exception {
|
List<MesPdProcessParamTemplateImportExcelVO> list = ExcelUtils.read(file, MesPdProcessParamTemplateImportExcelVO.class);
|
return success(templateService.importProcessParamTemplateList(list, updateSupport));
|
}
|
|
// ==================== 拼接 VO ====================
|
|
private MesPdProcessParamTemplateRespVO buildTemplateRespVO(MesPdProcessParamTemplateDO template) {
|
return buildTemplateRespVOList(Collections.singletonList(template)).get(0);
|
}
|
|
private List<MesPdProcessParamTemplateRespVO> buildTemplateRespVOList(List<MesPdProcessParamTemplateDO> list) {
|
if (CollUtil.isEmpty(list)) {
|
return Collections.emptyList();
|
}
|
// 1. 批量获取所有模板的明细
|
Map<Long, List<MesPdProcessParamTemplateDetailDO>> detailMap = new HashMap<>();
|
Set<Long> allUnitMeasureIds = new HashSet<>();
|
for (MesPdProcessParamTemplateDO template : list) {
|
List<MesPdProcessParamTemplateDetailDO> details =
|
templateService.getProcessParamTemplateDetailListByTemplateId(template.getId());
|
detailMap.put(template.getId(), details);
|
for (MesPdProcessParamTemplateDetailDO detail : details) {
|
if (detail.getUnitMeasureId() != null) {
|
allUnitMeasureIds.add(detail.getUnitMeasureId());
|
}
|
}
|
}
|
// 2. 批量获取计量单位
|
Map<Long, MesMdUnitMeasureDO> unitMeasureMap = unitMeasureService.getUnitMeasureMap(allUnitMeasureIds);
|
// 3. 拼接 VO
|
return BeanUtils.toBean(list, MesPdProcessParamTemplateRespVO.class, vo -> {
|
List<MesPdProcessParamTemplateDetailDO> details = detailMap.get(vo.getId());
|
if (CollUtil.isNotEmpty(details)) {
|
List<MesPdProcessParamTemplateRespVO.DetailResp> detailResps = BeanUtils.toBean(
|
details, MesPdProcessParamTemplateRespVO.DetailResp.class, detailResp ->
|
MapUtils.findAndThen(unitMeasureMap, detailResp.getUnitMeasureId(),
|
unitMeasure -> detailResp.setUnitMeasureName(unitMeasure.getName()))
|
);
|
vo.setDetails(detailResps);
|
}
|
});
|
}
|
|
}
|
