com.github.xiaoyuge5201.excel.ExcelUtils Maven / Gradle / Ivy
package com.github.xiaoyuge5201.excel;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.ss.usermodel.*;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.InputStream;
import java.util.HashMap;
import java.util.Map;
/**
* excel读取类：兼容xls和xlsx
*
* @author yugb
*/
public class ExcelUtils {
private Workbook wb;
private Sheet sheet;
private Row row;
public ExcelUtils(String filepath) {
if (filepath == null) {
return;
}
String ext = filepath.substring(filepath.lastIndexOf("."));
try {
InputStream is = new FileInputStream(filepath);
if (".xls".equals(ext)) {
wb = new HSSFWorkbook(is);
} else if (".xlsx".equals(ext)) {
wb = new XSSFWorkbook(is);
} else {
wb = null;
}
} catch (FileNotFoundException e) {
e.printStackTrace();
} catch (IOException e) {
e.printStackTrace();
}
}
/**
* 读取Excel表格表头的内容
*
* @return String 表头内容的数组
*/
public String[] readExcelTitle() throws Exception {
if (wb == null) {
throw new Exception("Workbook对象为空！");
}
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] = row.getCell(i).getCellFormula();
}
return title;
}
/**
* 读取Excel数据内容
*
* @return Map 包含单元格数据内容的Map对象
*/
public Map> readExcelContent() throws Exception {
if (wb == null) {
throw new Exception("Workbook对象为空！");
}
Map> content = new HashMap>();
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;
Map cellValue = new HashMap();
while (j < colNum) {
Object obj = getCellFormatValue(row.getCell(j));
cellValue.put(j, obj);
j++;
}
content.put(i, cellValue);
}
return content;
}
/**
* 根据Cell类型设置数据
*
* @param cell 单元格
* @return
*/
private Object getCellFormatValue(Cell cell) {
Object cellvalue = "";
if (cell != null) {
// 判断当前Cell的Type
switch (cell.getCellType()) {
case Cell.CELL_TYPE_NUMERIC:// 如果当前Cell的Type为NUMERIC
case Cell.CELL_TYPE_FORMULA: {
// 判断当前的cell是否为Date
if (DateUtil.isCellDateFormatted(cell)) {
// 如果是Date类型则，转化为Data格式
// data格式是带时分秒的：2013-7-10 0:00:00 cell.getDateCellValue().toLocaleString();
// data格式是不带带时分秒的：2013-7-10 cell.getDateCellValue();
cellvalue = cell.getDateCellValue();
} else {// 如果是纯数字
// 取得当前Cell的数值
cellvalue = String.valueOf(cell.getNumericCellValue());
}
break;
}
case Cell.CELL_TYPE_STRING:// 如果当前Cell的Type为STRING
// 取得当前的Cell字符串
cellvalue = cell.getRichStringCellValue().getString();
break;
default:// 默认的Cell值
cellvalue = "";
}
} else {
cellvalue = "";
}
return cellvalue;
}
public static void main(String[] args) {
try {
String filepath = "C:\\Users\\yugb\\Desktop\\01808011003.xlsx";
ExcelUtils excelReader = new ExcelUtils(filepath);
// 读取Excel
Map> map = excelReader.readExcelContent();
System.out.println(map.get(0));
System.out.println("获得Excel表格的内容:");
for (int i = 1; i <= map.size(); i++) {
map.get(i).get(0).toString();
System.out.println(map.get(i).get(0));
}
} catch (FileNotFoundException e) {
System.out.println("未找到指定路径的文件!");
e.printStackTrace();
} catch (Exception e) {
e.printStackTrace();
}
}
}