楚江erp：样品反馈单导出功能开发

yeqianyong
1 parent c9baf587
Showing 1 changed file with 100 additions and 0 deletions
xingyun-sc/src/main/java/com/lframework/xingyun/sc/controller/sample/SampleResultFeedbackController.java
--- a/xingyun-sc/src/main/java/com/lframework/xingyun/sc/controller/sample/SampleResultFeedbackController.java
View file @2d757cc
+++ b/xingyun-sc/src/main/java/com/lframework/xingyun/sc/controller/sample/SampleResultFeedbackController.java
View file @2d757cc
 package com.lframework.xingyun.sc.controller.sample;
+import cn.hutool.core.io.resource.ClassPathResource;
 import com.lframework.xingyun.sc.bo.sample.feedback.GetSampleResultFeedbackBo;
 import com.lframework.xingyun.sc.bo.sample.feedback.QuerySampleResultFeedbackBo;
+import com.lframework.xingyun.sc.entity.SampleFeedbackTrackingDetail;
+import com.lframework.xingyun.sc.utils.ExcelUtil;
+import com.lframework.xingyun.sc.utils.LatexFormulaExcelExporterUtil;
+import com.lframework.xingyun.sc.utils.ResponseUtil;
 import com.lframework.xingyun.sc.vo.sample.feedback.QuerySampleResultFeedbackVo;
 import com.lframework.xingyun.sc.service.sample.SampleResultFeedbackService;
 import com.lframework.xingyun.sc.entity.SampleResultFeedback;
@@ -10,6 +15,7 @@ import com.lframework.starter.web.core.components.resp.PageResult;
 import com.lframework.starter.web.core.components.resp.InvokeResult;
 import javax.annotation.Resource;
+import javax.servlet.http.HttpServletResponse;
 import javax.validation.constraints.NotBlank;
 import io.swagger.annotations.ApiImplicitParam;
 import com.lframework.starter.web.core.components.resp.InvokeResultBuilder;
@@ -17,6 +23,11 @@ import com.lframework.starter.common.exceptions.impl.DefaultClientException;
 import io.swagger.annotations.ApiOperation;
 import com.lframework.starter.common.utils.CollectionUtil;
 import io.swagger.annotations.Api;
+import lombok.extern.slf4j.Slf4j;
+import org.apache.commons.lang3.StringUtils;
+import org.apache.poi.ss.usermodel.Sheet;
+import org.apache.poi.ss.usermodel.Workbook;
+import org.apache.poi.xssf.usermodel.XSSFWorkbook;
 import org.springframework.web.bind.annotation.DeleteMapping;
 import com.lframework.starter.web.core.controller.DefaultBaseController;
 import com.lframework.starter.web.core.annotations.security.HasPermission;
@@ -24,7 +35,12 @@ import org.springframework.validation.annotation.Validated;
 import org.springframework.web.bind.annotation.*;
 import javax.validation.Valid;
+import java.io.InputStream;
+import java.time.format.DateTimeFormatter;
+import java.util.ArrayList;
+import java.util.HashMap;
 import java.util.List;
+import java.util.Map;
 import java.util.stream.Collectors;
 /**
@@ -32,6 +48,7 @@ import java.util.stream.Collectors;
  *
  */
 @Api(tags = "产品试样结果反馈单")
+@Slf4j
 @Validated
 @RestController
 @RequestMapping("/sample/feedback")
@@ -86,4 +103,87 @@ public class SampleResultFeedbackController extends DefaultBaseController {
         sampleResultFeedbackService.deleteById(id);
         return InvokeResultBuilder.success();
     }
