poi导出excel根据列合并单元格

业务需求是要求把一个树结构的数据导出成excel格式的文件，大概是这样的：

 

 

 上代码：

/**
 *     用于保存需要合并的单元格
 */
public class MergeModel {
    private String content;//内容
    private int rowIndex;//记录相同内容的开始行号
    private int cellIndex;//列号
    public String getContent() {
        return content;
    }
    public void setContent(String content) {
        this.content = content;
    }
    public int getRowIndex() {
        return rowIndex;
    }
    public void setRowIndex(int rowIndex) {
        this.rowIndex = rowIndex;
    }
    public int getCellIndex() {
        return cellIndex;
    }
    public void setCellIndex(int cellIndex) {
        this.cellIndex = cellIndex;
    }
    
}

 

因为是树结构，就会存在这么一种情况：

　　a目录和b目录为同级目录，a目录下最后一个子目录和b目录下的第一个子目录名称相同，在合并时就会发生合并错位情况，所以需要在合并时将子目录也合并；这里不考虑同一目录下名字相同的情况。

import java.io.FileOutputStream;
import java.io.IOException;
import java.text.DecimalFormat;
import java.util.Collections;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.regex.Pattern;

import org.apache.poi.ss.usermodel.HorizontalAlignment;
import org.apache.poi.ss.util.CellRangeAddress;
import org.apache.poi.xssf.usermodel.XSSFCell;
import org.apache.poi.xssf.usermodel.XSSFCellStyle;
import org.apache.poi.xssf.usermodel.XSSFRow;
import org.apache.poi.xssf.usermodel.XSSFSheet;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;

import poiExcel.model.MergeModel;

