Java操作Excel（使用POI）

背景說明

以前寫過使用 JXL 操作Excel的例子，但JXL對於Excel 2007版本以后的文件（即擴展名為 .xlsx）無法讀取，也找不到可以支持的包。所以，有時不得不用 POI 來操作Excel。正好前幾天用了一下，記錄備查。

 

示例代碼

1、讀取Excel類。這是網上找的一個代碼，親測可用。（不必讀懂，拷走直接用，只要知道如何調就行，參行第2節代碼）

import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.util.ArrayList;
import java.util.List;
import org.apache.poi.hssf.usermodel.HSSFCell;
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 org.apache.poi.ss.usermodel.Workbook;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;

/**
 * 
 * @描述：測試excel讀取
 *               導入的jar包
 *               poi-3.8-beta3-20110606.jar
 *               poi-ooxml-3.8-beta3-20110606.jar
 *               poi-examples-3.8-beta3-20110606.jar
 *               poi-excelant-3.8-beta3-20110606.jar
 *               poi-ooxml-schemas-3.8-beta3-20110606.jar
 *               poi-scratchpad-3.8-beta3-20110606.jar
 *               xmlbeans-2.3.0.jar
 *               dom4j-1.6.1.jar
 *               jar包官網下載地址：http://poi.apache.org/download.html
 *               下載poi-bin-3.8-beta3-20110606.zipp
 * 
 * @作者：建寧
 * @時間：2012-08-29 下午16:27:15
 */
public class ImportExecl
{

    /** 總行數 */
    private int totalRows = 0;

    /** 總列數 */
    private int totalCells = 0;

    /** 錯誤信息 */
    private String errorInfo;

    /** 構造方法 */
    public ImportExecl()
    {

    }

    /**
     * @描述：得到總行數
     * @作者：建寧
     * @時間：2012-08-29 下午16:27:15
     * @參數：@return
     * @返回值：int
     */
    public int getTotalRows()
    {
        return totalRows;
    }

    /**
     * @描述：得到總列數
     * @作者：建寧
     * @時間：2012-08-29 下午16:27:15
     * @參數：@return
     * @返回值：int
     */
    public int getTotalCells()
    {
        return totalCells;
    }

    /**
     * @描述：得到錯誤信息
     * @作者：建寧
     * @時間：2012-08-29 下午16:27:15
     * @參數：@return
     * @返回值：String
     */
    public String getErrorInfo()
    {
        return errorInfo;
    }

    /**
     * @描述：驗證excel文件
     * @作者：建寧
     * @時間：2012-08-29 下午16:27:15
     * @參數：@param filePath　文件完整路徑
     * @參數：@return
     * @返回值：boolean
     */
    public boolean validateExcel(String filePath)
    {
        /** 檢查文件名是否為空或者是否是Excel格式的文件 */

        if (filePath == null || !(WDWUtil.isExcel2003(filePath) || WDWUtil.isExcel2007(filePath)))
        {
            errorInfo = "文件名不是excel格式";
            return false;
        }

        /** 檢查文件是否存在 */
        File file = new File(filePath);
        if (file == null || !file.exists())
        {
            errorInfo = "文件不存在";
            return false;
        }
        return true;
    }

    /**
     * @描述：根據文件名讀取excel文件
     * @作者：建寧
     * @時間：2012-08-29 下午16:27:15
     * @參數：@param filePath 文件完整路徑
     * @參數：@return
     * @返回值：List
     */
    public List<List<String>> read(String filePath)
    {
        List<List<String>> dataLst = new ArrayList<List<String>>();
        InputStream is = null;
        try
        {
            /** 驗證文件是否合法 */
            if (!validateExcel(filePath))
            {
                System.out.println(errorInfo);
                return null;
            }

            /** 判斷文件的類型，是2003還是2007 */
            boolean isExcel2003 = true;
            if (WDWUtil.isExcel2007(filePath))
            {
                isExcel2003 = false;
            }

            /** 調用本類提供的根據流讀取的方法 */
            File file = new File(filePath);
            is = new FileInputStream(file);
            dataLst = read(is, isExcel2003);
            is.close();
        }
        catch (Exception ex)
        {
            ex.printStackTrace();
        }
        finally
        {
            if (is != null)
            {
                try
                {
                    is.close();
                }
                catch (IOException e)
                {
                    is = null;
                    e.printStackTrace();
                }
            }
        }

        /** 返回最后讀取的結果 */
        return dataLst;
    }

    /**
     * @描述：根據流讀取Excel文件
     * @作者：建寧
     * @時間：2012-08-29 下午16:40:15
     * @參數：@param inputStream
     * @參數：@param isExcel2003
     * @參數：@return
     * @返回值：List
     */
    public List<List<String>> read(InputStream inputStream, boolean isExcel2003)
    {
        List<List<String>> dataLst = null;
        try
        {
            /** 根據版本選擇創建Workbook的方式 */
            Workbook wb = null;
            if (isExcel2003)
            {
                wb = new HSSFWorkbook(inputStream);
            }
            else
            {
                wb = new XSSFWorkbook(inputStream);
            }
            dataLst = read(wb);
        }
        catch (IOException e)
        {
            e.printStackTrace();
        }
        return dataLst;
    }

