From 3c34085a21511610477f555cebf3f6fa42272402 Mon Sep 17 00:00:00 2001 From: zss <zss@example.com> Date: 星期二, 07 十一月 2023 18:32:53 +0800 Subject: [PATCH] 产品结构导入数据 --- mes-basic/src/main/java/com/chinaztt/mes/basic/controller/ParamController.java | 58 +++++++++++++++++++++++++++++++++++++++++++++++----------- 1 files changed, 47 insertions(+), 11 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 5c3169a..eec09c0 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 @@ -28,6 +28,10 @@ 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.ExcelListener; +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; @@ -39,6 +43,7 @@ import org.springframework.web.bind.annotation.*; import org.springframework.web.multipart.MultipartFile; +import javax.servlet.http.HttpServletResponse; import java.io.IOException; import java.util.*; import java.util.stream.Collectors; @@ -70,12 +75,14 @@ @GetMapping("/page") @PreAuthorize("@pms.hasPermission('technology_param_view','product_workbench')") public R getParamPage(Page page, ParamDTO param) { + List<Map<String,Object>> all = new ArrayList<>(); List<SysDictItem> dict = dictUtils.getDict("technology_param"); - System.out.println(dict); - Page page1 = paramService.page(page, QueryWrapperUtil.gen(param)); + QueryWrapper<Param> gen = QueryWrapperUtil.gen(param); + gen.lambda().eq(Param::getParentId,0L); + Page page1 = paramService.page(page, gen); List records = page1.getRecords(); - List<Map>all = new ArrayList<Map>(); 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()); @@ -83,27 +90,54 @@ QueryWrapper<Param>queryWrapper=new QueryWrapper<>(); queryWrapper.lambda().eq(Param::getParentId,pojo.get("id")); List<Param> secondList = paramService.list(queryWrapper); - List<Map>secondMapList = new ArrayList<>(); + //娌℃湁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()); }); - System.out.println(second); 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); }); - System.out.println(third); - second.put("children",third); - secondMapList.add(second); }); - pojo.put("children",secondMapList); - all.add(pojo); }); page1.setRecords(all); return R.ok(page1); @@ -186,9 +220,11 @@ * @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(); + EasyExcelUtils.readModelExcel(ParamTemplate.class,file); } catch (IOException e) { e.printStackTrace(); } -- Gitblit v1.9.3