com.wudaosoft.reports.excel.ExcelReader Maven / Gradle / Ivy
/**
* Copyright 2009-2017 Wudao Software Studio(wudaosoft.com)
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package com.wudaosoft.reports.excel;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.lang.reflect.Field;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.math.BigDecimal;
import java.text.ParseException;
import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.Collections;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import jxl.Cell;
import jxl.Sheet;
import jxl.Workbook;
import jxl.read.biff.BiffException;
/**
* @author changsoul.wu
*
*/
public class ExcelReader {
private List columns = new ArrayList(0);
private Map columnMap = new HashMap();
public List readExcel(String filePath, Class clazz) throws BiffException, IOException, InstantiationException, IllegalAccessException {
return readExcel(new FileInputStream(filePath), clazz);
}
public List readExcel(File file, Class clazz) throws BiffException, IOException, InstantiationException, IllegalAccessException {
return readExcel(new FileInputStream(file), clazz);
}
public List readExcel(InputStream stream, Class clazz) throws BiffException, IOException, InstantiationException, IllegalAccessException {
columns.clear();
columnMap.clear();
// 获取Excel文件对象
Workbook rwb = Workbook.getWorkbook(stream);
// 获取文件的指定工作表 默认的第一个
Sheet sheet = rwb.getSheet(0);
// 行数(表头的目录不需要，从1开始)
int rows = sheet.getRows();
int cols = sheet.getColumns();
if(rows <= 1 || cols == 0)
return Collections.emptyList();
processAnnotation(clazz);
for(int i = 0; i < cols; i++) {
Cell cell = sheet.getCell(i, 0);
if(cell == null || cell.getContents() == null)
continue;
for(Column column : columns) {
if(column.getName().equals(cell.getContents().trim()))
columnMap.put(i, column);
}
}
List list = new ArrayList();
for (int i = 1; i < rows; i++) {
T obj = clazz.newInstance();
// 列数
for (int j = 0; j < cols; j++) {
Column column = columnMap.get(j);
if(column == null)
continue;
Cell cell = sheet.getCell(j, i);
String val = "";
if(cell != null)
val = cell.getContents();
try {
Method method = column.getSetMethod();
if(method == null)
continue;
String type = method.getParameterTypes()[0].getName();
if (type.equals("java.lang.String")) {
column.getSetMethod().invoke(obj, val);
} else if(type.equals("java.lang.Integer")) {
column.getSetMethod().invoke(obj, Integer.valueOf(val.trim()));
} else if(type.equals("java.lang.Float")) {
column.getSetMethod().invoke(obj, Float.valueOf(val.trim()));
} else if(type.equals("java.lang.Double")) {
column.getSetMethod().invoke(obj, Double.valueOf(val.trim()));
} else if(type.equals("java.lang.Long")) {
column.getSetMethod().invoke(obj, Long.valueOf(val.trim()));
} else if(type.equals("java.lang.Short")) {
column.getSetMethod().invoke(obj, Short.valueOf(val.trim()));
} else if(type.equals("java.util.Date")) {
column.getSetMethod().invoke(obj, new SimpleDateFormat(column.getPattern()).parse(val.trim()));
} else if(type.equals("java.math.BigDecimal")) {
column.getSetMethod().invoke(obj, new BigDecimal(val.trim()));
} else {
column.getSetMethod().invoke(obj, val);
}
} catch (IllegalArgumentException e) {
} catch (InvocationTargetException e) {
} catch (ParseException e) {
}
}
}
return list;
}
private void processAnnotation(Class clazz) {
Field[] fields = clazz.getDeclaredFields();
columns = new ArrayList(150);
for(Field field : fields) {
if(field.isAnnotationPresent(ExcelColumn.class)) {
ExcelColumn ec = field.getAnnotation(ExcelColumn.class);
Column c = new Column();
c.setName(ec.name());
c.setField(field.getName());
c.setLength(ec.length());
c.setIndex(ec.index());
c.setScale(ec.scale());
c.setIsDateColumn(ec.isDate());
c.setIsNumber(ec.isNumber());
c.setIsMoneyColumn(ec.isMoney());
c.setIsHyperlink(ec.isHyperlink());
c.setPattern(ec.pattern());
c.setLinkText(ec.linkText());
c.setTrueText(ec.trueText());
c.setFalseText(ec.falseText());
c.setRoundingMode(ec.roundingMode());
c.setAlign(ec.align());
Method m = null;
try {
m = clazz.getMethod(getSetMethodName(field));
} catch (SecurityException e) {
} catch (NoSuchMethodException e) {
}
if(m == null) {
try {
m = clazz.getMethod(field.getName());
} catch (SecurityException e) {
} catch (NoSuchMethodException e) {
}
}
c.setSetMethod(m);
columns.add(c);
}
}
}
private String getSetMethodName(Field field) {
byte[] items = field.getName().getBytes();
items[0] = (byte) ((char) items[0] - 'a' + 'A');
return "set" + new String(items);
}
}