开博第一篇: 一个Excel转换Java集合的工具类

一个把普通Excel文件转换成集合的工具类, 第一次使用POI,代码有些稚嫩,已经经过了测试, 现在的版本不支持Excel的公式, 只能针对数字和字符串。


package com.tarena.crm.util;

import java.io.FileInputStream;
import java.lang.reflect.InvocationTargetException;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Iterator;
import java.util.Properties;

import org.apache.commons.beanutils.BeanUtils;
import org.apache.log4j.Logger;
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 com.tarena.crm.entity.ConsulterProxy;

public class ExcelToCollectionUtil {
/**
* Logger for this class
*/
private static final Logger logger = Logger
.getLogger(ExcelToCollectionUtil.class);

/**
* Excel和对象成员的映射集合.
*/
private Properties pro;

/**
* 列数
*/
private int count;
/**
* Excel封装对象.
*/
private HSSFWorkbook hb;

/**
* 坏数据静态异常类.
*/
public static class ExcelBadRecordException extends Exception {
private StringBuilder ms = new StringBuilder();

private boolean flag = false;

public void addMessage(String s) {
flag = true;
ms.append(s + "\n");
}

public ExcelBadRecordException() {
super();
}

@Override
public String getMessage() {
return ms.toString();
}

public boolean hasError() {
return flag;
}

}

private ExcelBadRecordException exception = new ExcelBadRecordException();

/**
* 集合中所存储的对象类型.
*/
private Class clazz;

public ExcelToCollectionUtil(Properties pro, HSSFWorkbook hb, Class clazz) {
super();
this.pro = pro;
this.hb = hb;
this.clazz = clazz;
count=pro.keySet().size();
}

/**
* 从Excel导入到Collection
*/
public Collection toCollection() {
Collection table = new ArrayList(); // 保存数据的集合
HSSFSheet sheet = hb.getSheetAt(0); // 得到第一个sheet , 应该改进成可以选择的.
HSSFRow head = sheet.getRow(0); // 得到第一行表头

for (int i = 1;; i++) {
HSSFRow row = sheet.getRow(i); // 得到一行记录

if (row == null) // 判断是否为最后记录, 应该判断记录是否有效.
break;
HSSFCell cell = row.getCell((short) 0);
if (cell == null
|| (cell.getCellType() == HSSFCell.CELL_TYPE_STRING
&& cell.getRichStringCellValue().getString().trim().equals(""))
|| cell.getCellType() == HSSFCell.CELL_TYPE_BLANK)
break; //判断每一行第一个单元格是否为有效数据.

Object object = null;
try {
object = this.exchangObjectFromRow(head, row); // 通过一条记录产生一个对象
} catch (Exception e) {
// 遇到无效的数据跳过
e.printStackTrace();
exception.addMessage("第" + i + "条记录格式有错误!"); // 将无效记录异常内容存在ExcelBadRecordException对象里
continue;
}
table.add(object); // 将产生的对象存入集合中
}
return table;
}

/**
* 抛出无效数据异常.
*
* @throws ExcelBadRecordException
*/
public void showException() throws ExcelBadRecordException {
if (exception.hasError()) {
throw exception;
}
}

/**
* 将一个row对象数据转换成javabean对象.
*
* @param head
* @param row
* @return
* @throws InvocationTargetException
* @throws IllegalAccessException
*/
private Object exchangObjectFromRow(HSSFRow head, HSSFRow row)
throws Exception {
Object object = this.newInstance(clazz);
for (int j = 0;j<count; j++) {
HSSFCell cell = row.getCell((short) j); // 得到一个单元格.
if(cell==null) //假如为空跳过.
continue;
String attribute = head.getCell((short) j).getRichStringCellValue()
.getString(); // 得到Excel中该列列名.
String property = pro.getProperty(attribute); // 从Properties中得到对应的成员名
Object value = null; // 成员对象
if (property == null || property.equals("") || attribute.equals("")) // 防止有没有对应的字段映射
continue;
switch (cell.getCellType()) { // 判断单元格的数据类型,做不同处理.
case HSSFCell.CELL_TYPE_STRING: // 判断是字符串类型
value = cell.getRichStringCellValue().getString(); // 得到对象的成员对象.
break;
case HSSFCell.CELL_TYPE_NUMERIC: // 判断是数字类型
if (HSSFDateUtil.isCellDateFormatted(cell)) { // 判断是标准日期类型(1996-1-1)
value = cell.getDateCellValue(); // 得到对象的成员对象.
} else {
value = new Double(cell.getNumericCellValue()).longValue();
}
break;
case HSSFCell.CELL_TYPE_BLANK: // 判断是空白类型
value = "";
break;
default:
logger.debug("第" + j + "列为非字符串和数字的类型,类型编号为"
+ cell.getCellType());
}

BeanUtils.setProperty(object, property, value);

}
if (logger.isDebugEnabled()) {
logger.debug(BeanUtils.describe(object));
}
return object;
}

/**
* 创建对象.
*
* @param clazz
* @return
*/
private Object newInstance(Class clazz) {
try {
return clazz.newInstance();
} catch (Exception e) {
logger.error(e.getMessage());
}
return null;
}

public static void main(String[] args) throws Exception {
// URL url=new URL("http://localhost:8080/crm/temp/text.xls");
// InputStream is=url.openStream();
FileInputStream is = new FileInputStream("e:\\project\\ConsulterTemplate.xls");

HSSFWorkbook book = new HSSFWorkbook(is);
Properties pro = new Properties();
pro.load(ExcelToCollectionUtil.class.getClassLoader()
.getResourceAsStream("ConsulterProxy.properties"));
ExcelToCollectionUtil util = new ExcelToCollectionUtil(pro, book,
ConsulterProxy.class);
Collection col = util.toCollection();
Iterator it = col.iterator();
while (it.hasNext()) {
ConsulterProxy p=(ConsulterProxy)it.next();
//System.out.println(p.getProvinceString());
}

util.showException();


}

}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值