楚江erp：订货单批量导出

@@ -8,6 +8,7 @@ import com.lframework.starter.bpm.mappers.FlowTaskWrapperMapper;
 import com.lframework.starter.bpm.vo.flow.task.QueryTodoTaskListVo;
 import com.lframework.starter.common.exceptions.impl.DefaultClientException;
 import com.lframework.starter.common.utils.CollectionUtil;
+import com.lframework.starter.mq.core.utils.ExportTaskUtil;
 import com.lframework.starter.web.core.annotations.security.HasPermission;
 import com.lframework.starter.web.core.components.resp.InvokeResult;
 import com.lframework.starter.web.core.components.resp.InvokeResultBuilder;
@@ -22,6 +23,8 @@ import com.lframework.xingyun.sc.bo.order.GetPurchaseOrderInfoBo;
 import com.lframework.xingyun.sc.entity.ContractDistributorStandard;
 import com.lframework.xingyun.sc.entity.PurchaseOrderInfo;
 import com.lframework.xingyun.sc.entity.PurchaseOrderLine;
+import com.lframework.xingyun.sc.enums.ExportType;
+import com.lframework.xingyun.sc.excel.order.OrderInfoExportTaskWorker;
 import com.lframework.xingyun.sc.handlers.TransactorHandler;
 import com.lframework.xingyun.sc.service.contract.ContractDistributorStandardService;
 import com.lframework.xingyun.sc.service.order.PurchaseOrderInfoService;
@@ -515,6 +518,14 @@ public class PurchaseOrderInfoController extends DefaultBaseController {
         }
     }
 
