場景:
需要判斷傳入的excel 檔案 的列名是不是強一緻匹對, 那麼就需要拿出這些家夥
實作:
ExcelUtil.java
/**
* @Author: JCccc
* @Date: 2022-8-31 02:49
* @Description:
*/
public class ExcelUtil {
protected static final Logger logger = LoggerFactory.getLogger(ExcelUtil.class);
private static final String XLS = "xls";
private static final String XLSX = "xlsx";
private static final String SPLIT = ".";
public static Workbook getWorkbook(MultipartFile file) {
Workbook workbook=null;
try {
// 擷取Excel字尾名
String fileName = file.getOriginalFilename();
if (StringUtils.isEmpty(fileName) || fileName.lastIndexOf(SPLIT) < 0) {
logger.warn("解析Excel失敗,因為擷取到的Excel檔案名非法!");
return null;
}
String fileType = fileName.substring(fileName.lastIndexOf(SPLIT) + 1, fileName.length());
// 擷取Excel工作簿
if (fileType.equalsIgnoreCase(XLS)) {
workbook = new HSSFWorkbook(file.getInputStream());
} else if (fileType.equalsIgnoreCase(XLSX)) {
workbook = new XSSFWorkbook(file.getInputStream());
}
} catch (IOException e) {
e.printStackTrace();
}
return workbook;
}
public static List<String> getSheetTitles( Workbook workbook) {
//拿第一個sheet表
Sheet sheet = workbook.getSheetAt(0);
// 校驗sheet是否合法
if (Objects.isNull(sheet)) {
return null;
}
// 擷取第一行資料(假如第一行就是列名)
Row sheetTitleRow = sheet.getRow(sheet.getFirstRowNum());
// 取出最後一列
short lastCellNum = sheetTitleRow.getLastCellNum();
List<String> sheetTitleList = new LinkedList<>();
for (int i = 0; i < lastCellNum; i++) {
//取出每一列的名
String cellValue = sheetTitleRow.getCell(i).getStringCellValue();
sheetTitleList.add(cellValue);
}
return sheetTitleList;
}
}
使用:
Workbook workbook = ExcelUtil.getWorkbook(file);
if (Objects.nonNull(workbook)){
List<String> sheetTitlesFromFile = ExcelUtil.getSheetTitles(workbook);
}