定义需要导入的EXCEL模板，EXCEL格式变更，只需要变更对应的枚举信息就可以，如果定义多个EXCEL模板，
只要定义多个枚举就行
package org.util.excel;
import lombok.AllArgsConstructor;
import lombok.Getter;
import org.apache.commons.lang3.StringUtils;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;
public class importExcelTitle {
@Getter
@AllArgsConstructor
public enum TaskimportExcel {
PUBLIC_DATA_ID("ID", "publicDataId", 0, true),
PUBLIC_DATA_NAME("名称", "publicDataName", 40, true),
TASK_USER_NAME("任务担当人", "taskUserName", 0, false),
END_TIME("任务到期日", "endTime", 0, false),
;
private final String titleCn;
private final String valueKey;
private final Integer titleLength;
private final Boolean isRequired;
public static Map getKeyValue() {
return Arrays.stream(TaskimportExcel.values()).collect(
Collectors.toMap(TaskimportExcel::getTitleCn, TaskimportExcel::getValueKey));
}
public static Map getExcelTitle() {
return Arrays.stream(TaskimportExcel.values()).collect(
Collectors.toMap(TaskimportExcel::getValueKey, title -> title));
}
public static List checkData(TaskimportExcel taskimportExcel, String value) {
return checkDataCommon(value, taskimportExcel.getIsRequired(), taskimportExcel.getTitleCn(), taskimportExcel.getTitleLength());
}
}
private static List checkDataCommon(String value, Boolean isRequired, String titleCn, Integer titleLength) {
List result = new ArrayList<>();
if (Boolean.TRUE.equals(isRequired) && StringUtils.isBlank(value)) {
result.add(String.format("%s信息未填", titleCn));
}
if (titleLength != 0 && value != null && value.length() > titleLength) {
result.add(String.format("%s过长", titleCn));
}
return result;
}
}
导入的excel处理
对导入的excel进行处理，按照上述importExcelTitle.java中的配置，将导入的excel转化成List
Map
package org.util.excel;
import lombok.extern.slf4j.Slf4j;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.ss.usermodel.*;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
import java.io.IOException;
import java.io.InputStream;
import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
@Slf4j
public class importExcelUtil {
private importExcelUtil() {
}
private static final String EXCEL_2003L = ".xls";
private static final String EXCEL_2007U = ".xlsx";
private static final String GENERAL = "General";
private static final String M_D_YY = "m/d/yy";
public static List> parseExcel(InputStream in, String fileName, Map titleKey) throws IOException {
// 根据文件名来创建Excel工作薄
Workbook work = getWorkbook(in, fileName);
Sheet sheet;
Row row;
Cell cell;
// 返回数据
List> ls = new ArrayList<>();
// 遍历Excel中所有的sheet
sheet = work.getSheetAt(0);
if (sheet == null) {
return ls;
}
// 取第一行标题
Row topRow = sheet.getRow(0);
if (null == topRow) {
return ls;
}
row = sheet.getRow(0);
String[] title;
if (row == null) {
return ls;
}
title = new String[row.getLastCellNum()];
for (int y = row.getFirstCellNum(); y < row.getLastCellNum(); y++) {
cell = row.getCell(y);
title[y] = (String) getCellValue(cell);
}
// 遍历当前sheet中的所有行
// 坑：sheet.getLastRowNum() 最后一行行标，比行数小1
for (int j = 1; j <= sheet.getLastRowNum(); j++) {
row = sheet.getRow(j);
Map m = new HashMap<>(row.getLastCellNum());
// 遍历所有的列
for (int y = row.getFirstCellNum(); y < row.getLastCellNum(); y++) {
cell = row.getCell(y);
String key = title[y].replace(" ", "");
if (titleKey.get(key) == null) {
log.error("导入excel出现错误的标题：" + key);
throw new RuntimeException("excel出现错误的标题：" + key);
}
m.put(titleKey.get(key), getCellValue(cell).toString().trim());
}
ls.add(m);
}
work.close();
return ls;
}
public static Workbook getWorkbook(InputStream inStr, String fileName) throws IOException {
Workbook wb;
String fileType = fileName.substring(fileName.lastIndexOf("."));
if (EXCEL_2003L.equals(fileType)) {
// 2003-
wb = new HSSFWorkbook(inStr);
} else if (EXCEL_2007U.equals(fileType)) {
// 2007+
wb = new XSSFWorkbook(inStr);
} else {
throw new RuntimeException("解析的文件格式有误！");
}
return wb;
}
public static Object getCellValue(Cell cell) {
Object value = new Object();
// 日期格式化
SimpleDateFormat sdf = new SimpleDateFormat("yyy-MM-dd");
if (cell == null) {
return "";
}
switch (cell.getCellType()) {
case STRING:
case FORMULA:
value = cell.getRichStringCellValue().getString();
break;
case NUMERIC:
if (GENERAL.equals(cell.getCellStyle().getDataFormatString())) {
double cellValue = cell.getNumericCellValue();
value = replace(String.valueOf(cellValue));
} else if (M_D_YY.equals(cell.getCellStyle().getDataFormatString())) {
value = sdf.format(cell.getDateCellValue());
} else {
DataFormatter formatter = new DataFormatter();
// 直接获取到单元格的值
value = formatter.formatCellValue(cell);
}
break;
case BOOLEAN:
value = cell.getBooleanCellValue();
break;
case BLANK:
value = "";
break;
default:
break;
}
return value;
}
public static String replace(String s) {
char a = '.';
if (s.indexOf(a) >= 0) {
//去掉多余的0
s = s.replaceAll("0+$", "");
//如最后一位是.则去掉
s = s.replaceAll("[.]$", "");
}
return s;
}
}
调用导入任务信息
测试调用， 拿到List
@ApiOperation(value = "导入任务信息")
@PostMapping("/importTask")
public void importTask(@RequestParam(value = "file") MultipartFile fileToUpload) {
List> excelList;
try {
excelList = importExcelUtil.parseExcel(
fileToUpload.getInputStream(),
fileToUpload.getOriginalFilename(),
importExcelTitle.TaskimportExcel.getKeyValue());
} catch (IOException e) {
log.error("解析excel时失败" + e.getMessage());
}
}
