java

hongjing_z 2019-07-31

展開全文

先開始接到這個需求的時候，第一想法是這得要多復雜的算才能實現(xiàn)，因為不是每一列的合并的行數(shù)一樣，比如第一列1-4行的內容都相同，那么需要合并，而第二列1-2相同，所以1和2合并，3-4相同，3和4合并；然后一層一層的，就和樹一樣，逐層向下；
毋庸置疑，通過算法肯定能實現(xiàn)，但是我覺得太繞了，一下子也想不出來什么好的算法；后來我就想到約瑟夫環(huán)的問題，一個典型的java面向對象的例子；后來就通過建一個單元格對象，屬性有單元格內容，與單元格同一列的上一行內容，起始行，起始列，就這四個屬性；詳細解釋我放在代碼中：

model對象：

/**
 * Created by zelei.fan on 2017/3/20.
 */
public class PoiModel {
    private String content;
    private String oldContent;
    private int rowIndex;
    private int cellIndex;
    public String getOldContent() {
        return oldContent;
    }
    public void setOldContent(String oldContent) {
        this.oldContent = oldContent;
    }
    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;
    }
}

Test:

import cn.yoho.perf.common.model.PoiModel;
import com.beust.jcommander.internal.Maps;
import com.google.common.collect.Lists;
import org.apache.poi.ss.usermodel.Cell;
import org.apache.poi.ss.usermodel.Row;
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 java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.util.*;
/**
 * Created by zelei.fan on 2017/3/14.
 */
