|
|
@@ -1126,7 +1126,134 @@ public class NewExcelUtil<T> {
|
|
|
{
|
|
|
return importExcel(StringUtils.EMPTY, is, titleNum);
|
|
|
}
|
|
|
+//
|
|
|
+ public List<T> importExcelSet( InputStream is, int titleNum) throws Exception {
|
|
|
+ this.type = Excel.Type.IMPORT;
|
|
|
+ this.wb = WorkbookFactory.create(is);
|
|
|
+ List<T> list = new ArrayList<T>();
|
|
|
+ Iterator<Sheet> sheetIterator = wb.sheetIterator();
|
|
|
+ while (sheetIterator.hasNext()) {
|
|
|
+ Sheet sheet = sheetIterator.next();
|
|
|
+ if (sheet == null) {
|
|
|
+ throw new IOException("文件sheet不存在");
|
|
|
+ }
|
|
|
+ boolean isXSSFWorkbook = !(wb instanceof HSSFWorkbook);
|
|
|
+ Map<String, PictureData> pictures;
|
|
|
+ if (isXSSFWorkbook) {
|
|
|
+ pictures = getSheetPictures07((XSSFSheet) sheet, (XSSFWorkbook) wb);
|
|
|
+ } else {
|
|
|
+ pictures = getSheetPictures03((HSSFSheet) sheet, (HSSFWorkbook) wb);
|
|
|
+ }
|
|
|
+ // 获取最后一个非空行的行下标,比如总行数为n,则返回的为n-1
|
|
|
+ int rows = sheet.getLastRowNum();
|
|
|
+
|
|
|
+ if (rows > 0) {
|
|
|
+ // 定义一个map用于存放excel列的序号和field.
|
|
|
+ Map<String, Integer> cellMap = new HashMap<String, Integer>();
|
|
|
+ // 获取表头
|
|
|
+ Row heard = sheet.getRow(titleNum);
|
|
|
+ for (int i = 0; i < heard.getPhysicalNumberOfCells(); i++) {
|
|
|
+ Cell cell = heard.getCell(i);
|
|
|
+ if (StringUtils.isNotNull(cell)) {
|
|
|
+ String value = this.getCellValue(heard, i).toString();
|
|
|
+ cellMap.put(value, i);
|
|
|
+ } else {
|
|
|
+ cellMap.put(null, i);
|
|
|
+ }
|
|
|
+ }
|
|
|
+ // 有数据时才处理 得到类的所有field.
|
|
|
+ List<Object[]> fields = this.getFields();
|
|
|
+ Map<Integer, Object[]> fieldsMap = new HashMap<Integer, Object[]>();
|
|
|
+ for (Object[] objects : fields) {
|
|
|
+ Excel attr = (Excel) objects[1];
|
|
|
+ Integer column = cellMap.get(attr.name());
|
|
|
+ if (column != null) {
|
|
|
+ fieldsMap.put(column, objects);
|
|
|
+ }
|
|
|
+ }
|
|
|
+ int errorCount = 0;
|
|
|
+ try {
|
|
|
+ for (int i = titleNum + 1; i <= rows; i++) {
|
|
|
+ errorCount = i + 1;
|
|
|
+ // 从第2行开始取数据,默认第一行是表头.
|
|
|
+ Row row = sheet.getRow(i);
|
|
|
+ // 判断当前行是否是空行
|
|
|
+ if (isRowEmpty(row)) {
|
|
|
+ continue;
|
|
|
+ }
|
|
|
+ T entity = null;
|
|
|
+ for (Map.Entry<Integer, Object[]> entry : fieldsMap.entrySet()) {
|
|
|
+ Object val = this.getCellValue(row, entry.getKey());
|
|
|
+
|
|
|
+ // 如果不存在实例则新建.
|
|
|
+ entity = (entity == null ? clazz.newInstance() : entity);
|
|
|
+ // 从map中得到对应列的field.
|
|
|
+ Field field = (Field) entry.getValue()[0];
|
|
|
+ Excel attr = (Excel) entry.getValue()[1];
|
|
|
+ // 取得类型,并根据对象类型设置值.
|
|
|
+ Class<?> fieldType = field.getType();
|
|
|
+ if (String.class == fieldType) {
|
|
|
+ String s = Convert.toStr(val);
|
|
|
+ if (StringUtils.endsWith(s, ".0")) {
|
|
|
+ val = StringUtils.substringBefore(s, ".0");
|
|
|
+ } else {
|
|
|
+ String dateFormat = field.getAnnotation(Excel.class).dateFormat();
|
|
|
+ if (StringUtils.isNotEmpty(dateFormat)) {
|
|
|
+ val = parseDateToStr(dateFormat, val);
|
|
|
+ } else {
|
|
|
+ val = Convert.toStr(val);
|
|
|
+ }
|
|
|
+ }
|
|
|
+ } else if ((Integer.TYPE == fieldType || Integer.class == fieldType) && StringUtils.isNumeric(Convert.toStr(val))) {
|
|
|
+ val = Convert.toInt(val);
|
|
|
+ } else if ((Long.TYPE == fieldType || Long.class == fieldType) && StringUtils.isNumeric(Convert.toStr(val))) {
|
|
|
+ val = Convert.toLong(val);
|
|
|
+ } else if (Double.TYPE == fieldType || Double.class == fieldType) {
|
|
|
+ val = Convert.toDouble(val);
|
|
|
+ } else if (Float.TYPE == fieldType || Float.class == fieldType) {
|
|
|
+ val = Convert.toFloat(val);
|
|
|
+ } else if (BigDecimal.class == fieldType) {
|
|
|
+ val = Convert.toBigDecimal(val);
|
|
|
+ } else if (Date.class == fieldType) {
|
|
|
+ if (val instanceof String) {
|
|
|
+ val = DateUtils.parseDate((String) val);
|
|
|
+ } else if (val instanceof Double) {
|
|
|
+ val = DateUtil.getJavaDate((Double) val);
|
|
|
+ }
|
|
|
+ } else if (Boolean.TYPE == fieldType || Boolean.class == fieldType) {
|
|
|
+ val = Convert.toBool(val, false);
|
|
|
+ }
|
|
|
+ if (StringUtils.isNotNull(fieldType)) {
|
|
|
+ String propertyName = field.getName();
|
|
|
+ if (StringUtils.isNotEmpty(attr.targetAttr())) {
|
|
|
+ propertyName = field.getName() + "." + attr.targetAttr();
|
|
|
+ } else if (StringUtils.isNotEmpty(attr.readConverterExp())) {
|
|
|
+ val = reverseByExp(Convert.toStr(val), attr.readConverterExp(), attr.separator());
|
|
|
+ } else if (!attr.handler().equals(ExcelHandlerAdapter.class)) {
|
|
|
+ val = dataFormatHandlerAdapter(val, attr);
|
|
|
+ } else if (ColumnType.IMAGE == attr.cellType() && StringUtils.isNotEmpty((CharSequence) pictures)) {
|
|
|
+ PictureData image = pictures.get(row.getRowNum() + "_" + entry.getKey());
|
|
|
+ if (image == null) {
|
|
|
+ val = "";
|
|
|
+ } else {
|
|
|
+ byte[] data = image.getData();
|
|
|
+ val = FileUtils.writeImportBytes(data);
|
|
|
+ }
|
|
|
+ }
|
|
|
+ ReflectUtils.invokeSetter(entity, propertyName, val);
|
|
|
+ }
|
|
|
+ }
|
|
|
|
|
|
+ list.add(entity);
|
|
|
+ }
|
|
|
+ } catch (Exception e) {
|
|
|
+ throw new BusinessException(String.format("表格第%s行内容错误,请检查修改后再上传。", errorCount));
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ }
|
|
|
+ return list;
|
|
|
+ }
|
|
|
|
|
|
/**
|
|
|
* 对excel表单指定表格索引名转换成list
|