+    @ApiOperation("订货单批量打印")
+    @GetMapping("/batchPrintPurchaseOrder")
+    public InvokeResult<Void> batchPrintPurchaseOrder( QueryPurchaseOrderInfoVo vo) {
+        vo.setExportType(ExportType.ORDER_EXPORT_ZIP.getCode());
+        ExportTaskUtil.exportTask("订货单信息", OrderInfoExportTaskWorker.class, vo);
+        return InvokeResultBuilder.success();
+    }
+
     private File convertExcelToPdf(File excelFile) throws IOException, InterruptedException {
         if (!excelFile.exists()) {
             throw new IllegalArgumentException("Excel 文件不存在: " + excelFile.getAbsolutePath());

@@ -12,6 +12,7 @@ public enum ExportType implements BaseEnum<String> {
     RECEIVABLE_LEDGER_REPORT("RECEIVABLE_LEDGER_REPORT", "应收款台账报表"),
     SHIPMENT_DETAIL_REPORT("SHIPMENT_DETAIL_REPORT", "发货明细报表"),
     ORDER_DETAIL_REPORT("ORDER_DETAIL_REPORT", "订单明细报表"),
+    ORDER_EXPORT_ZIP("ORDER_EXPORT_ZIP", "订货单信息"),
 
 
     ;

+package com.lframework.xingyun.sc.excel.order;
+
+import com.lframework.starter.mq.core.components.export.ExportTaskWorker;
+import com.lframework.starter.web.core.components.excel.ExcelModel;
+import com.lframework.starter.web.core.components.resp.PageResult;
+import com.lframework.starter.web.core.utils.ApplicationUtil;
+import com.lframework.starter.web.core.utils.JsonUtil;
+import com.lframework.xingyun.sc.entity.PurchaseOrderInfo;
+import com.lframework.xingyun.sc.service.order.PurchaseOrderInfoService;
+import com.lframework.xingyun.sc.vo.order.QueryPurchaseOrderInfoVo;
+
+
+public class OrderInfoExportTaskWorker implements
+        ExportTaskWorker<QueryPurchaseOrderInfoVo, PurchaseOrderInfo, ExcelModel> {
+
+
+    @Override
+    public QueryPurchaseOrderInfoVo parseParams(String json) {
+        return JsonUtil.parseObject(json, QueryPurchaseOrderInfoVo.class);
+    }
+
+    @Override
+    public PageResult<PurchaseOrderInfo> getDataList(int pageIndex, int pageSize, QueryPurchaseOrderInfoVo params) {
+        PurchaseOrderInfoService orderInfoService = ApplicationUtil.getBean(PurchaseOrderInfoService.class);
+        return orderInfoService.query(pageIndex, pageSize, params);
+    }
+
+    @Override
+    public ExcelModel exportData(PurchaseOrderInfo purchaseOrderInfo) {
+        return null;
+    }
+
+    @Override
+    public Class<ExcelModel> getModelClass() {
+        return null;
+    }
+}

-package com.lframework.xingyun.sc.excel.ledger;
+package com.lframework.xingyun.sc.handlers;
 
 
 import cn.hutool.core.io.resource.ClassPathResource;
@@ -9,8 +9,14 @@ import com.lframework.starter.web.core.utils.JsonUtil;
 import com.lframework.starter.web.inner.entity.SysDataDicItem;
 import com.lframework.starter.web.inner.service.system.SysDataDicItemService;
 import com.lframework.xingyun.sc.bo.ledger.receipt.ReceiptLedgerReportDetail;
+import com.lframework.xingyun.sc.entity.PurchaseOrderInfo;
+import com.lframework.xingyun.sc.entity.PurchaseOrderLine;
+import com.lframework.xingyun.sc.excel.ledger.LedgerReportExportModel;
+import com.lframework.xingyun.sc.service.order.PurchaseOrderLineService;
 import com.lframework.xingyun.sc.utils.ExcelUtil;
+import com.lframework.xingyun.sc.utils.LatexFormulaExcelExporterUtil;
 import com.lframework.xingyun.sc.vo.ledger.receipt.ReceiptLedgerReportVo;
+import com.lframework.xingyun.sc.vo.order.QueryPurchaseOrderInfoVo;
 import lombok.extern.slf4j.Slf4j;
 import org.apache.commons.collections4.CollectionUtils;
 import org.apache.commons.lang3.StringUtils;
@@ -19,17 +25,29 @@ import org.apache.poi.ss.util.CellRangeAddress;
 import org.apache.poi.xssf.usermodel.XSSFWorkbook;
 import org.springframework.stereotype.Component;
 
-import java.io.File;
-import java.io.FileOutputStream;
-import java.io.InputStream;
+import javax.annotation.Resource;
+import java.io.*;
+import java.math.BigDecimal;
+import java.math.RoundingMode;
+import java.nio.file.Files;
+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;
+import java.util.zip.ZipEntry;
+import java.util.zip.ZipOutputStream;
 
 
 @Slf4j
 @Component
-public class LedgerReportExportHandler implements ExportHandler {
+public class BusinessDataExportHandler implements ExportHandler {
+
+
+    @Resource
+    private PurchaseOrderLineService purchaseOrderLineService;
+
 
     @Override
     public void exportLedgerReport(Object params, Object dataList, File xlsxFile, String exportType) {
@@ -134,6 +152,159 @@ public class LedgerReportExportHandler implements ExportHandler {
         }
     }
 
+    @Override
+    public void exportOrderInfo(Object params, Object dataList, File file) {
+        // 订单信息
+        List<PurchaseOrderInfo> orderInfoList = JsonUtil.parseList(JsonUtil.toJsonString(dataList), PurchaseOrderInfo.class);
+        if (CollectionUtils.isEmpty(orderInfoList)) {
+            return;
+        }
+        List<String> orderIds = orderInfoList.stream().map(PurchaseOrderInfo::getId).collect(Collectors.toList());
+        Map<String, List<PurchaseOrderLine>> orderLineMap = new HashMap<>();
+        // 获取物料行数据
+        List<PurchaseOrderLine> orderLineList = purchaseOrderLineService.listByOrderIds(orderIds, true);
+        if (CollectionUtils.isNotEmpty(orderLineList)) {
+            for (PurchaseOrderLine orderLine: orderLineList) {
+                String orderId = orderLine.getPurchaseOrderId();
+                List<PurchaseOrderLine> list = orderLineMap.computeIfAbsent(orderId, k -> new ArrayList<>());
+                list.add(orderLine);
+            }
+        }
+        // 查询条件
+        QueryPurchaseOrderInfoVo queryVo = JsonUtil.parseObject(JsonUtil.toJsonString(params), QueryPurchaseOrderInfoVo.class);
+        // 获取导出模版
+        ClassPathResource templateResource = new ClassPathResource("templates/purchaseOrderTemplate.xlsx");
+        if ("PRODUCTION_PROCESS".equals(queryVo.getTemplateType())) {
+            templateResource = new ClassPathResource("templates/purchaseOrderTemplateForProduction.xlsx");
+        }
+        try (FileOutputStream fosZip = new FileOutputStream(file);
+             ZipOutputStream zos = new ZipOutputStream(fosZip)) {
+            for (PurchaseOrderInfo orderInfo : orderInfoList) {
+                try (InputStream inputStream = templateResource.getStream();
+                     Workbook workbook = new XSSFWorkbook(inputStream)) {
+                    Sheet sheet = workbook.getSheetAt(0);
+                    DateTimeFormatter dateFormatter = DateTimeFormatter.ofPattern("yyyy-MM-dd");
+                    int startRow = 7;
+                    BigDecimal totalQuantity = BigDecimal.ZERO;
+                    List<PurchaseOrderLine> purchaseOrderLineList = orderLineMap.get(orderInfo.getId());
+                    if (CollectionUtils.isNotEmpty(purchaseOrderLineList)) {
+                        for (int i = startRow + 1; i < startRow + purchaseOrderLineList.size(); i++) {
+                            ExcelUtil.copyRow(workbook, sheet, startRow, i);
+                        }
+                        for (PurchaseOrderLine currentOrderLine : purchaseOrderLineList) {
+                            ExcelUtil.setCellValue(sheet, startRow, 1, currentOrderLine.getIndustry());
+                            ExcelUtil.setCellValue(sheet, startRow, 2, currentOrderLine.getQuality());
+                            ExcelUtil.setCellValue(sheet, startRow, 3, currentOrderLine.getBrand());
+                            List<LatexFormulaExcelExporterUtil.FormulaComponent> formulaComponentList = new ArrayList<>(3);
+                            if (currentOrderLine.getThickness() != null) {
+                                LatexFormulaExcelExporterUtil.FormulaComponent formulaComponent = new LatexFormulaExcelExporterUtil.FormulaComponent();
+                                formulaComponent.setBase(currentOrderLine.getThickness());
+                                formulaComponent.setSup(currentOrderLine.getThicknessTolPos());
+                                formulaComponent.setSub(currentOrderLine.getThicknessTolNeg());
+                                formulaComponentList.add(formulaComponent);
+                            }
+                            if (currentOrderLine.getWidth() != null) {
+                                LatexFormulaExcelExporterUtil.FormulaComponent formulaComponent = new LatexFormulaExcelExporterUtil.FormulaComponent();
+                                formulaComponent.setBase(currentOrderLine.getWidth());
+                                formulaComponent.setSup(currentOrderLine.getWidthTolPos());
+                                formulaComponent.setSub(currentOrderLine.getWidthTolNeg());
+                                formulaComponentList.add(formulaComponent);
+                            }
+                            if (currentOrderLine.getLength() != null) {
+                                LatexFormulaExcelExporterUtil.FormulaComponent formulaComponent = new LatexFormulaExcelExporterUtil.FormulaComponent();
+                                formulaComponent.setBase(currentOrderLine.getLength());
+                                formulaComponent.setSup(currentOrderLine.getLengthTolPos());
+                                formulaComponent.setSub(currentOrderLine.getLengthTolNeg());
+                                formulaComponentList.add(formulaComponent);
+                            }
+                            String latex = LatexFormulaExcelExporterUtil.convertToLatex(formulaComponentList);
+                            if (StringUtils.isNotBlank(latex)) {
+                                LatexFormulaExcelExporterUtil.insertLatexImageToCell(workbook, sheet, latex, startRow, 4, 1, 7);
+                            }
+                            ExcelUtil.setCellValue(sheet, startRow, 11, currentOrderLine.getStatus());
+                            ExcelUtil.setCellValue(sheet, startRow, 13, currentOrderLine.getQuantity());
+                            ExcelUtil.setCellValue(sheet, startRow, 14, currentOrderLine.getSalesPrice());
+                            ExcelUtil.setCellValue(sheet, startRow, 16, currentOrderLine.getDeliveryDate());
+                            ExcelUtil.setCellValue(sheet, startRow, 17, currentOrderLine.getAssessmentExceedsAgreement());
+                            if (currentOrderLine.getQuantity() != null) {
+                                totalQuantity = totalQuantity.add(currentOrderLine.getQuantity());
+                            }
+                            startRow++;
+                        }
+                    }
+                    Map<String, Object> dataMap = new HashMap<>();
+                    SysDataDicItemService sysDataDicItemService = ApplicationUtil.getBean(SysDataDicItemService.class);
+                    SysDataDicItem supplyUnitDicItem = sysDataDicItemService.findByCode("SUPPLIER", orderInfo.getSupplyUnit());
+                    dataMap.put("supplyUnit", supplyUnitDicItem == null ? "" : supplyUnitDicItem.getName());
+                    dataMap.put("orderNo", orderInfo.getOrderNo());
+                    dataMap.put("orderingUnitName", orderInfo.getOrderingUnitName());
+                    dataMap.put("customerTier", orderInfo.getCustomerTier());
+                    dataMap.put("orderDate", orderInfo.getOrderDate() == null ? "" : orderInfo.getOrderDate().format(dateFormatter));
+                    dataMap.put("totalQuantity", totalQuantity.setScale(1, RoundingMode.HALF_UP));
+                    SysDataDicItem esDicItem = sysDataDicItemService.findByCode("APPLICABLE_STANDARD", orderInfo.getExecutionStandard());
+                    dataMap.put("executionStandard", esDicItem == null ? "" : esDicItem.getName());
+                    dataMap.put("deliveryMethod", orderInfo.getDeliveryMethod());
+                    dataMap.put("invoicingStatus", orderInfo.getInvoicingStatus());
+                    dataMap.put("pieceWeightHeader", orderInfo.getPieceWeightHeader());
+                    dataMap.put("surface", orderInfo.getSurface());
+                    dataMap.put("tolerance", orderInfo.getTolerance());
+                    dataMap.put("performanceance", orderInfo.getPerformance());
+                    dataMap.put("packaging", orderInfo.getPackaging());
+                    dataMap.put("shippingCost", orderInfo.getShippingCost());
+                    dataMap.put("remarks", orderInfo.getRemarks());
+                    dataMap.put("createUser", orderInfo.getCreateBy());
+                    dataMap.put("element", orderInfo.getElement());
+                    String et = orderInfo.getEnterpriseType();
+                    dataMap.put("enterpriseType",
+                            "DEALER".equals(et) ? "经销商" :
+                                    "TERMINAL".equals(et) ? "终端" :
+                                            "FOREIGN".equals(et) ? "外贸" : "");
+                    dataMap.put("deptName", orderInfo.getDeptName() == null ? "" : orderInfo.getDeptName());
+                    dataMap.put("regionName", orderInfo.getRegionName() == null ? "" : orderInfo.getRegionName());
+                    dataMap.put("stockUpCompanyName", orderInfo.getStockUpCompanyName() == null ? "" : orderInfo.getStockUpCompanyName());
+                    dataMap.put("priceListNo", orderInfo.getPriceListNo());
+                    ExcelUtil.processTemplate(workbook, dataMap);
+                    if ("PDF".equals(queryVo.getExportFileType())) {
+                        // 输出PDF
+                        File tempExcel = File.createTempFile("purchase_order_" + orderInfo.getOrderNo(), ".xlsx");
+                        try (FileOutputStream fos = new FileOutputStream(tempExcel)) {
+                            workbook.write(fos);
+                        }
+                        File pdfFile = ExcelUtil.convertExcelToPdf(tempExcel, "/usr/bin/libreoffice --headless --convert-to pdf --outdir %s %s");
+                        try (InputStream pdfIn = new FileInputStream(pdfFile)) {
+                            ZipEntry entry = new ZipEntry(orderInfo.getOrderNo() + "-订货单打印.pdf");
+                            zos.putNextEntry(entry);
+                            byte[] buffer = new byte[8192];
+                            int len;
+                            while ((len = pdfIn.read(buffer)) != -1) {
+                                zos.write(buffer, 0, len);
+                            }
+                            zos.closeEntry();
+                        } finally {
+                            if (tempExcel != null && tempExcel.exists()) {
+                                Files.delete(tempExcel.toPath());
+                            }
+                            if (pdfFile != null && pdfFile.exists()) {
+                                Files.delete(pdfFile.toPath());
+                            }
+                        }
+                    } else {
+                        ByteArrayOutputStream baos = new ByteArrayOutputStream();
+                        workbook.write(baos);
+                        ZipEntry entry = new ZipEntry(orderInfo.getOrderNo() + "-订货单打印.xlsx");
+                        zos.putNextEntry(entry);
+                        zos.write(baos.toByteArray());
+                        zos.closeEntry();
+                    }
+                }
+            }
+            zos.flush();
+        } catch (Exception e) {
+            log.error("订货单批量打印失败: {}", e.getMessage(), e);
+            throw new DefaultClientException(e.getMessage());
+        }
+    }
+
     private int setCellValue(Workbook workbook, Sheet sheet, int rowIndex, List<ReceiptLedgerReportDetail> detailList) {
         if (CollectionUtils.isEmpty(detailList)) {
             return rowIndex;

@@ -13,6 +13,12 @@ public class QueryPurchaseOrderInfoVo extends PageVo implements BaseVo, Serializ
     private static final long serialVersionUID = 1L;
 
     /**
+     * 订货单ID集合
+     */
+    @ApiModelProperty("订货单ID集合")
+    private List<String> ids;
+
+    /**
      * 查询类型
      * REVOKE：撤销
      * CHANGE：变更
@@ -141,4 +147,25 @@ public class QueryPurchaseOrderInfoVo extends PageVo implements BaseVo, Serializ
      */
     @ApiModelProperty("状态集合")
     private List<String> includeStatus;
+
+    /**
+     * 导出类型
+     */
+    @ApiModelProperty("导出类型")
+    private String exportType;
+
+    /**
+     * 导出文件类型
+     * EXCEL/PDF
+     */
+    @ApiModelProperty("导出文件类型")
+    private String exportFileType;
+
+    /**
+     * 模版类型
+     * NORMAL：普通模版
+     * PRODUCTION_PROCESS：生产工艺模版
+     */
+    @ApiModelProperty("导出类型")
+    private String templateType;
 }

@@ -195,6 +195,12 @@
                     #{status}
                 </foreach>
             </if>
+            <if test="vo.ids != null and vo.ids.size() > 0">
+                AND tb.id IN
+                <foreach collection="vo.ids" item="id" open="(" separator="," close=")">
+                    #{id}
+                </foreach>
+            </if>
         </where>
         ORDER BY tb.update_time DESC
     </select>