public class ExportExcel {
    /**
     *     创建excel文件
     * @param objData 数据
     * @param fileName 文件名
     * @param sheetName sheet名
     * @param columns 表头
     * @param mergeIndex 需要合并的列号集合
     * @return 
     */
    public static int exportToExcelForXlsx(List<List<Object>> objData,String fileName, String sheetName, List<String> columns,List<Integer> mergeIndex,boolean isTree) {
        int flag = 0;
        Collections.sort(mergeIndex);//将列号排序
        // 创建工作薄
        XSSFWorkbook wb = new XSSFWorkbook();
        // sheet1
        XSSFSheet sheet1 = wb.createSheet(sheetName);
        
        //设置样式
        XSSFCellStyle style = wb.createCellStyle();
        style.setAlignment(HorizontalAlignment.CENTER);//水平对齐
        
        //表头
        sheet1.createFreezePane(0, 1);//冻结表头
        XSSFRow sheet1row1 = sheet1.createRow((short) 0);
        sheet1row1.setHeight((short) 480);
        //写入表头
        if (columns != null && columns.size() > 0) {
            for(int i=0;i<columns.size();i++) {
                String column = columns.get(i);
                //列
                XSSFCell cell = sheet1row1.createCell(i);
                cell.setCellValue(column);
            }
        }
        
        int dataSatrtIndex = 1;//数据开始行
        boolean isMerge = false;
        if(mergeIndex != null && mergeIndex.size() != 0) {
            isMerge = true;
        }
        
        //写入数据
        if (objData != null && objData.size() > 0) {
            Map<Integer,MergeModel> poiModels = new HashMap<Integer, MergeModel>();
            
            //循环写入表中数据
            int i = 0;
            for ( ; i < objData.size(); i++) {
                //数据行
                XSSFRow row = sheet1.createRow((short) (i+dataSatrtIndex));
                //行内循环，既单元格（列）
                List<Object> list = objData.get(i);
                DecimalFormat decimalFormat = new DecimalFormat("0.00");
                int j = 0;
                for (Object o : list) {
                    //数据列
                    String content = "";
                    if(o != null) {
                        if (o.toString().contains(".") && isNumeric(o.toString())) {
                            content = decimalFormat.format(Float.valueOf(o.toString()));
                        } else if (o.toString().contains("-") && o.toString().contains(":")) {
                            content = String.valueOf(o).split("\\.")[0];
                        }else {
                            content = String.valueOf(o);
                        }
                    }
                    
                    if(isMerge && mergeIndex.contains(j)) {
                        //如果该列需要合并
                        MergeModel poiModel = poiModels.get(j);
                        if(poiModel == null) {
                            poiModel = new MergeModel();
                            poiModel.setContent(content);
                            poiModel.setRowIndex(i + dataSatrtIndex);
                            poiModel.setCellIndex(j);
                            poiModels.put(j, poiModel);
                        }else {
                            if(!poiModel.getContent().equals(content)) {
                                //如果不同了，则将前面的数据合并写入
　　　　　　　　　　　　　　　　　　　　if(isTree){
　　　　　　　　　　　　　　　　　　　　　　//此列向后的所有列都进行一次写入合并操作，并清空。
　　　　　　　　　　　　　　　　　　　　　　//树结构中存在这种情况，a目录和b目录为同级目录，a目录下最后一个子目录和b目录下的第一个子目录名称相同，防止本来不应该合并的单元格被合并
　　　　　　　　　　　　　　　　　　　　　　addMergedRegionValue(sheet1,poiModels,mergeIndex,i + dataSatrtIndex,poiModel.getCellIndex());
　　　　　　　　　　　　　　　　　　　　}else{
　　　　　　　　　　　　　　　　　　　　　　XSSFRow lastRow = sheet1.getRow(poiModel.getRowIndex());
　　　　　　　　　　　　　　　　　　　　　　XSSFCell lastCell = lastRow.createCell(poiModel.getCellIndex());//创建列
　　　　　　　　　　　　　　　　　　　　　　lastCell.setCellValue(poiModel.getContent());
　　　　　　　　　　　　　　　　　　　　　　//合并单元格
　　　　　　　　　　　　　　　　　　　　　　if(poiModel.getRowIndex() != i + dataSatrtIndex - 1) {
　　　　　　　　　　　　　　　　　　　　　　　　sheet1.addMergedRegion(new CellRangeAddress(poiModel.getRowIndex(), i + dataSatrtIndex - 1, poiModel.getCellIndex(), poiModel.getCellIndex()));
　　　　　　　　　　　　　　　　　　　　　　}
　　　　　　　　　　　　　　　　　　　　}

　　　　　　　　　　　　　　　　　　　//将新数据存入
                                poiModel.setContent(content);
                                poiModel.setRowIndex(i + dataSatrtIndex);
                                poiModel.setCellIndex(j);
                                poiModels.put(j, poiModel);
                            }
                        }
                        row.createCell(j);//创建单元格
                    }else {//该列不需要合并
                        //数据列
                        XSSFCell cell = row.createCell(j);
                        cell.setCellValue(content);
                    }
                    
                    j++;
                }
            
            }
            
            //将最后一份存入
            if(poiModels != null && poiModels.size() != 0) {
                for(Integer key : poiModels.keySet()) {
                    MergeModel poiModel = poiModels.get(key);
                    XSSFRow lastRow = sheet1.getRow(poiModel.getRowIndex());
                    XSSFCell lastCell = lastRow.getCell(poiModel.getCellIndex());
                    lastCell.setCellValue(poiModel.getContent());
                    //合并单元格
                    if(poiModel.getRowIndex() != i + dataSatrtIndex -1) {
                        sheet1.addMergedRegion(new CellRangeAddress(poiModel.getRowIndex(), i + dataSatrtIndex - 1, poiModel.getCellIndex(), poiModel.getCellIndex()));
                    }
                }
            }
        } else {
            flag = -1;
        }
        
        //设置固定列宽，poi的列宽设置有点操蛋，大概规律网上有不少版本自行百度
        //这里大概是143像素
        for (int i = 0; i < columns.size(); i++) {
            sheet1.setColumnWidth(i, 4550);
        }
        
        FileOutputStream out = null;  
        try {    
            out = new FileOutputStream("E:\\" + fileName + ".xlsx");
            wb.write(out);
        } catch (Exception ex) {
            try {  
                out.flush();  
                out.close();  
            }catch (IOException e){ 
                flag = 0;
                e.printStackTrace();  
            }  
        }
        return flag;
    }
    
