From f4cfca6f2c8a5cb4e8e0fdc00d76a710283d6b06 Mon Sep 17 00:00:00 2001
From: 李林 <z1292839451@163.com>
Date: 星期五, 12 四月 2024 18:05:48 +0800
Subject: [PATCH] 报告生成功能

---
 inspect-server/src/main/java/com/yuanchu/mom/service/impl/InsOrderPlanServiceImpl.java |  313 +++++++++++++++++++++++++++++++++++++++++++---------
 1 files changed, 258 insertions(+), 55 deletions(-)

diff --git a/inspect-server/src/main/java/com/yuanchu/mom/service/impl/InsOrderPlanServiceImpl.java b/inspect-server/src/main/java/com/yuanchu/mom/service/impl/InsOrderPlanServiceImpl.java
index 78981e5..36c8f4a 100644
--- a/inspect-server/src/main/java/com/yuanchu/mom/service/impl/InsOrderPlanServiceImpl.java
+++ b/inspect-server/src/main/java/com/yuanchu/mom/service/impl/InsOrderPlanServiceImpl.java
@@ -13,6 +13,9 @@
 import com.deepoove.poi.XWPFTemplate;
 import com.deepoove.poi.config.Configure;
 import com.deepoove.poi.config.ConfigureBuilder;
+import com.deepoove.poi.data.*;
+import com.deepoove.poi.data.style.Style;
+import com.deepoove.poi.data.style.TableStyle;
 import com.yuanchu.mom.common.GetLook;
 import com.yuanchu.mom.common.PrintChina;
 import com.yuanchu.mom.dto.InsOrderPlanDTO;
@@ -26,7 +29,8 @@
 import com.yuanchu.mom.utils.GiveCode;
 import com.yuanchu.mom.utils.QueryWrappers;
 import com.yuanchu.mom.vo.InsOrderPlanVO;
-import lombok.AllArgsConstructor;
+import org.apache.poi.xwpf.usermodel.TableRowAlign;
+import org.apache.poi.xwpf.usermodel.XWPFTable;
 import org.springframework.beans.factory.annotation.Value;
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
@@ -34,14 +38,17 @@
 
 import javax.annotation.Resource;
 import java.io.FileNotFoundException;
-import java.io.FileOutputStream;
 import java.io.IOException;
+import java.nio.file.Files;
+import java.nio.file.Paths;
 import java.time.LocalDateTime;
 import java.time.format.DateTimeFormatter;
 import java.util.*;
+import java.util.concurrent.atomic.AtomicReference;
 import java.util.stream.Collectors;
 import java.util.concurrent.CompletableFuture;
 import java.util.concurrent.atomic.AtomicInteger;