+
+
+    /**
+     * 导出反馈单数据
+     * 支持PDF、EXCEL俩种格式
+     * @param id            主键ID
+     * @param response      响应体
+     */
+    @ApiOperation("规格变更打印")
+    @GetMapping("/export")
+    public void export(@NotBlank(message = "id不能为空") String id, @NotBlank(message = "导出类型不能为空") String exportType
+            , HttpServletResponse response) {
+        SampleResultFeedback data = sampleResultFeedbackService.findById(id);
+        if (data == null) {
+            throw new DefaultClientException("样品反馈单数据不存在！");
+        }
+        ClassPathResource templateResource = new ClassPathResource("templates/sampleFeedbackTemplate.xlsx");
+        try (InputStream inputStream = templateResource.getStream();
+             Workbook workbook = new XSSFWorkbook(inputStream)) {
+            Sheet sheet = workbook.getSheetAt(0);
+            ResponseUtil.setExcelResponseHead(response, data.getOrderNo() + "-试样结果反馈单.xlsx");
+            DateTimeFormatter dateFormatter = DateTimeFormatter.ofPattern("yyyy-MM-dd");
+            List<SampleFeedbackTrackingDetail> detailList = data.getDetailList();
+            // ========== 起始行：5==========
+            int startRow = 4;
+            for (int i = startRow + 1; i < startRow + detailList.size(); i++) {
+                ExcelUtil.copyRow(workbook, sheet, startRow, i);
+            }
+            for (int i = 0; i < detailList.size(); i++) {
+                SampleFeedbackTrackingDetail detail = detailList.get(i);
+                int rowIdx = startRow + i;
+                ExcelUtil.setCellValue(sheet, rowIdx, 1, detail.getBrand());
+                List<LatexFormulaExcelExporterUtil.FormulaComponent> formulaComponentList = new ArrayList<>(3);
+                if (detail.getThickness() != null) {
+                    LatexFormulaExcelExporterUtil.FormulaComponent formulaComponent = new LatexFormulaExcelExporterUtil.FormulaComponent();
+                    formulaComponent.setBase(detail.getThickness());
+                    formulaComponent.setSup(detail.getThicknessTolPos());
+                    formulaComponent.setSub(detail.getThicknessTolNeg());
+                    formulaComponentList.add(formulaComponent);
+                }
+                if (detail.getWidth() != null) {
+                    LatexFormulaExcelExporterUtil.FormulaComponent formulaComponent = new LatexFormulaExcelExporterUtil.FormulaComponent();
+                    formulaComponent.setBase(detail.getWidth());
+                    formulaComponent.setSup(detail.getWidthTolPos());
+                    formulaComponent.setSub(detail.getWidthTolNeg());
+                    formulaComponentList.add(formulaComponent);
+                }
+                if (detail.getLength() != null) {
+                    LatexFormulaExcelExporterUtil.FormulaComponent formulaComponent = new LatexFormulaExcelExporterUtil.FormulaComponent();
+                    formulaComponent.setBase(detail.getLength());
+                    formulaComponent.setSup(detail.getLengthTolPos());
+                    formulaComponent.setSub(detail.getLengthTolNeg());
+                    formulaComponentList.add(formulaComponent);
+                }
+                String latex = LatexFormulaExcelExporterUtil.convertToLatex(formulaComponentList);
+                if (StringUtils.isNotBlank(latex)) {
+                    LatexFormulaExcelExporterUtil.insertLatexImageToCell(workbook, sheet, latex, rowIdx, 2, 1, 2);
+                }
+                ExcelUtil.setCellValue(sheet, rowIdx, 4, detail.getStatus());
+                ExcelUtil.setCellValue(sheet, rowIdx, 5, detail.getShipmentDate());
+                ExcelUtil.setCellValue(sheet, rowIdx, 6, detail.getQuantity());
+                ExcelUtil.setCellValue(sheet, rowIdx, 7, detail.getYieldBatchNo());
+            }
+            // ========== 填充顶部固定字段 ==========
+            Map<String, Object> dataMap = new HashMap<>();
+            dataMap.put("customerName", data.getCustomerName());
+            dataMap.put("workshopName", data.getWorkshopName());
+            dataMap.put("orderNo", data.getOrderNo());
+
+            ExcelUtil.processTemplate(workbook, dataMap);
+            if ("PDF".equals(exportType)) {
+                // 输出为PDF文档
+            } else {
+                // 输出
+                workbook.write(response.getOutputStream());
+                response.getOutputStream().flush();
+            }
+        } catch (Exception e) {
+            log.error("导出反馈单导出失败：", e);
+            throw new DefaultClientException("导出反馈单导出失败：" + e.getMessage());
+        }
+    }
+
 }