    /**
     * @描述：讀取數據
     * @作者：建寧
     * @時間：2012-08-29 下午16:50:15
     * @參數：@param Workbook
     * @參數：@return
     * @返回值：List<List<String>>
     */
    private List<List<String>> read(Workbook wb)
    {
        List<List<String>> dataLst = new ArrayList<List<String>>();
        
        /** 得到第一個shell */
        Sheet sheet = wb.getSheetAt(0);

        /** 得到Excel的行數 */
        this.totalRows = sheet.getPhysicalNumberOfRows();

        /** 得到Excel的列數 */
        if (this.totalRows >= 1 && sheet.getRow(0) != null)
        {
            this.totalCells = sheet.getRow(0).getPhysicalNumberOfCells();
        }

        /** 循環Excel的行 */
        for (int r = 0; r < this.totalRows; r++)
        {
            Row row = sheet.getRow(r);
            if (row == null)
            {
                continue;
            }
            List<String> rowLst = new ArrayList<String>();

            /** 循環Excel的列 */
            for (int c = 0; c < this.getTotalCells(); c++)
            {
                Cell cell = row.getCell(c);
                String cellValue = "";
                if (null != cell)
                {
                    // 以下是判斷數據的類型
                    switch (cell.getCellType())
                    {
                    case HSSFCell.CELL_TYPE_NUMERIC: // 數字
                        cellValue = cell.getNumericCellValue() + "";
                        break;

                    case HSSFCell.CELL_TYPE_STRING: // 字符串
                        cellValue = cell.getStringCellValue();
                        break;

                    case HSSFCell.CELL_TYPE_BOOLEAN: // Boolean
                        cellValue = cell.getBooleanCellValue() + "";
                        break;

                    case HSSFCell.CELL_TYPE_FORMULA: // 公式
                        cellValue = cell.getCellFormula() + "";
                        break;

                    case HSSFCell.CELL_TYPE_BLANK: // 空值
                        cellValue = "";
                        break;

                    case HSSFCell.CELL_TYPE_ERROR: // 故障
                        cellValue = "非法字符";
                        break;

                    default:
                        cellValue = "未知類型";
                        break;
                    }
                }
                rowLst.add(cellValue);
            }

            /** 保存第r行的第c列 */
            dataLst.add(rowLst);
        }
        return dataLst;
    }

    /**
     * @描述：main測試方法
     * @作者：建寧
     * @時間：2012-08-29 下午17:12:15
     * @參數：@param args
     * @參數：@throws Exception
     * @返回值：void
     */
    public static void main(String[] args) throws Exception
    {
        ImportExecl poi = new ImportExecl();
        // List<List<String>> list = poi.read("d:/aaa.xls");
        List<List<String>> list = poi.read("運單.xlsx");
        if (list != null)
        {
            for (int i = 0; i < list.size(); i++)
            {
                System.out.print("第" + (i) + "行");
                List<String> cellList = list.get(i);
                for (int j = 0; j < cellList.size(); j++)
                {
                    // System.out.print("    第" + (j + 1) + "列值：");
                    System.out.print("    " + cellList.get(j));
                }
                System.out.println();
            }
        }
    }
}

/**
 * @描述：工具類
 * @作者：建寧
 * @時間：2012-08-29 下午16:30:40
 */
class WDWUtil
{

    /**
     * @描述：是否是2003的excel，返回true是2003
     * @作者：建寧
     * @時間：2012-08-29 下午16:29:11
     * @參數：@param filePath　文件完整路徑
     * @參數：@return
     * @返回值：boolean
     */
    public static boolean isExcel2003(String filePath)
    {
        return filePath.matches("^.+\\.(?i)(xls)$");
    }

    /**
     * 
     * @描述：是否是2007的excel，返回true是2007
     * @作者：建寧
     * @時間：2012-08-29 下午16:28:20
     * @參數：@param filePath　文件完整路徑
     * @參數：@return
     * @返回值：boolean
     */
    public static boolean isExcel2007(String filePath)
    {
        return filePath.matches("^.+\\.(?i)(xlsx)$");
    }
}

 

2、讀取調用的代碼：

ImportExecl poi = new ImportExecl();
List<List<String>> list = poi.read("運單.xlsx");

 

3、生成Excel的代碼：

    /**
     * 創建一個Excel
     * @param fileName 文件名
     * @param dataList 數據
     * @throws IOException
     */
    public static void buildXLSX(String fileName, List<String[]> dataList) {
        try
        {
            // 聲明一個工作薄
            XSSFWorkbook workBook = null;
            workBook = new XSSFWorkbook();
            // 生成一個表格
            XSSFSheet sheet = workBook.createSheet();
            workBook.setSheetName(0,"info");
   
            //插入需導出的數據
            for(int i=0;i<dataList.size();i++){
                XSSFRow row = sheet.createRow(i);
                String[] oneRowData =  dataList.get(i);
                for(int j=0;j<oneRowData.length;j++)
                {
                    row.createCell(j).setCellValue(oneRowData[j]);
                }
            }
            File  file = new File("輸出\\"+fileName);
            //文件輸出流
            FileOutputStream outStream = new FileOutputStream(file);
            workBook.write(outStream);
            outStream.flush();
            outStream.close();
        }
        catch (Exception e)
        {
            e.printStackTrace();
        }
    }