Java读取Excel表格到对象之优化
在前面,虽然可以从excel读取转化为对象,但是如果一个类的字段太多,每次都要通过set来设值,很浪费时间, 还有许多类似的类如果都要导入的话,为每个类设计一个方法显得太愚蠢,所以决定设计一个工具类,能够自动转换为想要的,通过反射和泛型来实现:
注意!在这里面默认的为表格的第一列对应类的第一个变量,如果对应相隔几个位置或者类有继承的变量,可以通过修改循环中的i和i与表格列的差来使用。
public <T> List<T> readExcel(File file, Class<T> c) throws Exception {
//获取文件名字
String fileName = file.getName();
//获取文件类型
String fileType = fileName.substring(fileName.lastIndexOf(".") + 1);
System.out.println(" **** fileType:" + fileType);
//获取输入流
InputStream stream = new FileInputStream(file);
//获取工作薄
Workbook xssfWorkbook = null;
if (fileType.equals("xls")) {
xssfWorkbook = new HSSFWorkbook(stream);
} else if (fileType.equals("xlsx")) {
xssfWorkbook = new XSSFWorkbook(stream);
} else {
System.out.println("您输入的excel格式不正确");
}
List<T> list = new ArrayList<T>();
// Read the Sheet
Sheet Sheet = xssfWorkbook.getSheetAt(0);
// Read the Row 从0开始
for (int rowNum = 0; rowNum <= Sheet.getLastRowNum(); rowNum++) {
Row Row = Sheet.getRow(rowNum);
if (Row != null) {
//判断这行记录是否存在
if (Row.getLastCellNum() < 1 || "".equals(getValue(Row.getCell(1)))) {
continue;
}
//获取每一行
//反射获取实例
T t = c.newInstance();
//获取私有变量
Field[] fields = c.getDeclaredFields();
//循环赋值
for(int i=0;i<fields.length-1;i++){
fields[i].setAccessible(true);
fields[i].set(t, getValue(Row.getCell(i)));
}
list.add(t);
}
}
return list;
}
private String getValue(Cell cell){
int type = CellFormat.ultimateType(cell);
if(type == Cell.CELL_TYPE_BOOLEAN)
{
return String.valueOf(cell.getBooleanCellValue());
}
else if(type == Cell.CELL_TYPE_NUMERIC)
{
return String.valueOf(cell.getNumericCellValue());
}
else if(type == Cell.CELL_TYPE_BLANK)
{
return "";
}
else
{
return cell.getStringCellValue().trim();
}
}
注意!在这里面默认的为表格的第一列对应类的第一个变量,如果对应相隔几个位置或者类有继承的变量,可以通过修改循环中的i和i与表格列的差来使用。