+import java.util.stream.Stream;
 
 /**
  * 妫�楠屼换鍔�-涓氬姟瀹炵幇灞�
@@ -76,6 +83,7 @@
     @Resource
     private InsReportMapper insReportMapper;
 
+    @Resource
     private InsProductResultMapper insProductResultMapper;
 
     @Override
@@ -120,50 +128,7 @@
         List<SampleProductDto> list = JSON.parseArray(JSON.toJSONString(map.get("sampleProduct")), SampleProductDto.class);
         for (SampleProductDto samples : list) {
             if (BeanUtil.isEmpty(samples.getInsProduct())) continue;
-            for (InsProduct product : samples.getInsProduct()) {
-                if (product.getTemplateId() == null) {
-                    product.setTemplate(new ArrayList<>());
-                    continue;
-                }
-                if (set.add(product.getTemplateId())) {
-                    map2.put(product.getTemplateId(), standardTemplateService.getStandTempThingById(product.getTemplateId()) + "");
-                }
-                String thing = map2.get(product.getTemplateId());
-                if (StrUtil.isNotEmpty(thing)) {
-                    JSONObject sheet = JSON.parseObject(JSON.toJSONString(JSON.parseArray(JSON.toJSONString(JSON.parseObject(thing).get("data"))).get(0)));
-                    JSONObject config = JSON.parseObject(JSON.toJSONString(sheet.get("config")));
-                    List<JSONObject> cellData = JSON.parseArray(JSON.toJSONString(sheet.get("celldata")), JSONObject.class);
-                    List<JSONObject> tableData = new ArrayList<>();
-                    Map<String, Object> style = new HashMap<>();
-//                    style.put("borderInfo", config.get("borderInfo"));
-                    style.put("rowlen", config.get("rowlen"));
-                    style.put("columnlen", config.get("columnlen"));
-                    int r = -1;
-                    if (BeanUtil.isNotEmpty(cellData)) {
-                        for (JSONObject o : cellData) {
-                            JSONObject v = JSON.parseObject(JSON.toJSONString(o.get("v")));
-                            if (BeanUtil.isNotEmpty(v.get("ps"))) {
-                                JSONObject ps = JSON.parseObject(JSON.toJSONString(v.get("ps")));
-                                if (ps.get("value").equals("妫�楠岄」") && v.get("v").equals(product.getInspectionItem())) {
-                                    tableData.add(o);
-                                    r = Integer.parseInt(o.get("r") + "");
-                                    continue;
-                                }
-                            }
-                            if (tableData.size() > 0) {
-                                if ((Integer.parseInt(o.get("r") + "") == r + 1 || Integer.parseInt(o.get("r") + "") == r) && !(o.get("v") + "").equals("{\"ht\":\"0\"}")) {
-                                    r = Integer.parseInt(o.get("r") + "");
-                                    tableData.add(o);
-                                } else {
-                                    break;
-                                }
-                            }
-                        }
-                    }
-                    product.setTemplate(tableData);
-                    product.setStyle(style);
-                }
-            }
+            getTemplateThing(set, map2, samples);
         }
         map.put("sampleProduct", list);
         return map;
@@ -190,9 +155,9 @@
                     JSONObject insValue = JSON.parseObject(JSON.toJSONString(o));
                     Map<String, Object> map = new HashMap<>();
                     map.put("v", JSON.parseObject(JSON.toJSONString(insValue.get("v"))).get("v"));
-                    if (insValue.get("u") == null && StrUtil.isNotEmpty(JSON.parseObject(JSON.toJSONString(insValue.get("v"))).get("v").toString())) {
+                    if ((insValue.get("u") == null || insValue.get("u").equals("")) && StrUtil.isNotEmpty(JSON.parseObject(JSON.toJSONString(insValue.get("v"))).get("v").toString())) {
                         map.put("u", userId);
-                    }else {
+                    } else {
                         map.put("u", insValue.get("u"));
                     }
                     iv.add(map);
@@ -212,7 +177,7 @@
                 JSONObject equipValue = JSON.parseObject(JSON.toJSONString(jo.get("equipValue")));
                 try {
                     result.setEquipValue(JSON.parseObject(JSON.toJSONString(equipValue.get("v"))).get("v") + "");
-                }catch (Exception ignored){
+                } catch (Exception ignored) {
                     result.setEquipValue(null);
                 }
                 JSONObject insResult = JSON.parseObject(JSON.toJSONString(jo.get("insResult")));
@@ -228,7 +193,7 @@
                 }
                 insProduct.setUpdateUser(userId);
                 insProductMapper.updateById(insProduct);
-                if(count.get() == 0){
+                if (count.get() == 0) {
                     InsSample insSample = insSampleMapper.selectById(insProductMapper.selectById(insProduct.getId()).getInsSampleId());
                     insSample.setInsState(1);
                     count.addAndGet(1);
@@ -262,34 +227,268 @@
             InsOrder insOrder = insOrderMapper.selectById(orderId);
             Map<String, String> user = insProductMapper.selectUserById(insOrder.getUserId());
             List<SampleProductDto> samples = insSampleMapper.selectSampleProductListByOrderId(orderId);
-            samples.forEach(System.out::println);
             InsReport insReport = new InsReport();
             insReport.setCode(giveCode.giveCode("JCZX/TX-", "ins_report", "-", "yyMMdd"));
             insReport.setInsOrderId(orderId);
+            Set<Integer> set = new HashSet<>();
+            Map<Integer, String> map2 = new HashMap<>();
+            List<Map<String, Object>> tables = new ArrayList<>();
+            Set<String> standardMethod = new HashSet<>();
+            Set<String> deviceSet = new HashSet<>();
+            samples.forEach(a -> {
+                Set<String> templateSet = new HashSet<>();
+                getTemplateThing(set, map2, a);
+                a.getInsProduct().forEach(b->{
+                    String[] split = b.getMethod().replaceAll("]", "").replaceAll("\\[","").replaceAll("\"","").split(",");
+                    standardMethod.addAll(Arrays.asList(split));
+                    if(b.getInsProductResult()!=null){
+                        if(b.getInsProductResult().getEquipValue()!=null&& !Objects.equals(b.getInsProductResult().getEquipValue(), "")){
+                            deviceSet.add(b.getInsProductResult().getEquipValue());
+                        }
+                    }
+                    if (b.getTemplate().size() == 0) {
+                        return;
+                    }
+                    templateSet.add(JSON.toJSONString(b.getTemplate()));
+                });
+                AtomicInteger index = new AtomicInteger();
+                templateSet.forEach(tem->{
+                    Set<Integer> set2 = new HashSet<>();
+                    List<RowRenderData> rows = new ArrayList<>();
+                    List<TextRenderData> text = new ArrayList<>();
+                    RowRenderData rowRenderData;
+                    Set<String> delSet = new HashSet<>();
+                    List<JSONObject> temp = JSON.parseArray(tem, JSONObject.class);
+                    Map<String, InsProduct> pMap = new HashMap<>();
+                    Set<String> delRSet = new HashSet<>();
+                    for (JSONObject jo1 : temp) {
+                        JSONObject v = JSON.parseObject(JSON.toJSONString(jo1.get("v")));
+                        if (v.get("ps") != null) {
+                            if ("妫�楠屽��".equals(JSON.parseObject(JSON.toJSONString(v.get("ps"))).get("value") + "") || "璁惧".equals(JSON.parseObject(JSON.toJSONString(v.get("ps"))).get("value") + "")) {
+                                delSet.add(jo1.get("c") + "");
+                            }
+                            if ("妫�楠屽瓙椤�".equals(JSON.parseObject(JSON.toJSONString(v.get("ps"))).get("value") + "")) {
+                                int count2 = 0;
+                                for (InsProduct p : a.getInsProduct()) {
+                                    if(v.get("v").equals(p.getInspectionItemSubclass())){
+                                        pMap.put(jo1.get("r")+"", p);
+                                        break;
+                                    }else{
+                                        count2++;
+                                    }
+                                }
+                                if(count2 == a.getInsProduct().size()){
+                                    delRSet.add(jo1.get("r")+"");
+                                }
+                            }
+                        }
+                    }
+                    for (int i = 0; i < temp.size(); i++) {
+                        JSONObject jo1 = temp.get(i);
+                        if (delRSet.stream().anyMatch(e -> e.equals(jo1.get("r") + ""))) {
+                            continue;
+                        }
+                        if (delSet.stream().anyMatch(e -> e.equals(jo1.get("c") + ""))) {
+                            continue;
+                        }
+                        TextRenderData textRenderData = new TextRenderData();
+                        if (set2.add(Integer.parseInt(jo1.get("r") + ""))) {
+                            if (text.size() > 0) {
+                                TextRenderData[] text2 = text.toArray(new TextRenderData[0]);
+                                rowRenderData = Rows.of(text2).center().rowAtleastHeight(1).create();
+                                rows.add(rowRenderData);
+                                text = new ArrayList<>();
+                            }
+                        }
+                        JSONObject v = JSON.parseObject(JSON.toJSONString(jo1.get("v")));
+                        InsProduct p = pMap.get(jo1.get("r") + "");
+                        if (p!=null && v.get("ps") != null) {
+                            String value = JSON.parseObject(JSON.toJSONString(v.get("ps"))).get("value") + "";
+                            if (value.equals("瑕佹眰鍊�")) {
+                                textRenderData.setText(p.getAsk());
+                            } else if (value.equals("鍗曚綅")) {
+                                textRenderData.setText(p.getUnit());
+                            } else if (value.equals("缁撹")) {
+                                switch (p.getInsResult()) {
+                                    case 1:
+                                        textRenderData.setText("鈭�");
+                                        break;
+                                    case 0:
+                                        textRenderData.setText("脳");
+                                        break;
+                                    default:
+                                        textRenderData.setText("-");
+                                        break;
+                                }
+                            } else if (value.equals("搴忓彿")) {
+                                index.getAndIncrement();
+                                textRenderData.setText(index + "");
+                            } else if (value.equals("璁$畻鍊�")) {
+                                JSONArray jsonArray = JSON.parseArray(p.getInsProductResult().getComValue());
+                                textRenderData.setText((JSON.parseObject(JSON.toJSONString(jsonArray.get(0))).get("v") + "").equals("") ? "/" : JSON.parseObject(JSON.toJSONString(jsonArray.get(0))).get("v") + "");
+                            } else if (value.equals("鏈�缁堝��")) {
+                                textRenderData.setText(p.getLastValue());
+                            } else if (value.equals("鏍峰搧缂栧彿")) {
+                                textRenderData.setText(a.getSampleCode());
+                            } else if (value.equals("鏍峰搧鍨嬪彿")) {
+                                textRenderData.setText(a.getModel());
+                            } else if (value.equals("鏍峰搧鍚嶇О")) {
+                                textRenderData.setText(a.getSample());
+                            } else {
+                                textRenderData.setText(v.get("v") == null ? "" : v.get("v") + "");
+                            }
+                        } else if (p==null && v.get("ps") != null) {
+                            String value = JSON.parseObject(JSON.toJSONString(v.get("ps"))).get("value") + "";
+                            p = pMap.get(pMap.keySet().iterator().next());
+                            if (value.equals("鏈�缁堝��")) {
+                                textRenderData.setText(p.getLastValue());
+                            } else if (value.equals("缁撹")) {
+                                switch (p.getInsResult()) {
+                                    case 1:
+                                        textRenderData.setText("鉁�");
+                                        break;
+                                    case 0:
+                                        textRenderData.setText("鉁�");
+                                        break;
+                                    default:
+                                        textRenderData.setText("-");
+                                        break;
+                                }
+                            } else if (value.equals("鏍峰搧缂栧彿")) {
+                                textRenderData.setText(a.getSampleCode());
+                            } else if (value.equals("鏍峰搧鍨嬪彿")) {
+                                textRenderData.setText(a.getModel());
+                            } else if (value.equals("鏍峰搧鍚嶇О")) {
+                                textRenderData.setText(a.getSample());
+                            } else {
+                                textRenderData.setText(v.get("v") == null ? "" : v.get("v") + "");
+                            }
+                        } else {
+                            textRenderData.setText(v.get("v") == null ? "" : v.get("v") + "");
+                        }
+                        Style style = new Style();
+                        style.setFontFamily(v.get("ff") == null ? "寰蒋闆呴粦" : v.get("ff") + "");
+//                        style.setFontSize(v.get("fs")==null?14:Integer.parseInt(v.get("fs")+""));
+                        style.setColor(v.get("fc") == null ? "000000" : (v.get("fc") + "").replace("#", ""));
+                        textRenderData.setStyle(style);
+                        text.add(textRenderData);
+                        if (i == temp.size() - 1) {
+                            TextRenderData[] text2 = text.toArray(new TextRenderData[0]);
+                            rowRenderData = Rows.of(text2).rowAtleastHeight(1).center().create();
+                            rows.add(rowRenderData);
+                            text = new ArrayList<>();
+                        }
+                    }
+                    TableRenderData tableRenderData = new TableRenderData();
+                    tableRenderData.setRows(rows);
+                    TableStyle tableStyle = new TableStyle();
+                    /*JSONObject styleJo = JSON.parseObject(JSON.toJSONString(b.getStyle().get("columnlen")));
+                    List<Integer> colLen = new ArrayList<>();
+                    int colSize = rows.get(0).getCells().size();
+                    int colOne = Integer.parseInt(b.getTemplate().get(0).get("r") + "");
+                    for (int i = colOne; i < colSize + colOne; i++) {
+                        if (styleJo.get(i + "") == null) {
+                            colLen.add(pxToCm(100));
+                        } else {
+                            int i1 = Integer.parseInt(styleJo.get(i + "") + "");
+                            colLen.add(pxToCm(i1));
+                        }
+                    }*/
+//                    tableStyle.setColWidths(ArrayUtils.toPrimitive(colLen.toArray(new Integer[0])));
+                    tableStyle.setAlign(TableRowAlign.CENTER);
+                    tableStyle.setWidth(XWPFTable.DEFAULT_PERCENTAGE_WIDTH);
+                    tableRenderData.setTableStyle(tableStyle);
+                    Map<String, Object> table = new HashMap<>();
+                    table.put("table", tableRenderData);
+                    table.put("report", insReport);
+                    tables.add(table);
+                });
+            });
             String url;
             try {
                 url = ResourceUtils.getURL("classpath:static/report-template.docx").getPath();
             } catch (FileNotFoundException e) {
                 throw new ErrorException("鎵句笉鍒版ā鏉挎枃浠�");
             }
