如何使用poi将包含头部的jtable保存到excel？

import org.apache.poi.ss.usermodel.*;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;

import javax.swing.*;
import java.io.FileOutputStream;
import java.io.IOException;

public class JTableToExcel {
    public static void main(String[] args) {
        JTable table = new JTable(); // 假设已经有一个包含头部的JTable对象

        try (Workbook workbook = new XSSFWorkbook()) {
            Sheet sheet = workbook.createSheet("Sheet1");

            // 创建表头行
            Row headerRow = sheet.createRow(0);
            TableColumnModel columnModel = table.getColumnModel();
            for (int i = 0; i < columnModel.getColumnCount(); i++) {
                String headerValue = columnModel.getColumn(i).getHeaderValue().toString();
                Cell cell = headerRow.createCell(i);
                cell.setCellValue(headerValue);
            }

            // 写入数据行
            for (int row = 0; row < table.getRowCount(); row++) {
                Row dataRow = sheet.createRow(row + 1);
                for (int col = 0; col < table.getColumnCount(); col++) {
                    Object value = table.getValueAt(row, col);
                    Cell cell = dataRow.createCell(col);
                    if (value != null) {
                        cell.setCellValue(value.toString());
                    }
                }
            }

            // 保存Excel文件
            try (FileOutputStream outputStream = new FileOutputStream("output.xlsx")) {
                workbook.write(outputStream);
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
}