import org.apache.poi.ss.usermodel.*;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.apache.commons.lang3.StringUtils;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;
/**
 * 災(zāi)情速報(bào)員 Excel 導(dǎo)出服務(wù)
 * 核心特性：
 * 1. 列數(shù)與表頭自動(dòng)聯(lián)動(dòng)，增刪列僅改表頭數(shù)組，無(wú)需修改循環(huán)條件
 * 2. 完整細(xì)框線、統(tǒng)一行高、表頭加粗居中，格式規(guī)范美觀
 * 3. Redis 數(shù)據(jù)安全轉(zhuǎn)換，避免類(lèi)型強(qiáng)轉(zhuǎn)異常
 * 4. 空數(shù)據(jù)友好提示、字段空值安全處理，防止導(dǎo)出報(bào)錯(cuò)
 * 5. 資源安全關(guān)閉，避免內(nèi)存泄漏
 * @author 編程實(shí)戰(zhàn)博客
 */
public class ExcelExportService {
    private static final Logger log = LoggerFactory.getLogger(ExcelExportService.class);
    /**
     * 統(tǒng)一行高（20點(diǎn)為Excel舒適行高，可根據(jù)需求調(diào)整為18/22）
     */
    private static final short ROW_HEIGHT = 20;
    /**
     * 【核心】表頭全局唯一入口，所有列數(shù)相關(guān)邏輯自動(dòng)聯(lián)動(dòng)
     * 增刪列：僅需修改此數(shù)組，其他地方無(wú)需改動(dòng)
     */
    private static final String[] HEADERS = {
        "序號(hào)", "姓名", "電話", "震中距離", "市州", "區(qū)縣", "鄉(xiāng)鎮(zhèn)/街道", "村/社區(qū)", "備注"
    };
    /**
     * 導(dǎo)出災(zāi)情速報(bào)員數(shù)據(jù)為Excel字節(jié)流
     * @param key Redis緩存key（用于獲取數(shù)據(jù)）
     * @return Excel字節(jié)流（可直接響應(yīng)前端下載）
     */
    public ByteArrayOutputStream exportToExcel(String key) {
        // 初始化空集合，避免空指針異常
        List<DisasterOfficer> doList = new ArrayList<>();
        // 1. 安全從Redis獲取數(shù)據(jù)，避免類(lèi)型強(qiáng)轉(zhuǎn)異常
        if (StringUtils.isNotBlank(key)) {
            Object cacheObj = redisCache.getCacheObject(Constants.DO_Redis + "_" + key);
            // 類(lèi)型校驗(yàn)：確保緩存數(shù)據(jù)是List<DisasterOfficer>
            if (cacheObj instanceof List<?>) {
                doList = (List<DisasterOfficer>) cacheObj;
            } else if (cacheObj != null) {
                log.error("Redis緩存數(shù)據(jù)類(lèi)型異常，key:{}，預(yù)期List<DisasterOfficer>", Constants.DO_Redis + "_" + key);
                return new ByteArrayOutputStream();
            }
        }
        Workbook workbook = null;
        ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
        try {
            // 2. 創(chuàng)建Excel工作簿（XSSFWorkbook對(duì)應(yīng)XLSX格式）
            workbook = new XSSFWorkbook();
            Sheet sheet = workbook.createSheet("災(zāi)情速報(bào)員");
            // 3. 創(chuàng)建單元格樣式（表頭樣式+數(shù)據(jù)樣式，均帶完整框線）
            CellStyle headerStyle = createHeaderCellStyle(workbook);
            CellStyle dataStyle = createDataCellStyle(workbook);
            // 4. 構(gòu)建表頭（自動(dòng)適配表頭列數(shù)）
            Row headerRow = sheet.createRow(0);
            headerRow.setHeightInPoints(ROW_HEIGHT);
            buildHeader(headerRow, headerStyle);
            // 5. 填充數(shù)據(jù)（自動(dòng)適配表頭列數(shù)，空數(shù)據(jù)自動(dòng)提示）
            fillData(sheet, doList, dataStyle);
            // 6. 將Excel寫(xiě)入字節(jié)流，返回給前端下載
            workbook.write(outputStream);
            log.info("災(zāi)情速報(bào)員Excel導(dǎo)出成功，數(shù)據(jù)條數(shù)：{}", doList.size());
        } catch (IOException e) {
            log.error("災(zāi)情速報(bào)員Excel導(dǎo)出失敗", e); // 打印完整異常堆棧，便于排查
        } finally {
            // 7. 強(qiáng)制關(guān)閉Workbook，釋放重量級(jí)資源，避免內(nèi)存泄漏（核心！）
            if (workbook != null) {
                try {
                    workbook.close();
                } catch (IOException e) {
                    log.error("關(guān)閉Excel工作簿失敗", e);
                }
            }
        }
        return outputStream;
    }
    /**
     * 創(chuàng)建表頭樣式：加粗+水平垂直居中+完整細(xì)框線
     */
    private CellStyle createHeaderCellStyle(Workbook workbook) {
        CellStyle style = workbook.createCellStyle();
        // 字體設(shè)置：加粗、12號(hào)字
        Font font = workbook.createFont();
        font.setBold(true);
        font.setFontHeightInPoints((short) 12);
        style.setFont(font);
        // 對(duì)齊方式：水平居中+垂直居中
        style.setAlignment(HorizontalAlignment.CENTER);
        style.setVerticalAlignment(VerticalAlignment.CENTER);
        // 完整框線：上、下、左、右均為細(xì)實(shí)線
        style.setBorderTop(BorderStyle.THIN);
        style.setBorderBottom(BorderStyle.THIN);
        style.setBorderLeft(BorderStyle.THIN);
        style.setBorderRight(BorderStyle.THIN);
        return style;
    }
    /**
     * 創(chuàng)建數(shù)據(jù)行樣式：水平垂直居中+完整細(xì)框線
     */
    private CellStyle createDataCellStyle(Workbook workbook) {
        CellStyle style = workbook.createCellStyle();
        style.setAlignment(HorizontalAlignment.CENTER);
        style.setVerticalAlignment(VerticalAlignment.CENTER);
        // 與表頭統(tǒng)一框線樣式，保證整體美觀
        style.setBorderTop(BorderStyle.THIN);
        style.setBorderBottom(BorderStyle.THIN);
        style.setBorderLeft(BorderStyle.THIN);
        style.setBorderRight(BorderStyle.THIN);
        return style;
    }
    /**
     * 構(gòu)建表頭：自動(dòng)適配表頭列數(shù)，設(shè)置列寬、單元格值和樣式
     */
    private void buildHeader(Row headerRow, CellStyle style) {
        Sheet sheet = headerRow.getSheet();
        // 循環(huán)次數(shù) = 表頭數(shù)組長(zhǎng)度，自動(dòng)適配列數(shù)
        for (int i = 0; i < HEADERS.length; i++) {
            Cell cell = headerRow.createCell(i);
            cell.setCellValue(HEADERS[i]); // 表頭文本
            cell.setCellStyle(style);      // 應(yīng)用表頭樣式
            sheet.setColumnWidth(i, 20 * 256); // 統(tǒng)一列寬（20個(gè)字符，適配大部分文本）
        }
    }
    /**
     * 填充數(shù)據(jù)：自動(dòng)適配表頭列數(shù)，支持空數(shù)據(jù)提示、空值安全處理
     */
    private void fillData(Sheet sheet, List<DisasterOfficer> doList, CellStyle style) {
        int columnCount = HEADERS.length; // 自動(dòng)獲取總列數(shù)，與表頭聯(lián)動(dòng)
        // 場(chǎng)景1：無(wú)數(shù)據(jù)/數(shù)據(jù)過(guò)期，顯示友好提示
        if (doList == null || doList.isEmpty()) {
            Row emptyRow = sheet.createRow(1);
            emptyRow.setHeightInPoints(ROW_HEIGHT); // 與數(shù)據(jù)行統(tǒng)一行高
            // 提示信息放在第一列
            Cell tipCell = emptyRow.createCell(0);
            tipCell.setCellValue("查詢(xún)無(wú)數(shù)據(jù)或者已過(guò)期，請(qǐng)重新查詢(xún)下載！");
            tipCell.setCellStyle(style);
            // 自動(dòng)補(bǔ)全所有列，保證框線完整（循環(huán)次數(shù)=總列數(shù)）
            for (int j = 1; j < columnCount; j++) {
                emptyRow.createCell(j).setCellStyle(style);
            }
            return;
        }
        // 場(chǎng)景2：有數(shù)據(jù)，循環(huán)填充（序號(hào)從1開(kāi)始）
        for (int i = 0; i < doList.size(); i++) {
            DisasterOfficer officer = doList.get(i);
            Row dataRow = sheet.createRow(i + 1); // 行號(hào)從1開(kāi)始（0行為表頭）
            dataRow.setHeightInPoints(ROW_HEIGHT);
            // 列順序與表頭嚴(yán)格對(duì)應(yīng)，空值自動(dòng)轉(zhuǎn)為空字符串
            dataRow.createCell(0).setCellValue(i + 1); // 序號(hào)
            dataRow.createCell(1).setCellValue(getSafeValue(officer.getContacts())); // 姓名
            dataRow.createCell(2).setCellValue(getSafeValue(officer.getPhone())); // 電話
            dataRow.createCell(3).setCellValue(getSafeValue(officer.getDistance())); // 震中距離
            dataRow.createCell(4).setCellValue(getSafeValue(officer.getCity())); // 市州
            dataRow.createCell(5).setCellValue(getSafeValue(officer.getCounty())); // 區(qū)縣
            dataRow.createCell(6).setCellValue(getSafeValue(officer.getTowns())); // 鄉(xiāng)鎮(zhèn)/街道
            dataRow.createCell(7).setCellValue(getSafeValue(officer.getVillage())); // 村/社區(qū)
            dataRow.createCell(8).setCellValue(getSafeValue(officer.getRemark())); // 備注
            // 為所有單元格應(yīng)用樣式（循環(huán)次數(shù)=總列數(shù)，自動(dòng)適配）
            for (int j = 0; j < columnCount; j++) {
                dataRow.getCell(j).setCellStyle(style);
            }
        }
    }
    /**
     * 空值安全處理工具方法：null轉(zhuǎn)為空字符串，避免導(dǎo)出異常
     */
    private String getSafeValue(Object value) {
        return value == null ? "" : value.toString();
    }
    // ===================== 以下為項(xiàng)目依賴(lài)類(lèi)（僅作示意，需替換為實(shí)際類(lèi)） =====================
    /**
     * 常量類(lèi)：Redis緩存key前綴
     */
    private static class Constants {
        public static String DO_Redis = "DISASTER_OFFICER"; // 災(zāi)情速報(bào)員緩存前綴
    }
    /**
     * Redis緩存工具類(lèi)（示意），實(shí)際項(xiàng)目中替換為自己的Redis工具
     */
    private static class redisCache {
        public static Object getCacheObject(String key) {
            // 實(shí)際邏輯：從Redis獲取數(shù)據(jù)，此處僅作占位
            return null;
        }
    }
    /**
     * 災(zāi)情速報(bào)員實(shí)體類(lèi)（已包含備注字段）
     */
    private static class DisasterOfficer {
        private String contacts;  // 姓名
        private String phone;     // 電話
        private String distance;  // 震中距離
        private String city;      // 市州
        private String county;    // 區(qū)縣
        private String towns;     // 鄉(xiāng)鎮(zhèn)/街道
        private String village;   // 村/社區(qū)
        private String remark;    // 備注（新增字段）
        // getter方法（必須，用于獲取字段值）
        public String getContacts() { return contacts; }
        public String getPhone() { return phone; }
        public String getDistance() { return distance; }
        public String getCity() { return city; }
        public String getCounty() { return county; }
        public String getTowns() { return towns; }
        public String getVillage() { return village; }
        public String getRemark() { return remark; }
    }
}

// 全局唯一表頭，增刪列僅改這里
private static final String[] HEADERS = {"序號(hào)", "姓名", ..., "備注"};

// 正確寫(xiě)法（無(wú)報(bào)錯(cuò)，兼容Apache Commons包）
if (StringUtils.isNotBlank(key)) { ... }