public class Test{
    /**
     * @param title 標題集合 tilte的長度應該與list中的model的屬性個數(shù)一致
     * @param maps 內容集合
     * @param mergeIndex 合并單元格的列
     */
    public static String createExcel(String[] title, Map<String/*sheet名*/, List<Map<String/*對應title的值*/, String>>> maps, int[] mergeIndex){
        if (title.length==0){
            return null;
        }
        /*初始化excel模板*/
        Workbook workbook = new XSSFWorkbook();
        Sheet sheet = null;
        int n = 0;
        /*循環(huán)sheet頁*/
        for(Map.Entry<String, List<Map<String/*對應title的值*/, String>>> entry : maps.entrySet()){
            /*實例化sheet對象并且設置sheet名稱，book對象*/
            try {
                sheet = workbook.createSheet();
                workbook.setSheetName(n, entry.getKey());
                workbook.setSelectedTab(0);
            }catch (Exception e){
                e.printStackTrace();
            }
            /*初始化head，填值標題行（第一行）*/
            Row row0 = sheet.createRow(0);
            for(int i = 0; i<title.length; i++){
                /*創(chuàng)建單元格，指定類型*/
                Cell cell_1 = row0.createCell(i, Cell.CELL_TYPE_STRING);
                cell_1.setCellValue(title[i]);
            }
            /*得到當前sheet下的數(shù)據(jù)集合*/
            List<Map<String/*對應title的值*/, String>> list = entry.getValue();
            /*遍歷該數(shù)據(jù)集合*/
            List<PoiModel> poiModels = Lists.newArrayList();
            if(null!=workbook){
                Iterator iterator = list.iterator();
                int index = 1;/*這里1是從excel的第二行開始，第一行已經(jīng)塞入標題了*/
                while (iterator.hasNext()){
                    Row row = sheet.createRow(index);
                    /*取得當前這行的map，該map中以key，value的形式存著這一行值*/
                    Map<String, String> map = (Map<String, String>)iterator.next();
                    /*循環(huán)列數(shù)，給當前行塞值*/
                    for(int i = 0; i<title.length; i++){
                        String old = "";
                        /*old存的是上一行統(tǒng)一位置的單元的值，第一行是最上一行了，所以從第二行開始記*/
                        if(index > 1){
                            old = poiModels.get(i)==null?"":poiModels.get(i).getContent();
                        }
                        /*循環(huán)需要合并的列*/
                        for(int j = 0; j < mergeIndex.length; j++){
                            if(index == 1){
                                /*記錄第一行的開始行和開始列*/
                                PoiModel poiModel = new PoiModel();
                                poiModel.setOldContent(map.get(title[i]));
                                poiModel.setContent(map.get(title[i]));
                                poiModel.setRowIndex(1);
                                poiModel.setCellIndex(i);
                                poiModels.add(poiModel);
                                break;
                            }else if(i > 0 && mergeIndex[j] == i){/*這邊i>0也是因為第一列已經(jīng)是最前一列了，只能從第二列開始*/
                                /*當前同一列的內容與上一行同一列不同時，把那以上的合并, 或者在當前元素一樣的情況下，前一列的元素并不一樣，這種情況也合并*/
                                /*如果不需要考慮當前行與上一行內容相同，但是它們的前一列內容不一樣則不合并的情況，把下面條件中||poiModels.get(i).getContent().equals(map.get(title[i])) && !poiModels.get(i - 1).getOldContent().equals(map.get(title[i-1]))去掉就行*/
                                if(!poiModels.get(i).getContent().equals(map.get(title[i])) || poiModels.get(i).getContent().equals(map.get(title[i])) && !poiModels.get(i - 1).getOldContent().equals(map.get(title[i-1]))){
                                    /*當前行的當前列與上一行的當前列的內容不一致時，則把當前行以上的合并*/
                                    CellRangeAddress cra=new CellRangeAddress(poiModels.get(i).getRowIndex()/*從第二行開始*/, index - 1/*到第幾行*/, poiModels.get(i).getCellIndex()/*從某一列開始*/, poiModels.get(i).getCellIndex()/*到第幾列*/);
                                    //在sheet里增加合并單元格
                                    sheet.addMergedRegion(cra);
                                    /*重新記錄該列的內容為當前內容，行標記改為當前行標記，列標記則為當前列*/
                                    poiModels.get(i).setContent(map.get(title[i]));
                                    poiModels.get(i).setRowIndex(index);
                                    poiModels.get(i).setCellIndex(i);
                                }
                            }
                            /*處理第一列的情況*/
                            if(mergeIndex[j] == i && i == 0 && !poiModels.get(i).getContent().equals(map.get(title[i]))){
                                /*當前行的當前列與上一行的當前列的內容不一致時，則把當前行以上的合并*/
                                CellRangeAddress cra=new CellRangeAddress(poiModels.get(i).getRowIndex()/*從第二行開始*/, index - 1/*到第幾行*/, poiModels.get(i).getCellIndex()/*從某一列開始*/, poiModels.get(i).getCellIndex()/*到第幾列*/);
                                //在sheet里增加合并單元格
                                sheet.addMergedRegion(cra);
                                /*重新記錄該列的內容為當前內容，行標記改為當前行標記*/
                                poiModels.get(i).setContent(map.get(title[i]));
                                poiModels.get(i).setRowIndex(index);
                                poiModels.get(i).setCellIndex(i);
                            }
                            /*最后一行沒有后續(xù)的行與之比較，所有當?shù)阶詈笠恍袝r則直接合并對應列的相同內容*/
                            if(mergeIndex[j] == i && index == list.size()){
                                CellRangeAddress cra=new CellRangeAddress(poiModels.get(i).getRowIndex()/*從第二行開始*/, index/*到第幾行*/, poiModels.get(i).getCellIndex()/*從某一列開始*/, poiModels.get(i).getCellIndex()/*到第幾列*/);
                                //在sheet里增加合并單元格
                                sheet.addMergedRegion(cra);
                            }
                        }
                        Cell cell = row.createCell(i, Cell.CELL_TYPE_STRING);
                        cell.setCellValue(map.get(title[i]));
                        /*在每一個單元格處理完成后，把這個單元格內容設置為old內容*/
                        poiModels.get(i).setOldContent(old);
                    }
                    index++;
                }
            }
            n++;
        }
        /*生成臨時文件*/
        FileOutputStream out = null;
        String localPath = null;
        File tempFile = null;
        String fileName = String.valueOf(new Date().getTime()/1000);
        try {
            tempFile = File.createTempFile(fileName, ".xlsx");
            localPath = tempFile.getAbsolutePath();
            out = new FileOutputStream(localPath);
            workbook.write(out);
        }catch (IOException e){
            e.printStackTrace();
        }finally {
            try {
                out.flush();
                out.close();
            }catch (IOException e){
                e.printStackTrace();
            }
        }
        return localPath;
    }
    public static void main(String[] args) throws IOException{
        /*此處標題的數(shù)組則對應excel的標題*/
        String[] title = {"id","標題","描述","負責人","開始時間"};
        List<Map<String, String>> list = Lists.newArrayList();
        /*這邊是制造一些數(shù)據(jù)，注意每個list中map的key要和標題數(shù)組中的元素一致*/
        for(int i = 0; i<10; i++){
            HashMap<String, String> map = com.google.common.collect.Maps.newHashMap();
            if(i > 5){
                if(i<7){
                    map.put("id","333");
                    map.put("標題","mmmm");
                }else {
                    map.put("id","333");
                    map.put("標題","aaaaa");
                }
            }else if (i >3){
                map.put("id","222");
                map.put("標題","哈哈哈哈");
            }else if (i>1 && i<3){
                map.put("id","222");
                map.put("標題","hhhhhhhh");
            }else {
                map.put("id","222");
                map.put("標題","bbbb");
            }
            map.put("描述","sssssss");
            map.put("負責人","vvvvv");
            map.put("開始時間","2017-02-27 11:20:26");
            list.add(map);
        }
        Map<String/*此處的key為每個sheet的名稱，一個excel中可能有多個sheet頁*/, List<Map<String/*此處key對應每一列的標題*/, String>>/*該list為每個sheet頁的數(shù)據(jù)*/> map = Maps.newHashMap();
        map.put("測試合并數(shù)據(jù)", list);
        System.out.println(createExcel(title, map, new int[]{0,1,2}/*此處數(shù)組為需要合并的列，可能有的需求是只需要某些列里面相同內容合并*/));
    }
}