- 使用poi可以让java对Excel表格进行解析表格转换为list集合
- 对查询出来的list进行导出表格等操作
引入依赖
<!-- 解析excel表格 -->
<dependency>
<groupId>org.apache.poi</groupId>
<artifactId>poi-ooxml</artifactId>
<version>3.17</version>
</dependency>
工具类
/**
* @author xy
* 2020.12.21下午
*/
public class ExcelUtil {
/**
* 解析校验表格头规则,可自定义
*/
private static final String CHECK = "姓名手机号一级行业二级行业 公司或店铺名地区详细地址";
/**
* 导入表格解析为map返回
*/
public static Map<Integer, List<String>> importExcel(File file) {
if (!file.exists()) {
throw new RuntimeException("文件不存在");
}
FileInputStream fis = null;
Workbook workBook = null;
try {
fis = new FileInputStream(file);
// 使用XSSFWorkbook
workBook = new XSSFWorkbook(fis);
// 获取第一个sheet(表格头)
Sheet sheet = workBook.getSheetAt(0);
//第一个参数表示行数 第二个List保存该行的cell(单元格)数据
Map<Integer, List<String>> map = new HashMap<>(8);
int i = 0;
//排除科学计数
DecimalFormat df = new DecimalFormat("0");
Row row1 = sheet.getRow(0);
short lastCellNum = row1.getLastCellNum();
StringBuilder sb = new StringBuilder();
for (Cell cell : row1) {
sb.append(cell.toString());
}
if (lastCellNum != 7 || !CHECK.equals(sb.toString())) {
return null;
}
for (Row row : sheet) {
map.put(i, new ArrayList<>());
// 遍历当前行的所有cell
for (Cell cell : row) {
switch (cell.getCellTypeEnum()) {
case STRING:
// 如果是字符串则保存
map.get(i).add(cell.getRichStringCellValue().getString());
break;
case NUMERIC:
//将数值转换为字符串
map.get(i).add(df.format(cell.getNumericCellValue()));
default:
break;
}
}
i++;
}
return map;
} catch (Exception e) {
e.printStackTrace();
} finally { //关流
if (fis != null) {
try {
fis.close();
} catch (IOException e) {
e.printStackTrace();
}
}
if (workBook != null) {
try {
workBook.close();
} catch (IOException e) {
e.printStackTrace();
}
}
}
return null;
}
/**
* 根据参数产生excel表格
*
* @param map 每行的数据,key是行数,value是当前行内容
* @param strArray 表格头名,要跟列数一致
* @param fileName 文件名 建议用年月日分秒
*/
public static String createExcel(Map<String, List<String>> map, String[] strArray, String fileName) {
Collection<List<String>> values = map.values();
for (List<String> value : values) {
if (value.size() != strArray.length) {
return "表格头与导出的数据列数不一致,请检查数据。";
}
}
if (!fileName.endsWith(".xls")) {
fileName = fileName + ".xls";
}
// 第一步,创建一个webbook,对应一个Excel文件
HSSFWorkbook wb = new HSSFWorkbook();
// 第二步,在webbook中添加一个sheet,对应Excel文件中的sheet
HSSFSheet sheet = wb.createSheet("sheet1");
sheet.setDefaultColumnWidth(20);// 默认列宽
// 第三步,在sheet中添加表头第0行,注意老版本poi对Excel的行数列数有限制short
HSSFRow row = sheet.createRow((int) 0);
// 第四步,创建单元格,并设置值表头 设置表头居中
HSSFCellStyle style = wb.createCellStyle();
style.setAlignment(HorizontalAlignment.CENTER);
//设置表格边框
style.setBorderBottom(BorderStyle.valueOf(BORDER_THIN)); //下边框
style.setBorderLeft(BorderStyle.valueOf(BORDER_THIN));//左边框
style.setBorderTop(BorderStyle.valueOf(BORDER_THIN));//上边框
style.setBorderRight(BorderStyle.valueOf(BORDER_THIN));//右边框
//设置背景色
style.setFillForegroundColor(HSSFColor.YELLOW.index);
style.setFillPattern(FillPatternType.forInt(SOLID_FOREGROUND));
// 添加excel title
HSSFCell cell = null;
for (int i = 0; i < strArray.length; i++) {
cell = row.createCell((short) i);
cell.setCellValue(strArray[i]);
cell.setCellStyle(style);
}
// 第五步,写入实体数据
int i = 0;
for (String str : map.keySet()) {
row = sheet.createRow((int) i + 1);
List<String> list = map.get(str);
// 第四步,创建单元格,并设置值和样式
for (int j = 0; j < strArray.length; j++) {
HSSFCell cell1 = row.createCell((short) j);
HSSFCellStyle cellStyle = wb.createCellStyle();
cellStyle.setAlignment(HorizontalAlignment.CENTER);
//设置表格边框
cellStyle.setBorderBottom(BorderStyle.valueOf(BORDER_THIN)); //下边框
cellStyle.setBorderLeft(BorderStyle.valueOf(BORDER_THIN));//左边框
cellStyle.setBorderTop(BorderStyle.valueOf(BORDER_THIN));//上边框
cellStyle.setBorderRight(BorderStyle.valueOf(BORDER_THIN));//右边框
cell1.setCellStyle(cellStyle);
cell1.setCellValue(list.get(j));
}
try {
// 第六步,将文件存到指定位置,这里的PathUtil.getFullRealPath()是获取文件保存的tomcat路径,可以自己写
FileOutputStream fout = new FileOutputStream(PathUtil.getFullRealPath() + fileName);
wb.write(fout);
fout.close();
} catch (Exception e) {
e.printStackTrace();
}
i++;
}
return PathUtil.getFullRealPath() + fileName;
}
}