统计数据导出

cash-service/order-service/src/main/java/com/czg/service/order/service/impl/TableSummaryServiceImpl.java

@@ -1,16 +1,21 @@
 package com.czg.service.order.service.impl;
 
 import cn.hutool.core.collection.CollUtil;
+import com.alibaba.excel.write.handler.SheetWriteHandler;
+import com.czg.excel.ExcelExportUtil;
+import com.czg.excel.SheetData;
 import com.czg.order.param.TableSummaryParam;
 import com.czg.order.service.TableSummaryService;
 import com.czg.order.vo.TableSummaryExportVo;
 import com.czg.service.order.mapper.ShopTableOrderStatisticMapper;
 import jakarta.annotation.Resource;
+import jakarta.servlet.http.HttpServletResponse;
 import org.springframework.stereotype.Service;
 
 import java.math.BigDecimal;
 import java.time.LocalDate;
 import java.time.format.DateTimeFormatter;
+import java.util.ArrayList;
 import java.util.List;
 import java.util.Map;
 import java.util.stream.Collectors;
@@ -29,7 +34,7 @@ public class TableSummaryServiceImpl implements TableSummaryService {
 
 
     @Override
-    public List<TableSummaryExportVo> summaryExportList(TableSummaryParam param) {
+    public void summaryExportList(TableSummaryParam param, HttpServletResponse response) {
         if (param.getBeginDate() == null && param.getEndDate() == null) {
             // 获取当前日期
             LocalDate currentDate = LocalDate.now();
@@ -48,7 +53,8 @@ public class TableSummaryServiceImpl implements TableSummaryService {
         }
         List<TableSummaryExportVo> list = shopTableOrderStatisticMapper.findSummaryExportList(param);
         if (CollUtil.isEmpty(list)) {
-            return List.of();
+            ExcelExportUtil.exportToResponse(new ArrayList<>(), TableSummaryExportVo.class, "台桌统计", response);
+            return;
         }
         record TableSummary(String tableKey, BigDecimal totalSales, Map<Long, BigDecimal> productSales) {}
 
@@ -85,10 +91,106 @@ public class TableSummaryServiceImpl implements TableSummaryService {
                 vo.setSalesAmount(summary.productSales().getOrDefault(vo.getProductId(), BigDecimal.ZERO));
             }
         });
-        // 追加个空行用于处理表格样式
-        TableSummaryExportVo nullVo = new TableSummaryExportVo();
-        list.add(nullVo);
-        return list;
+
+        List<SheetWriteHandler> strategies = calculateMergeCells(list);
+        ExcelExportUtil.exportMultipleSheetsToResponse(
+                List.of(new SheetData()
+                        .setData(list)
+                        .setSheetName("台桌统计")
+                        .setHandlers(strategies)
+                        .setClazz(TableSummaryExportVo.class)),
+                "台桌统计",
+                response);
+    }
+
+    /**
+     * 计算需要合并的单元格信息
+     */
+    private List<SheetWriteHandler> calculateMergeCells(List<TableSummaryExportVo> list) {
+        List<SheetWriteHandler> mergeInfos = new ArrayList<>();
+
+        if (CollUtil.isEmpty(list)) {
+            return mergeInfos;
+        }
+
+        // 按createDate分组
+        Map<String, List<TableSummaryExportVo>> byCreateDate = list.stream()
+                .collect(Collectors.groupingBy(TableSummaryExportVo::getCreateDate));
+
+        List<String> keyList = byCreateDate.keySet().stream().sorted().toList();
+
+        // Excel行号从1开始（假设第0行是标题）
+        int startRow = 1;
+
+        for (String key : keyList) {
+            List<TableSummaryExportVo> dateGroup = byCreateDate.get(key);
+            int dateGroupSize = dateGroup.size();
+
+            // 合并createDate列（假设是第1列）
+            if (dateGroupSize > 1) {
+                mergeInfos.add(ExcelExportUtil.createMergeStrategy(startRow, startRow + dateGroupSize - 1, 0, 0));
+                mergeInfos.add(ExcelExportUtil.createMergeStrategy(startRow, startRow + dateGroupSize - 1, 9, 9));
+            }
+
+            // 在销售额组内按tableName分组
+            Map<String, List<TableSummaryExportVo>> byTableName = dateGroup.stream()
+                    .collect(Collectors.groupingBy(TableSummaryExportVo::getTableName));
+
+            List<String> nameKeyList = byTableName.keySet().stream().sorted().toList();
+
+            int currentRow = startRow;
+            for (String nameKey : nameKeyList) {
+                int nameGroupSize = byTableName.get(nameKey).size();
+
+                // 合并tableName列（假设是第4列，根据实际Excel列调整）
+                if (nameGroupSize > 1) {
+                    mergeInfos.add(ExcelExportUtil.createMergeStrategy(currentRow, currentRow + nameGroupSize - 1, 1, 1));
+                }
+
+                // 移动到下一组
+                currentRow += nameGroupSize;
+            }
+
+            // 在日期组内按totalSalesAmount分组（同台桌同日期）
+            Map<BigDecimal, List<TableSummaryExportVo>> byTotalSales = dateGroup.stream()
+                    .collect(Collectors.groupingBy(TableSummaryExportVo::getTotalSalesAmount));
+
+            List<BigDecimal> salesKeyList = byTotalSales.keySet().stream().sorted().toList();
+
+            int nameRow = startRow;
+
+            for (BigDecimal saleKey : salesKeyList) {
+                List<TableSummaryExportVo> salesGroup  = byTotalSales.get(saleKey);
+                int salesGroupSize = salesGroup.size();
+
+                // 合并totalSalesAmount列（假设是第8列，根据实际Excel列调整）
+                if (salesGroupSize > 1) {
+                    mergeInfos.add(ExcelExportUtil.createMergeStrategy(nameRow, nameRow + salesGroupSize - 1, 9, 9));
+                }
+
+//                // 在销售额组内按tableName分组
+//                Map<String, List<TableSummaryExportVo>> byTableName = salesGroup.stream()
+//                        .collect(Collectors.groupingBy(TableSummaryExportVo::getTableName));
+//
+//                List<String> nameKeyList = byTableName.keySet().stream().sorted().toList();
+//
+//                for (List<TableSummaryExportVo> nameGroup : byTableName.values()) {
+//                    int nameGroupSize = nameGroup.size();
+//
+//                    // 合并tableName列（假设是第4列，根据实际Excel列调整）
+//                    if (nameGroupSize > 1) {
+//                        mergeInfos.add(ExcelExportUtil.createMergeStrategy(currentRow, currentRow + nameGroupSize - 1, 1, 1));
+//                    }
+//
+//                    // 移动到下一组
+//                    currentRow += nameGroupSize;
+//                }
+            }
+
+            startRow += dateGroupSize;
+        }
+
+        return mergeInfos;
     }
 
 }