poi做Excel数据驱动,支持.xls和.xlsx格式的excel文档,比起jxl强大不少
2016-10-26 09:40
525 查看
import java.io.FileInputStream;
import java.io.InputStream;
import java.util.Iterator;
import java.util.LinkedHashMap;
import java.util.Map;
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.ss.usermodel.WorkbookFactory;
import org.testng.Assert;
public class PoiExcelProvider implements Iterator<Object[]> {
private Workbook book = null;
private Sheet sheet = null;
private int rowNum = 0;
private int currentRowNo = 0;
private short columnNum = 0;
private String[] columnnName;
/**获取第一行的所有单元格里面的类容:即key值*/
public PoiExcelProvider(String filePath,String sheetName) {
try {
InputStream inputStream = new FileInputStream(filePath);
book = WorkbookFactory.create(inputStream);
//取sheet
sheet = book.getSheet(sheetName);
rowNum = sheet.getLastRowNum()+1;//获取最后一行的行号,即获取一共有多少行
//获取当前sheet的第一行数据
Row row = sheet.getRow(0);
//第一行数据的一共有多少个单元格
columnNum = row.getLastCellNum();
//声明一个数组,长度为第一行所有单元格的长度
columnnName = new String[columnNum];
for (int i = 0; i < columnNum; i++) {
//获取第一行所有单元格的内容,存放到数组columnnName中
columnnName[i] = row.getCell(i).getStringCellValue();
}
this.currentRowNo++;//指向下一行
} catch (Exception e) {
e.printStackTrace();
Assert.fail("unable to read Excel data");
}
}
/*判断Excel是否还有内容*/
@Override
public boolean hasNext() {
//如果一行数据也没有或者当前行号超出最大行号,就关闭Excel
if (this.rowNum == 0 || this.currentRowNo >= this.rowNum) {
try {
book.close();
} catch (Exception e) {
e.printStackTrace();
}
return false;
} else {
// sheet下一行内容为空判定结束
if ((sheet.getRow(currentRowNo)).getCell(0).equals("")){
return false;
}
return true;
}
}
/**读取除key之外的Excel的所有内容*/
@Override
public Object[] next() {
//获取当前行对象
Row row = sheet.getRow(this.currentRowNo);
Map<String, String> data = new LinkedHashMap<String, String>();
for (int i = 0; i < this.columnNum; i++) {
String temp = "";
try {
temp = row.getCell(i).getStringCellValue();
} catch (ArrayIndexOutOfBoundsException ex) {
temp = "";
}
data.put(this.columnnName[i], temp);
}
Object object[] = new Object[1];
object[0] = data;
this.currentRowNo++;
return object;
}
@Override
public void remove() {
throw new UnsupportedOperationException("remove unsupported.");
}
}
import java.io.InputStream;
import java.util.Iterator;
import java.util.LinkedHashMap;
import java.util.Map;
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.ss.usermodel.WorkbookFactory;
import org.testng.Assert;
public class PoiExcelProvider implements Iterator<Object[]> {
private Workbook book = null;
private Sheet sheet = null;
private int rowNum = 0;
private int currentRowNo = 0;
private short columnNum = 0;
private String[] columnnName;
/**获取第一行的所有单元格里面的类容:即key值*/
public PoiExcelProvider(String filePath,String sheetName) {
try {
InputStream inputStream = new FileInputStream(filePath);
book = WorkbookFactory.create(inputStream);
//取sheet
sheet = book.getSheet(sheetName);
rowNum = sheet.getLastRowNum()+1;//获取最后一行的行号,即获取一共有多少行
//获取当前sheet的第一行数据
Row row = sheet.getRow(0);
//第一行数据的一共有多少个单元格
columnNum = row.getLastCellNum();
//声明一个数组,长度为第一行所有单元格的长度
columnnName = new String[columnNum];
for (int i = 0; i < columnNum; i++) {
//获取第一行所有单元格的内容,存放到数组columnnName中
columnnName[i] = row.getCell(i).getStringCellValue();
}
this.currentRowNo++;//指向下一行
} catch (Exception e) {
e.printStackTrace();
Assert.fail("unable to read Excel data");
}
}
/*判断Excel是否还有内容*/
@Override
public boolean hasNext() {
//如果一行数据也没有或者当前行号超出最大行号,就关闭Excel
if (this.rowNum == 0 || this.currentRowNo >= this.rowNum) {
try {
book.close();
} catch (Exception e) {
e.printStackTrace();
}
return false;
} else {
// sheet下一行内容为空判定结束
if ((sheet.getRow(currentRowNo)).getCell(0).equals("")){
return false;
}
return true;
}
}
/**读取除key之外的Excel的所有内容*/
@Override
public Object[] next() {
//获取当前行对象
Row row = sheet.getRow(this.currentRowNo);
Map<String, String> data = new LinkedHashMap<String, String>();
for (int i = 0; i < this.columnNum; i++) {
String temp = "";
try {
temp = row.getCell(i).getStringCellValue();
} catch (ArrayIndexOutOfBoundsException ex) {
temp = "";
}
data.put(this.columnnName[i], temp);
}
Object object[] = new Object[1];
object[0] = data;
this.currentRowNo++;
return object;
}
@Override
public void remove() {
throw new UnsupportedOperationException("remove unsupported.");
}
}
相关文章推荐
- 基于JXL和POI实现的读写Excel工具类(支持.xls和.xlsx格式)
- java导入和导出excel文档(支持xls,xlsx格式)
- Apache POI读写Excel文档入门(支持XLS和XLSX格式)
- POI使用:用poi接口不区分xls/xlsx格式解析Excel文档(41种日期格式解析方法,5种公式结果类型解析方法,3种常用数值类型精度控制办法)
- SSM+Maven+Bootstrap导入excel数据到数据库(支持XLSX,XLS格式)
- POI读取xls、xlsx(个人认为用JXL效果更好,但是JXL并不支持07版的Excel)
- java读写excel(POI,支持xls和xlsx两种格式)
- POI使用:用poi接口不区分xls/xlsx格式解析Excel文档(41种日期格式解析方法,5种公式结果类型解析方法,3种常用数值类型精度控制办法)
- java导入和导出excel文档(支持xls,xlsx格式)
- POI:支持xls/xlsx文件格式按cell类型解析相关内容(exls 2003/2007 兼容)
- 具有强大构建强大的Word报表,支持XAML数据绑定,文档格式转换,导出为PDF等功能的控件TX Text Control .NET for WPF介绍
- 【ASP.NET 进阶】仿百度文库文档在线预览(支持格式.pdf,.doc,docx,xls,xlsx,.ppt,pptx)
- java excel解析:poi与jxl的区别(excel版本问题:xls,xlsx)
- Java利用JXL&POI读取Excel(xls和xlsx)
- POI读取加密的EXCEL(兼容XLS,XLSX格式)
- poi导入导出excel文件,兼容.xls和.xlsx两种格式
- java 导入Excel 文件,兼容Excel 2003(后缀名:xls)及 2007(后缀名:xlsx)的文件,同时还支持csv格式的文件
- Java 读取Excel格式xls、xlsx数据工具类
- POI使用SAX处理大量数据的xlsx格式的Excel文件
- POI 3.8读取2003与2007格式EXCEL(xls、xlsx)