package cn.com.css.common.util;

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;

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();

String[] title = new String[colNum];

for (int i = 0; i < colNum; i++) {

title[i] = getCellFormatValue(row.getCell((short) i));

}

return title;

}

/**

* 读取Excel数据内容

*

* @param InputStream

* @return Map 包含单元格数据内容的Map对象

*/

public Map readExcelContent(InputStream is) {

Map content = new HashMap();

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 colNum1 = row.getPhysicalNumberOfCells(); // 获取有数据的列的个数

int colNum = 15;

// 正文内容应该从第二行开始,第一行为表头的标题

for (int i = 1; i <= rowNum; i++) {

row = sheet.getRow(i);

int j = 0;

while (j < colNum) {

String cell = getCellFormatValue(row.getCell((short) j)).trim();

if (cell == "") {

j++;

continue;

}

str += cell + " ";

j++;

}

str.trim();

// System.out.println(str);

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) {

// 判断当前Cell的Type

switch (cell.getCellType()) {

// 如果当前Cell的Type为NUMERIC

case HSSFCell.CELL_TYPE_NUMERIC: {

cellvalue = String.valueOf(cell.getNumericCellValue());

break;

}

case HSSFCell.CELL_TYPE_FORMULA: {

// 判断当前的cell是否为Date

if (HSSFDateUtil.isCellDateFormatted(cell)) {

// 如果是Date类型则,转化为Data格式

// 方法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;

}

// 如果当前Cell的Type为STRIN

case HSSFCell.CELL_TYPE_STRING:

// 取得当前的Cell字符串

cellvalue = cell.getRichStringCellValue().getString();

break;

// 默认的Cell值

default:

cellvalue = " ";

}

} else {

cellvalue = "";

}

return cellvalue;

}

}

Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