-            XWPFTemplate template = XWPFTemplate.compile(url).render(
+            StringBuilder standardMethod2 = new StringBuilder();
+            for (String s : standardMethod) {
+                standardMethod2.append("銆�").append(s);
+            }
+            standardMethod2.replace(0, 1,"");
+            tables.forEach(table->{
+                table.put("tableSize", tables.size() + 1);
+            });
+            List<Map<String, String>> deviceList = insOrderMapper.selectDeviceList(deviceSet);
+            ConfigureBuilder builder = Configure.builder();
+            builder.useSpringEL(true);
+            XWPFTemplate template = XWPFTemplate.compile(url, builder.build()).render(
                     new HashMap<String, Object>() {{
                         put("order", insOrder);
                         put("report", insReport);
                         put("user", user);
-                        put("samples", samples);
+//                        put("samples", samples);
                         put("sampleSize", samples.size() + samples.get(0).getUnit());
+                        put("tables", tables);
+                        put("tableSize", tables.size() + 1);
+                        put("standardMethod", standardMethod2);
+                        put("deviceList", deviceList);
                     }});
             try {
                 String name = LocalDateTime.now().format(DateTimeFormatter.ofPattern("yy_MM_dd_HH_mm_ss")) + ".docx";
-                template.writeAndClose(new FileOutputStream(wordUrl + "/" + name));
-                insReport.setUrl("/word/"+ name);
+                template.writeAndClose(Files.newOutputStream(Paths.get(wordUrl + "/" + name)));
+                insReport.setUrl("/word/" + name);
                 insReportMapper.insert(insReport);
             } catch (IOException e) {
                 throw new RuntimeException(e);
             }
         }
         return 1;
