java poi分批次導入Excel

最近換了新工作，公司要求導入Excel要分批次導入，並且是多線程的情況下執行導入，查了很多資料，沒看到比較復合的，就打算自己寫一個吧，可能有不足，希望指出。

上面說到多線程，這邊就不貼出代碼了，具體思路就是一個導入開辟一個線程，下面主要寫一下我的分批次的代碼思路：

分批次導入-方法一

先介紹我一開始的寫法：

通過一個（最大行數/設置讀取的行數）的 余數向上取整 來控制循環次數。

package oldboy;

import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.ss.usermodel.Cell;
import org.apache.poi.ss.usermodel.Row;
import org.apache.poi.ss.usermodel.Sheet;

import java.io.File;
import java.io.FileInputStream;

/**
 * Created by peng on 18/9/12.
 */
public class ExcelDemo {

    public static int READ_INDEX = 10;//這里設置每次讀取的行數

    public static void main(String[] args) throws Exception {
        testImport();
    }

    public static void testImport() throws Exception {
        boolean flag = true;
        //cycleNum 循環讀取次數
        int cycleNum = read("/Users/peng/Desktop/20180912/111.xls", 0);
        while (flag) {
            if (cycleNum != 0) {
                cycleNum = read("/Users/peng/Desktop/20180912/111.xls", cycleNum);
            } else {
                flag = false;
            }
        }
    }

    /**
     * 讀取Excel
     *
     * @param localPath 文件路徑
     * @param cycleNum  剩余的讀取次數
     * @return
     * @throws Exception
     */
    public static int read(String localPath, int cycleNum) throws Exception {
        File file = new File(localPath);
        FileInputStream inputStream = new FileInputStream(file);
        HSSFWorkbook book = new HSSFWorkbook(inputStream);
        Sheet sheet = book.getSheetAt(0); //sheet 從0開始
        Row row = sheet.getRow(0);      //獲取第一行
        int rowEndNum = getRow();  //取得最后一行的行號
        //向上取整
        int cycleIndex = (int) (Math.ceil((double) rowEndNum / (double) READ_INDEX));
        int cycleStartNum, cycleEndNum;
        if (cycleNum == 0) {//第一次循環進來
            cycleNum = cycleIndex;
            cycleStartNum = 1;
            if (rowEndNum > READ_INDEX) {
                cycleEndNum = READ_INDEX + 1;
            } else {//第一次循環如果最大行數小於設定的讀取行數
                cycleEndNum = rowEndNum;
            }
        } else {
            if (cycleNum == 1) {//最后一次循環的時候
                cycleStartNum = READ_INDEX * (cycleIndex - cycleNum) + 1;
                cycleEndNum = rowEndNum;
            } else {
                cycleStartNum = READ_INDEX * (cycleIndex - cycleNum) + 1;
                cycleEndNum = READ_INDEX + READ_INDEX * (cycleIndex - cycleNum) + 1;
            }
        }
        cycleNum -= 1;
        //每次循環的開始行數和結束行數
        System.out.println(cycleStartNum + "=======" + cycleEndNum);
        //System.out.println(rowEndNum);
        for (int i = cycleStartNum; i < cycleEndNum; i++) {//跳過第一行從第二行開始循環
            row = sheet.getRow(i);
            for (int j = 0; j < row.getLastCellNum(); j++) {//列循環開始,從第0列開始
                Cell cell = row.getCell(j);
                //ExcelUtils.getCellValue獲取單元格內容，這里忽略
                String cellValue = ExcelUtils.getCellValue(cell);
                //System.out.print(cellValue.concat("   |  "));
            }
            //System.out.println();
        }
        //返回剩余循環次數
        return cycleNum;
    }

    //這里獲取Excel的真實行數
    public static int getRow() {
        return 47;
    }

}

執行結果：

分批次導入-方法二

這個方法是我比較推薦的，最后我也是采用這種寫法，一次讀一行，每次讀到設定的行數就保存一次，讀到空行停止。

 

package oldboy;

import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.ss.usermodel.Cell;
import org.apache.poi.ss.usermodel.Row;
import org.apache.poi.ss.usermodel.Sheet;

import java.io.File;
import java.io.FileInputStream;
import java.util.ArrayList;
import java.util.List;

/**
 * Created by peng on 18/9/12.
 */
public class ExcelDemo1 {

    public static int READ_INDEX = 10;//這里設置每次讀取的行數

    public static void main(String[] args) throws Exception {
        testNewImport();
    }


    public static void testNewImport() throws Exception {
        String localPath = "/Users/peng/Desktop/20180912/111.xls";
        //每次讀READ_INDEX的量
        File file = new File(localPath);
        List<List<Object>> list = new ArrayList<>();
        int i = 0;
        while (true) {
            //如果是讀到下一行為空
            //每一行的list
            List<Object> rowList = readRow(file, i++);
            if (rowList != null) {
                list.add(rowList);
            } else {
                System.out.println("===================退出導入=============");
                System.out.println("===================保存剩下的=============" + i);
                //做保存動作
                list.clear();
                break;
            }
            if (list.size() == READ_INDEX) {
                //做保存動作
                System.out.println("===================保存=============" + i);
                list.clear();
            }
        }
    }

    public static List<Object> readRow(File file, int rowNum) throws Exception {
        FileInputStream inputStream = new FileInputStream(file);
        HSSFWorkbook book = new HSSFWorkbook(inputStream);
        Sheet sheet = book.getSheetAt(0);//sheet 從0開始
        List<Object> list = new ArrayList<>();
        Row row = sheet.getRow(rowNum);
        if (!isRowEmpty(row)) {
            for (int j = 0; j < row.getLastCellNum(); j++) {//列循環開始,從第0列開始
                Cell cell = row.getCell(j);
                if (cell == null) {
                    continue;
                }
                String cellValue;
                if (ExcelUtils.isMergedRegion(sheet, cell)) {
                    cellValue = ExcelUtils.getMergedRegionValue(sheet, cell);
                } else {
                    cellValue = ExcelUtils.getCellValue(cell);
                }
                list.add(cellValue);
                //System.out.print(cellValue.concat("   |  "));
            }
            //System.out.println();
        } else {
            return null;
        }

        return list;
    }

    /**
     * 判斷是否為空行
     *
     * @param row
     * @return
     */
    public static boolean isRowEmpty(Row row) {
        if (row == null) {
            return true;
        }
        for (int c = row.getFirstCellNum(); c < row.getLastCellNum(); c++) {
            Cell cell = row.getCell(c);
            if (cell != null && cell.getCellType() != Cell.CELL_TYPE_BLANK)
                return false;
        }
        return true;
    }

}

運行結果：

至於為什么一個是47一個是48大家自己控制一下就好了，很簡單，希望對大家有幫助，有不助的地方可以提出來交流一下，轉載注明出處，謝謝。