com.beyondb.io.ExcelReader.java Source code

Introduction

Here is the source code for com.beyondb.io.ExcelReader.java

Source

/*
 * To change this license header, choose License Headers in Project Properties.
 * To change this template file, choose Tools | Templates
 * and open the template in the editor.
 */

package com.beyondb.io;

import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.InputStream;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.HashMap;
import java.util.Map;
import org.apache.poi.hssf.usermodel.HSSFCell;
import org.apache.poi.hssf.usermodel.HSSFDateUtil;
import org.apache.poi.hssf.usermodel.HSSFRow;
import org.apache.poi.hssf.usermodel.HSSFSheet;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.poifs.filesystem.POIFSFileSystem;

/**
 * ?Excel
 */
public class ExcelReader {
    private POIFSFileSystem fs;
    private HSSFWorkbook wb;
    private HSSFSheet sheet;
    private HSSFRow row;

    /**
     * ?Excel
     * @param InputStream
     * @return String 
     */
    public String[] readExcelTitle(InputStream is) {
        try {
            fs = new POIFSFileSystem(is);
            wb = new HSSFWorkbook(fs);
        } catch (IOException e) {
            e.printStackTrace();
        }
        sheet = wb.getSheetAt(0);
        row = sheet.getRow(0);
        // 
        int colNum = row.getPhysicalNumberOfCells();
        System.out.println("colNum:" + colNum);
        String[] title = new String[colNum];
        for (int i = 0; i < colNum; i++) {
            //title[i] = getStringCellValue(row.getCell((short) i));
            title[i] = getCellFormatValue(row.getCell(i));
        }
        return title;
    }

    /**
     * ?Excel?
     * @param InputStream
     * @return Map ???Map
     */
    public Map<Integer, String> readExcelContent(InputStream is) {
        Map<Integer, String> content = new HashMap<Integer, String>();
        String str = "";
        try {
            fs = new POIFSFileSystem(is);
            wb = new HSSFWorkbook(fs);
        } catch (IOException e) {
            e.printStackTrace();
        }
        sheet = wb.getSheetAt(0);
        // 
        int rowNum = sheet.getLastRowNum();
        row = sheet.getRow(0);
        int colNum = row.getPhysicalNumberOfCells();
        // ,
        for (int i = 1; i <= rowNum; i++) {
            row = sheet.getRow(i);
            int j = 0;
            while (j < colNum) {
                // ???"-"??Stringreplace()?
                // ????javabean?javabean
                // str += getStringCellValue(row.getCell((short) j)).trim() +
                // "-";
                str += getCellFormatValue(row.getCell(j)).trim() + "    ";
                j++;
            }
            content.put(i, str);
            str = "";
        }
        return content;
    }

    /**
     * ????
     * 
     * @param cell Excel?
     * @return String ??
     */
    private String getStringCellValue(HSSFCell cell) {
        String strCell = "";
        switch (cell.getCellType()) {
        case HSSFCell.CELL_TYPE_STRING:
            strCell = cell.getStringCellValue();
            break;
        case HSSFCell.CELL_TYPE_NUMERIC:
            strCell = String.valueOf(cell.getNumericCellValue());
            break;
        case HSSFCell.CELL_TYPE_BOOLEAN:
            strCell = String.valueOf(cell.getBooleanCellValue());
            break;
        case HSSFCell.CELL_TYPE_BLANK:
            strCell = "";
            break;
        default:
            strCell = "";
            break;
        }
        if (strCell.equals("") || strCell == null) {
            return "";
        }
        if (cell == null) {
            return "";
        }
        return strCell;
    }

    //    /**
    //     * ????
    //     * 
    //     * @param cell
    //     *            Excel?
    //     * @return String ??
    //     */
    //    private String getDateCellValue(HSSFCell cell) {
    //        String result = "";
    //        try {
    //            int cellType = cell.getCellType();
    //            if (cellType == HSSFCell.CELL_TYPE_NUMERIC) {
    //                Date date = cell.getDateCellValue();
    //                result = (date.getYear() + 1900) + "-" + (date.getMonth() + 1)
    //                        + "-" + date.getDate();
    //            } else if (cellType == HSSFCell.CELL_TYPE_STRING) {
    //                String date = getStringCellValue(cell);
    //                result = date.replaceAll("[]", "-").replace("", "").trim();
    //            } else if (cellType == HSSFCell.CELL_TYPE_BLANK) {
    //                result = "";
    //            }
    //        } catch (Exception e) {
    //            System.out.println("??!");
    //            e.printStackTrace();
    //        }
    //        return result;
    //    }

    /**
     * ?HSSFCell?
     * @param cell
     * @return
     */
    private String getCellFormatValue(HSSFCell cell) {
        String cellvalue = "";
        if (cell != null) {
            // ?CellType
            switch (cell.getCellType()) {
            // ?CellTypeNUMERIC
            case HSSFCell.CELL_TYPE_NUMERIC:
            case HSSFCell.CELL_TYPE_FORMULA: {
                // ?cell?Date
                if (HSSFDateUtil.isCellDateFormatted(cell)) {
                    // DateData?

                    //1?data?2011-10-12 0:00:00
                    //cellvalue = cell.getDateCellValue().toLocaleString();

                    //2?data??2011-10-12
                    Date date = cell.getDateCellValue();
                    SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd");
                    cellvalue = sdf.format(date);

                }
                // 
                else {
                    // ??Cell
                    cellvalue = String.valueOf(cell.getNumericCellValue());
                }
                break;
            }
            // ?CellTypeSTRIN
            case HSSFCell.CELL_TYPE_STRING:
                // ??Cell
                cellvalue = cell.getRichStringCellValue().getString();
                break;
            // Cell
            default:
                cellvalue = " ";
            }
        } else {
            cellvalue = "";
        }
        return cellvalue;

    }

    public static void main(String[] args) {
        try {
            // ?Excel
            InputStream is = new FileInputStream("C:\\Users\\lbs\\Desktop\\test.xls");
            ExcelReader excelReader = new ExcelReader();
            String[] title = excelReader.readExcelTitle(is);
            System.out.println("Excel:");
            for (String s : title) {
                System.out.print(s + " ");
            }

            // ?Excel
            InputStream is2 = new FileInputStream("C:\\Users\\lbs\\Desktop\\test.xls");
            Map<Integer, String> map = excelReader.readExcelContent(is2);
            System.out.println("Excel:");
            for (int i = 1; i <= map.size(); i++) {
                System.out.println(map.get(i));
            }

        } catch (FileNotFoundException e) {
            System.out.println("!");
            e.printStackTrace();
        }
    }
}