From 22594e714c57c9e243fe9973515ea9467d71c2db Mon Sep 17 00:00:00 2001 From: Fixiaobai <fixiaobai@163.com> Date: 星期二, 07 十一月 2023 18:35:11 +0800 Subject: [PATCH] Changes11 --- mes-basic/src/main/java/com/chinaztt/mes/basic/controller/ParamController.java | 88 +++++++++++++++++++++++++++++++++++++++++--- 1 files changed, 82 insertions(+), 6 deletions(-) diff --git a/mes-basic/src/main/java/com/chinaztt/mes/basic/controller/ParamController.java b/mes-basic/src/main/java/com/chinaztt/mes/basic/controller/ParamController.java index 6610d21..c2b8a24 100644 --- a/mes-basic/src/main/java/com/chinaztt/mes/basic/controller/ParamController.java +++ b/mes-basic/src/main/java/com/chinaztt/mes/basic/controller/ParamController.java @@ -17,14 +17,19 @@ package com.chinaztt.mes.basic.controller; -import com.alibaba.excel.EasyExcel; +import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; import com.baomidou.mybatisplus.extension.plugins.pagination.Page; import com.chinaztt.mes.basic.dto.ParamDTO; +import com.chinaztt.mes.basic.dto.ParamPlusDto; import com.chinaztt.mes.basic.entity.Param; -import com.chinaztt.mes.basic.excel.ParamData; -import com.chinaztt.mes.basic.excel.ParamUploadListener; import com.chinaztt.mes.basic.service.ParamService; +import com.chinaztt.mes.basic.util.DictUtils; +import com.chinaztt.mes.common.util.JsonUtil; +import com.chinaztt.mes.common.util.StringUtils; +import com.chinaztt.mes.common.util.easyexcel.EasyExcelUtils; +import com.chinaztt.mes.common.util.easyexcel.template.ParamTemplate; import com.chinaztt.mes.common.wrapper.QueryWrapperUtil; +import com.chinaztt.ztt.admin.api.entity.SysDictItem; import com.chinaztt.ztt.common.core.util.R; import com.chinaztt.ztt.common.log.annotation.SysLog; import io.swagger.annotations.Api; @@ -35,6 +40,8 @@ import org.springframework.web.multipart.MultipartFile; import java.io.IOException; +import java.util.*; +import java.util.stream.Collectors; /** @@ -51,6 +58,7 @@ private final ParamService paramService; + private DictUtils dictUtils; /** * 鍒嗛〉鏌ヨ * @@ -62,7 +70,72 @@ @GetMapping("/page") @PreAuthorize("@pms.hasPermission('technology_param_view','product_workbench')") public R getParamPage(Page page, ParamDTO param) { - return R.ok(paramService.page(page, QueryWrapperUtil.gen(param))); + List<Map<String,Object>> all = new ArrayList<>(); + List<SysDictItem> dict = dictUtils.getDict("technology_param"); + QueryWrapper<Param> gen = QueryWrapperUtil.gen(param); + gen.lambda().eq(Param::getParentId,0L); + Page page1 = paramService.page(page, gen); + List records = page1.getRecords(); + records.forEach(l->{ + System.out.println("涓�绾ф暟鎹�--"+l); + Map pojo = JsonUtil.jsonToPojo(JsonUtil.jsonToString(l), Map.class); + Optional.ofNullable(pojo.get("type")).ifPresent(o->{ + pojo.put("type",dict.stream().filter(d-> Objects.equals(String.valueOf(d.getValue()),pojo.get("type"))).collect(Collectors.toList()).get(0).getLabel()); + }); + QueryWrapper<Param>queryWrapper=new QueryWrapper<>(); + queryWrapper.lambda().eq(Param::getParentId,pojo.get("id")); + List<Param> secondList = paramService.list(queryWrapper); + //娌℃湁children鐩存帴娣诲姞 + if(secondList.isEmpty()){ + Map<String,Object> oneMap = new HashMap<>(); + oneMap.put("code",pojo.get("code")); + oneMap.put("paramItem",pojo.get("parameterItem")); + oneMap.put("paramItemTwo",""); + oneMap.put("paramItemThree",""); + oneMap.put("oneId",pojo.get("id")); + oneMap.put("paramType",StringUtils.defaultString((String) pojo.get("type"),"")); + oneMap.put("paramFormat",StringUtils.defaultString((String) pojo.get("parameterFormat"),"")); + oneMap.put("unit",StringUtils.defaultString((String) pojo.get("unit"),"")); + all.add(oneMap); + } + secondList.forEach(s->{ + System.out.println("浜岀骇鏁版嵁--"+s); + Map second = JsonUtil.jsonToPojo(JsonUtil.jsonToString(s), Map.class); + Optional.ofNullable(second.get("type")).ifPresent(o->{ + second.put("type",dict.stream().filter(d-> Objects.equals(String.valueOf(d.getValue()),second.get("type"))).collect(Collectors.toList()).get(0).getLabel()); + }); + QueryWrapper<Param>queryWrapperSecond=new QueryWrapper<>(); + queryWrapperSecond.lambda().eq(Param::getParentId,second.get("id")); + List<Param> third = paramService.getBaseMapper().selectList(queryWrapperSecond); + if(third.isEmpty()){ + Param p = new Param(); + p.setType(s.getType()); + p.setParameterFormat(s.getParameterFormat()); + p.setUnit(s.getUnit()); + third.add(p); + } + third.forEach(t->{ + System.out.println("涓夌骇鏁版嵁--"+t); + Optional.ofNullable(t.getType()).ifPresent(o->{ + t.setType(dict.stream().filter(d-> Objects.equals(String.valueOf(d.getValue()),t.getType())).collect(Collectors.toList()).get(0).getLabel()); + }); + Map<String,Object> allMap = new HashMap<>(); + allMap.put("code",pojo.get("code")); + allMap.put("paramItem",StringUtils.defaultString((String) pojo.get("parameterItem"),"")); + allMap.put("oneId",pojo.get("id")); + allMap.put("twoId",second.get("id")); + allMap.put("threeId",t.getId()); + allMap.put("paramItemTwo",StringUtils.defaultString((String) second.get("parameterItem"),"")); + allMap.put("paramItemThree",StringUtils.defaultString(t.getParameterItem(),"")); + allMap.put("paramType",StringUtils.defaultString(t.getType(),"")); + allMap.put("paramFormat",StringUtils.defaultString(t.getParameterFormat(),"")); + allMap.put("unit", StringUtils.defaultString(t.getUnit(),"")); + all.add(allMap); + }); + }); + }); + page1.setRecords(all); + return R.ok(page1); } /** @@ -104,7 +177,7 @@ @SysLog("鏂板宸ュ簭鍙傛暟琛�") @PostMapping @PreAuthorize("@pms.hasPermission('technology_param_add')") - public R save(@RequestBody Param param) { + public R save(@RequestBody ParamPlusDto param) { return R.ok(paramService.fullSave(param)); } @@ -142,9 +215,12 @@ * @return */ @PostMapping("/upload") + @ApiOperation(value = "瀵煎叆", notes = "瀵煎叆") public R upload(@RequestParam("file") MultipartFile file) { try { - EasyExcel.read(file.getInputStream(), ParamData.class, new ParamUploadListener(paramService)).sheet().doRead(); +// EasyExcel.read(file.getInputStream(), ParamTemplate.class, new ExcelListener<ParamTemplate>()).sheet().doRead(); + List<ParamTemplate> paramTemplates = EasyExcelUtils.readModelExcel(ParamTemplate.class, file); + paramService.importExcelPlus(paramTemplates); } catch (IOException e) { e.printStackTrace(); } -- Gitblit v1.9.3