+    }
+
+    private void getTemplateThing(Set<Integer> set, Map<Integer, String> map2, SampleProductDto a) {
+        for (InsProduct product : a.getInsProduct()) {
+            if (product.getTemplateId() == null) {
+                product.setTemplate(new ArrayList<>());
+                continue;
+            }
+            if (set.add(product.getTemplateId())) {
+                map2.put(product.getTemplateId(), standardTemplateService.getStandTempThingById(product.getTemplateId()) + "");
+            }
+            String thing = map2.get(product.getTemplateId());
+            if (StrUtil.isNotEmpty(thing)) {
+                JSONObject sheet = JSON.parseObject(JSON.toJSONString(JSON.parseArray(JSON.toJSONString(JSON.parseObject(thing).get("data"))).get(0)));
+                JSONObject config = JSON.parseObject(JSON.toJSONString(sheet.get("config")));
+                List<JSONObject> cellData = JSON.parseArray(JSON.toJSONString(sheet.get("celldata")), JSONObject.class);
+                List<JSONObject> tableData = new ArrayList<>();
+                Map<String, Object> style = new HashMap<>();
+                style.put("rowlen", config.get("rowlen"));
+                style.put("columnlen", config.get("columnlen"));
+                int r = -1;
+                if (BeanUtil.isNotEmpty(cellData)) {
+                    for (JSONObject o : cellData) {
+                        JSONObject v = JSON.parseObject(JSON.toJSONString(o.get("v")));
+                        if (BeanUtil.isNotEmpty(v.get("ps"))) {
+                            JSONObject ps = JSON.parseObject(JSON.toJSONString(v.get("ps")));
+                            if (ps.get("value").equals("妫�楠岄」") && v.get("v").equals(product.getInspectionItem())) {
+                                tableData.add(o);
+                                r = Integer.parseInt(o.get("r") + "");
+                                continue;
+                            }
+                        }
+                        if (tableData.size() > 0) {
+                            if ((Integer.parseInt(o.get("r") + "") == r + 1 || Integer.parseInt(o.get("r") + "") == r) && !(o.get("v") + "").equals("{\"ht\":\"0\"}")) {
+                                r = Integer.parseInt(o.get("r") + "");
+                                tableData.add(o);
+                            } else {
+                                break;
+                            }
+                        }
+                    }
+                }
+                product.setTemplate(tableData);
+                product.setStyle(style);
+            }
+        }
     }
 
     @Override
@@ -303,4 +502,8 @@
         insOrderStateMapper.update(null, Wrappers.<InsOrderState>lambdaUpdate().eq(InsOrderState::getInsOrderId, orderId).eq(InsOrderState::getLaboratory, laboratory).set(InsOrderState::getInsTime, LocalDateTime.now()).set(InsOrderState::getInsState, 3));
         return 1;
     }
+
+    public int pxToCm(int px) {
+        return px / 9;
+    }
 }

--
Gitblit v1.9.3