    /**
     *     判断是不是数字
     * @param str
     * @return 
     */
    private static boolean isNumeric(String str){
        if(str == null || str.length() == 0) {
            return false;
        }
        Pattern pattern = Pattern.compile("^[-\\+]?[\\d]*$"); 
        return pattern.matcher(str).matches();
    }
　　
　　private static void addMergedRegionValue(XSSFSheet sheet,Map<Integer,MergeModel> poiModels,List<Integer> mergeIndex,int nowRowIndex,int nowCellIndex){
if(poiModels != null && poiModels.size() != 0 && mergeIndex != null && mergeIndex.size() != 0){
　　　　　　for(Integer index : mergeIndex){
　　　　　　　　if(index >= nowCellIndex){
　　　　　　　　　　MergeModel poiModel = poiModels.remove(index);//删除并获取value
　　　　　　　　　　if(poiModel != null){
　　　　　　　　　　　　XSSFRow lastRow = sheet.getRow(poiModel.getRowIndex());
　　　　　　　　　　　　XSSFCell lastCell = lastRow.createCell(poiModel.getCellIndex());//创建列
　　　　　　　　　　　　lastCell.setCellValue(poiModel.getContent());
　　　　　　　　　　　　//合并单元格
　　　　　　　　　　　　if(poiModel.getRowIndex() != nowRowIndex - 1) {
　　　　　　　　　　　　　　sheet.addMergedRegion(new CellRangeAddress(poiModel.getRowIndex(), nowRowIndex - 1, poiModel.getCellIndex(), poiModel.getCellIndex()));
　　　　　　　　　　　　}
　　　　　　　　　　}
　　　　　　　　}
　　　　　　}
　　　　}
　　}

}

 

import java.util.ArrayList;
import java.util.List;

import poiExcel.utils.ExportExcel;

public class Startover {
    public static void main(String[] args) {
         List<String> columns = new ArrayList<>();//标头
         columns.add("目录");
         columns.add("目录");
         columns.add("目录");
         columns.add("内容");
         String fileName = "文件名字";//文件名字
         String sheetName = "sheet名字";//sheet名字
         //内容数据
         List<List<Object>> exportData = new ArrayList<>();
         //行内的数据
         List<Object> rowData = new ArrayList<Object>();
         rowData.add("一级目录1");
         rowData.add("二级目录1");
         rowData.add("三级目录1");
         rowData.add("内容1");
         exportData.add(rowData);
         List<Object> rowData2 = new ArrayList<Object>();
         rowData2.add("一级目录1");
         rowData2.add("二级目录1");
         rowData2.add("三级目录1");
         rowData2.add("内容2");
         exportData.add(rowData2);
         List<Object> rowData3 = new ArrayList<Object>();
         rowData3.add("一级目录1");
         rowData3.add("二级目录1");
         rowData3.add("三级目录2");
         rowData3.add("内容3");
         exportData.add(rowData3);
         List<Object> rowData4 = new ArrayList<Object>();
         rowData4.add("一级目录1");
         rowData4.add("二级目录1");
         rowData4.add("三级目录2");
         rowData4.add("内容4");
         exportData.add(rowData4);
         List<Object> rowData5 = new ArrayList<Object>();
         rowData5.add("一级目录1");
         rowData5.add("二级目录2");
         rowData5.add("三级目录3");
         rowData5.add("内容5");
         exportData.add(rowData5);
         List<Object> rowData6 = new ArrayList<Object>();
         rowData6.add("一级目录1");
         rowData6.add("二级目录2");
         rowData6.add("三级目录3");
         rowData6.add("内容6");
         exportData.add(rowData6);
         //需要合并的列号
         List<Integer> mergeIndex = new ArrayList<Integer>();
         mergeIndex.add(0);
         mergeIndex.add(1);
         mergeIndex.add(2);
         int flag = ExportExcel.exportToExcelForXlsx(exportData, fileName, sheetName, columns, mergeIndex, true);
         System.out.println(flag);
    }
}