package cn.com.agree.poi.server.util;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.text.DecimalFormat;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;
import org.apache.poi.hssf.usermodel.HSSFDateUtil;
import org.apache.poi.hssf.usermodel.HSSFFormulaEvaluator;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.ss.usermodel.Cell;
import org.apache.poi.ss.usermodel.CellValue;
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.XSSFFormulaEvaluator;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
public class POIExcelUtil
{
//总行数
private int totalRows = 0;
//总列数
private int totalCells = 0;
//构造方法
public POIExcelUtil() {
}
/**
*
* Description:[根据文件名读取excel文件]</li>
* Created by [Huyvanpull] [Jan 20, 2010]</li>
* Midified by [modifier] [modified time]</li>
* @param fileName
* @return
* @throws Exception
*/
public List<ArrayList<String>> read(String fileName)
{
List<ArrayList<String>> dataLst = new ArrayList<ArrayList<String>>();
//检查文件名是否为空或者是否是Excel格式的文件
if (fileName == null || !fileName.matches("^.+\\.(?i)((xls)|(xlsx))$"))
{
System.out.println("非excel文档不能转换");
throw new RuntimeException("非excel文档不能转换");
}
boolean isExcel2003 = true;
// 对文件的合法性进行验证
if (fileName.matches("^.+\\.(?i)(xlsx)$"))
{
isExcel2003 = false;
}
//检查文件是否存在
File file = new File(fileName);
if (file == null || !file.exists())
{
return dataLst;
}
try
{
//调用本类提供的根据流读取的方法
dataLst = read(new FileInputStream(file), isExcel2003);
}
catch (Exception ex)
{
// ex.printStackTrace();
throw new RuntimeException(ex.getMessage());
}
//返回最后读取的结果
return dataLst;
}
/**
* <ul>
* <li>Description:[根据流读取Excel文件]</li>
* <li>Created by [Huyvanpull] [Jan 20, 2010]</li>
* <li>Midified by [modifier] [modified time]</li>
* <ul>
*
* @param inputStream
* @param isExcel2003
* @return
*/
public List<ArrayList<String>> read(InputStream inputStream,
boolean isExcel2003)
{
List<ArrayList<String>> dataLst = null;
try
{
//根据版本选择创建Workbook的方式
Workbook wb = isExcel2003 ? new HSSFWorkbook(inputStream)
: new XSSFWorkbook(inputStream);
dataLst = read(wb);
}
catch (Exception e)
{
// e.printStackTrace();
throw new RuntimeException(e.getMessage());
}
return dataLst;
}
/**
* <ul>
* <li>Description:[得到总行数]</li>
* <li>Created by [Huyvanpull] [Jan 20, 2010]</li>
* <li>Midified by [modifier] [modified time]</li>
* <ul>
*
* @return
*/
public int getTotalRows()
{
return totalRows;
}
/**
* <ul>
* <li>Description:[得到总列数]</li>
* <li>Created by [Huyvanpull] [Jan 20, 2010]</li>
* <li>Midified by [modifier] [modified time]</li>
* <ul>
*
* @return
*/
public int getTotalCells()
{
return totalCells;
}
/**
* <ul>
* <li>Description:[读取数据]</li>
* <li>Created by [Huyvanpull] [Jan 20, 2010]</li>
* <li>Midified by [modifier] [modified time]</li>
* <ul>
*
* @param wb
* @return
*/
private List<ArrayList<String>> read(Workbook wb)
{
List<ArrayList<String>> dataLst = new ArrayList<ArrayList<String>>();
//得到第一个shell
Sheet sheet = wb.getSheetAt(0);
this.totalRows = sheet.getPhysicalNumberOfRows();
this.totalCells = 0;
for (int r = 0; r < this.totalRows; r++)
{
Row row = sheet.getRow(r);
if (row == null)
{
continue;
}
if(row.getLastCellNum()>this.totalCells)
this.totalCells = row.getLastCellNum();
}
//循环Excel的行 */
for (int r = 0; r < this.totalRows; r++)
{
Row row = sheet.getRow(r);
if (row == null)
{
continue;
}
ArrayList<String> rowLst = new ArrayList<String>();
//循环Excel的列
for (short c = 0; c < this.getTotalCells(); c++)
{
Cell cell = row.getCell(c);
String cellValue = "";
if (cell == null)
{
rowLst.add(cellValue);
continue;
}
//处理数字型的,自动去零 */
if (Cell.CELL_TYPE_NUMERIC == cell.getCellType())
{
//在excel里,日期也是数字,在此要进行判断 */
if (HSSFDateUtil.isCellDateFormatted(cell))
{
Date date = cell.getDateCellValue();
cellValue = (date.getYear() + 1900) + "-" + (date.getMonth() + 1)
+ "-" + date.getDate()+" "+date.getHours()+":"+date.getMinutes()+":"+date.getSeconds() ;
}
else
{
cellValue = getRightStr(cell.getNumericCellValue() + "");
}
}
//处理字符串型 */
else if (Cell.CELL_TYPE_STRING == cell.getCellType())
{
cellValue = cell.getStringCellValue();
}
//处理布尔型 */
else if (Cell.CELL_TYPE_BOOLEAN == cell.getCellType())
{
cellValue = cell.getBooleanCellValue() + "";
}
else if (Cell.CELL_TYPE_FORMULA == cell.getCellType())
{
if(wb instanceof HSSFWorkbook) {
HSSFFormulaEvaluator hss = new HSSFFormulaEvaluator((HSSFWorkbook) wb);
try {
cellValue = hss.evaluate(cell).getNumberValue()+"";
} catch(Exception e) {
cellValue = hss.evaluate(cell).getStringValue()+"";
}
} else if(wb instanceof XSSFWorkbook){
XSSFFormulaEvaluator xss = new XSSFFormulaEvaluator((XSSFWorkbook) wb);
try {
cellValue = xss.evaluate(cell).getNumberValue()+"";
} catch(Exception e) {
cellValue = xss.evaluate(cell).getStringValue()+"";
}
}
}
//其它的,非以上几种数据类型 */
else
{
cellValue = cell.toString() + "";
}
rowLst.add(cellValue);
}
dataLst.add(rowLst);
}
return dataLst;
}
/**
* <ul>
* <li>Description:[正确地处理整数后自动加零的情况]</li>
* <li>Created by [Huyvanpull] [Jan 20, 2010]</li>
* <li>Midified by [modifier] [modified time]</li>
* <ul>
*
* @param sNum
* @return
*/
private String getRightStr(String sNum)
{
DecimalFormat decimalFormat = new DecimalFormat("#.00");
String resultStr = decimalFormat.format(new Double(sNum));
if (resultStr.matches("^[-+]?\\d+\\.[0]+$"))
{
resultStr = resultStr.substring(0, resultStr.indexOf("."));
}
return resultStr;
}
/**
* <ul>
* <li>Description:[测试main方法]</li>
* <li>Created by [Huyvanpull] [Jan 20, 2010]</li>
* <li>Midified by [modifier] [modified time]</li>
* <ul>
*
* @param args
* @throws Exception
*/
public static void main(String[] args) throws Exception
{
// String s = "";
// List<ArrayList<String>> dataLst = new POIExcelUtil()
// .read("E:\\test.xls");
// FileOutputStream fos = new FileOutputStream("c:\\d.txt");
// for (ArrayList<String> innerLst : dataLst)
// {
// StringBuffer rowData = new StringBuffer();
// for (String dataStr : innerLst)
// {
// rowData.append("\t").append(dataStr);
// }
// if (rowData.length() > 0)
// {
//// System.out.println(rowData.deleteCharAt(0).toString());
//// writ2Txt(rowData.deleteCharAt(0).toString()+"\r",fos);
// s = rowData.deleteCharAt(0).toString()+"\r\n";
// System.out.print(s);
// fos.write(s.getBytes());
// }
// }
// fos.close();
// System.out.println("OK");
}
/**
* 数据写入txt文档
* @throws IOException
*/
public static void writ2Txt(String s,FileOutputStream fos) throws IOException{
try {
fos.write(s.getBytes());
} catch (Exception e) {
e.printStackTrace();
}
}
}