feat: 丰富一下导出excel的内容和样式

servers/boe-server-all/src/main/java/com/xboe/module/boecase/service/impl/CaseAiChatServiceImpl.java

@@ -39,9 +39,7 @@ import org.apache.http.entity.StringEntity;
 import org.apache.http.impl.client.CloseableHttpClient;
 import org.apache.http.impl.client.HttpClients;
 import org.apache.http.util.EntityUtils;
-import org.apache.poi.ss.usermodel.Row;
+import org.apache.poi.ss.usermodel.*;
-import org.apache.poi.ss.usermodel.Sheet;
-import org.apache.poi.ss.usermodel.Workbook;
 import org.apache.poi.ss.util.CellRangeAddress;
 import org.apache.poi.xssf.usermodel.XSSFWorkbook;
 import org.jetbrains.annotations.NotNull;
@@ -933,18 +931,89 @@ public class CaseAiChatServiceImpl implements ICaseAiChatService {
 
         // 写入Excel文件
         Workbook workbook = new XSSFWorkbook();
-        Sheet sheet = workbook.createSheet("AI会话数据");
+        // 样式
+        CellStyle headerStyle = workbook.createCellStyle();
+        Font headerFont = workbook.createFont();
+        headerFont.setBold(true);
+        headerFont.setColor(IndexedColors.WHITE.getIndex());
+        headerStyle.setFont(headerFont);
+        headerStyle.setFillForegroundColor(IndexedColors.DARK_BLUE.getIndex());
+        headerStyle.setFillPattern(FillPatternType.SOLID_FOREGROUND);
+        headerStyle.setAlignment(HorizontalAlignment.CENTER);
+        headerStyle.setVerticalAlignment(VerticalAlignment.CENTER);
+        headerStyle.setBorderTop(BorderStyle.THIN);
+        headerStyle.setBorderBottom(BorderStyle.THIN);
+        headerStyle.setBorderLeft(BorderStyle.THIN);
+        headerStyle.setBorderRight(BorderStyle.THIN);
+
+        CellStyle dataStyle = workbook.createCellStyle();
+        dataStyle.setBorderTop(BorderStyle.THIN);
+        dataStyle.setBorderBottom(BorderStyle.THIN);
+        dataStyle.setBorderLeft(BorderStyle.THIN);
+        dataStyle.setBorderRight(BorderStyle.THIN);
+        dataStyle.setAlignment(HorizontalAlignment.CENTER);
+        dataStyle.setVerticalAlignment(VerticalAlignment.CENTER);
+
+        // 增加一个汇总Sheet
+        Sheet summarySheet = workbook.createSheet("汇总");
+        Row summaryHeaderRow = summarySheet.createRow(0);
+        Cell monthHeaderCell = summaryHeaderRow.createCell(0);
+        monthHeaderCell.setCellValue("月份");
+        monthHeaderCell.setCellStyle(headerStyle);
+        Cell sessionCountHeaderCell = summaryHeaderRow.createCell(1);
+        sessionCountHeaderCell.setCellValue("会话总次数");
+        sessionCountHeaderCell.setCellStyle(headerStyle);
+        Cell userCountHeaderCell = summaryHeaderRow.createCell(2);
+        userCountHeaderCell.setCellValue("会话总人数");
+        userCountHeaderCell.setCellStyle(headerStyle);
+        
+        Row summaryDataRow = summarySheet.createRow(1);
+        Cell monthDataCell = summaryDataRow.createCell(0);
+        monthDataCell.setCellValue(startTime.format(DateTimeFormatter.ofPattern("yyyy年MM月")));
+        monthDataCell.setCellStyle(dataStyle);
+        Cell sessionCountDataCell = summaryDataRow.createCell(1);
+        sessionCountDataCell.setCellValue(conversations.size());
+        sessionCountDataCell.setCellStyle(dataStyle);
+        Cell userCountDataCell = summaryDataRow.createCell(2);
+        userCountDataCell.setCellValue(conversations.stream().map(CaseAiConversations::getConversationUser).distinct().count());
+        userCountDataCell.setCellStyle(dataStyle);
+
+        // 列宽
+        for (int i = 0; i < 3; i++) {
+            summarySheet.autoSizeColumn(i);
+            summarySheet.setColumnWidth(i, summarySheet.getColumnWidth(i) + 500);
+        }
+
+        Sheet dataSheet = workbook.createSheet("AI会话数据");
         // 标题行
-        Row headerRow = sheet.createRow(0);
+        Row dataHeaderRow = dataSheet.createRow(0);
-        headerRow.createCell(0).setCellValue("会话ID");
+        Cell conversationIdHeaderCell = dataHeaderRow.createCell(0);
-        headerRow.createCell(1).setCellValue("会话名称");
+        conversationIdHeaderCell.setCellValue("会话ID");
-        headerRow.createCell(2).setCellValue("用户");
+        conversationIdHeaderCell.setCellStyle(headerStyle);
-        headerRow.createCell(3).setCellValue("提问");
+        Cell conversationNameHeaderCell = dataHeaderRow.createCell(1);
-        headerRow.createCell(4).setCellValue("回答");
+        conversationNameHeaderCell.setCellValue("会话名称");
-        headerRow.createCell(5).setCellValue("开始时间");
+        conversationNameHeaderCell.setCellStyle(headerStyle);
-        headerRow.createCell(6).setCellValue("问答时长(秒)");
+        Cell userHeaderCell = dataHeaderRow.createCell(2);
-        headerRow.createCell(7).setCellValue("消息状态");
+        userHeaderCell.setCellValue("用户ID");
-        headerRow.createCell(8).setCellValue("错误信息");
+        userHeaderCell.setCellStyle(headerStyle);
+        Cell queryHeaderCell = dataHeaderRow.createCell(3);
+        queryHeaderCell.setCellValue("提问");
+        queryHeaderCell.setCellStyle(headerStyle);
+        Cell answerHeaderCell = dataHeaderRow.createCell(4);
+        answerHeaderCell.setCellValue("回答");
+        answerHeaderCell.setCellStyle(headerStyle);
+        Cell startTimeHeaderCell = dataHeaderRow.createCell(5);
+        startTimeHeaderCell.setCellValue("开始时间");
+        startTimeHeaderCell.setCellStyle(headerStyle);
+        Cell durationHeaderCell = dataHeaderRow.createCell(6);
+        durationHeaderCell.setCellValue("问答时长(秒)");
+        durationHeaderCell.setCellStyle(headerStyle);
+        Cell statusHeaderCell = dataHeaderRow.createCell(7);
+        statusHeaderCell.setCellValue("消息状态");
+        statusHeaderCell.setCellStyle(headerStyle);
+        Cell errorMsgHeaderCell = dataHeaderRow.createCell(8);
+        errorMsgHeaderCell.setCellValue("错误信息");
+        errorMsgHeaderCell.setCellStyle(headerStyle);
 
         // 内容行
         if (!excelDataList.isEmpty()) {
@@ -958,48 +1027,78 @@ public class CaseAiChatServiceImpl implements ICaseAiChatService {
 
                     // 遍历每个消息
                     for (CaseAiMessageVo message : messages) {
-                        Row row = sheet.createRow(rowNum++);
+                        Row row = dataSheet.createRow(rowNum++);
                         // 填充每行数据
-                        row.createCell(0).setCellValue(excelData.getConversationId());
+                        Cell conversationIdCell = row.createCell(0);
-                        row.createCell(1).setCellValue(excelData.getConversationName());
+                        conversationIdCell.setCellValue(excelData.getConversationId());
-                        row.createCell(2).setCellValue(excelData.getUser());
+                        conversationIdCell.setCellStyle(dataStyle);
-                        row.createCell(3).setCellValue(message.getQuery() != null ? message.getQuery() : "");
+                        Cell conversationNameCell = row.createCell(1);
-                        row.createCell(4).setCellValue(message.getAnswer() != null ? message.getAnswer() : "");
+                        conversationNameCell.setCellValue(excelData.getConversationName());
+                        conversationNameCell.setCellStyle(dataStyle);
+                        Cell userCell = row.createCell(2);
+                        userCell.setCellValue(excelData.getUser());
+                        userCell.setCellStyle(dataStyle);
+                        Cell queryCell = row.createCell(3);
+                        queryCell.setCellValue(message.getQuery() != null ? message.getQuery() : "");
+                        queryCell.setCellStyle(dataStyle);
+                        Cell answerCell = row.createCell(4);
+                        answerCell.setCellValue(message.getAnswer() != null ? message.getAnswer() : "");
+                        answerCell.setCellStyle(dataStyle);
+                        Cell startTimeCell = row.createCell(5);
                         LocalDateTime messageStartTime = message.getStartTime();
                         if (messageStartTime != null) {
                             String startTimeStr = messageStartTime.format(DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss"));
-                            row.createCell(5).setCellValue(startTimeStr);
+                            startTimeCell.setCellValue(startTimeStr);
                         } else {
-                            row.createCell(5).setCellValue("");
+                            startTimeCell.setCellValue("");
                         }
-                        row.createCell(6).setCellValue(message.getDurationSeconds() != null ? message.getDurationSeconds() : 0);
+                        startTimeCell.setCellStyle(dataStyle);
+                        Cell durationCell = row.createCell(6);
+                        durationCell.setCellValue(message.getDurationSeconds() != null ? message.getDurationSeconds() : 0);
+                        durationCell.setCellStyle(dataStyle);
+                        Cell statusCell = row.createCell(7);
                         if (message.getStatus() != null) {
                             int status = message.getStatus();
                             CaseAiChatErrCodeEnum errCodeEnum = CaseAiChatErrCodeEnum.getByCode(status);
-                            row.createCell(7).setCellValue(errCodeEnum.getLabel());
+                            statusCell.setCellValue(errCodeEnum.getLabel());
                         }
+                        statusCell.setCellStyle(dataStyle);
+                        Cell errorMsgCell = row.createCell(8);
                         if (StringUtils.isNotBlank(message.getErrorMsg())) {
-                            row.createCell(8).setCellValue(message.getErrorMsg());
+                            errorMsgCell.setCellValue(message.getErrorMsg());
                         }
+                        errorMsgCell.setCellStyle(dataStyle);
                     }
 
                     // 合并单元格（会话ID、会话名称、用户三列）
                     // 参数说明：起始行号，结束行号，起始列号，结束列号
                     if (rowNum > startRow + 1) { // 只有当有多行时才合并
-                        sheet.addMergedRegion(new CellRangeAddress(startRow, rowNum - 1, 0, 0));
+                        dataSheet.addMergedRegion(new CellRangeAddress(startRow, rowNum - 1, 0, 0));
-                        sheet.addMergedRegion(new CellRangeAddress(startRow, rowNum - 1, 1, 1));
+                        dataSheet.addMergedRegion(new CellRangeAddress(startRow, rowNum - 1, 1, 1));
-                        sheet.addMergedRegion(new CellRangeAddress(startRow, rowNum - 1, 2, 2));
+                        dataSheet.addMergedRegion(new CellRangeAddress(startRow, rowNum - 1, 2, 2));
                     }
                 } else {
                     // 如果没有消息，则仍然创建一行显示基本信息
-                    Row row = sheet.createRow(rowNum++);
+                    Row row = dataSheet.createRow(rowNum++);
-                    row.createCell(0).setCellValue(excelData.getConversationId());
+                    Cell conversationIdCell = row.createCell(0);
-                    row.createCell(1).setCellValue(excelData.getConversationName());
+                    conversationIdCell.setCellValue(excelData.getConversationId());
-                    row.createCell(2).setCellValue(excelData.getUser());
+                    conversationIdCell.setCellStyle(dataStyle);
+                    Cell conversationNameCell = row.createCell(1);
+                    conversationNameCell.setCellValue(excelData.getConversationName());
+                    conversationNameCell.setCellStyle(dataStyle);
+                    Cell userCell = row.createCell(2);
+                    userCell.setCellValue(excelData.getUser());
+                    userCell.setCellStyle(dataStyle);
                 }
             }
         }
 
+        // 列宽
+        for (int i = 0; i < 9; i++) {
+            summarySheet.autoSizeColumn(i);
+            summarySheet.setColumnWidth(i, summarySheet.getColumnWidth(i) + 500);
+        }
+
         return workbook;
     }