一个导入excel的工具类,半自动,单元格读取部分需要自己手动实现
package com.home.util;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.ss.usermodel.*;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
import java.io.*;
import java.text.DecimalFormat;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
/**
* 导出excel,半自动
*
* @param <T>
*/
public abstract class UploadUtil<T> {
//如果抽象方法中需要操作中间变量可以直接使用
/**
* 实体对象
*/
public T t;
/**
* 单元格对象
*/
public Cell cell;
/**
* 处理上传的文件
*
* @param in
* @param fileName
* @return
* @throws Exception
*/
public List<T> getListByExcel(InputStream in, String fileName) throws Exception {
List<T> list = new ArrayList<>();
//创建Excel工作薄
Workbook work = this.getWorkbook(in, fileName);
if (null == work) {
throw new Exception("创建Excel工作薄为空!");
}
Sheet sheet = null;
Row row = null;
for (int i = 0; i < work.getNumberOfSheets(); i++) {
sheet = work.getSheetAt(i);
if (sheet == null) {
continue;
}
for (int j = sheet.getFirstRowNum(); j <= sheet.getLastRowNum(); j++) {
row = sheet.getRow(j);
if (row == null || row.getFirstCellNum() == j) {
continue;
}
list.add(buiderEntity(row));
}
}
work.close();
return list;
}
public abstract T buiderEntity(Row row);
/**
* 判断文件格式
*
* @param inStr
* @param fileName
* @return
* @throws Exception
*/
private Workbook getWorkbook(InputStream inStr, String fileName) throws Exception {
Workbook workbook = null;
String fileType = fileName.substring(fileName.lastIndexOf("."));
if (".xls".equals(fileType)) {
workbook = new HSSFWorkbook(inStr);
} else if (".xlsx".equals(fileType)) {
workbook = new XSSFWorkbook(inStr);
} else {
throw new RuntimeException("格式不对");
}
return workbook;
}
/**
* 输入流转file
*
* @param ins
* @param file
*/
public static void inputStreamToFile(InputStream ins, File file) {
try {
OutputStream os = new FileOutputStream(file);
int bytesRead = 0;
byte[] buffer = new byte[8192];
while ((bytesRead = ins.read(buffer, 0, 8192)) != -1) {
os.write(buffer, 0, bytesRead);
}
os.close();
ins.close();
} catch (Exception e) {
e.printStackTrace();
}
}
/**
* 获取单元格数据
*
* @param cell
* @return
*/
public static String getCellContent(Cell cell) {
return getCellContent(cell, null);
}
public static String getCellContent(Cell cell, DecimalFormat df) {
if (cell == null)
return "";
if (df == null) {
df = new DecimalFormat("####################");
}
if (cell.getCellType() == CellType.NUMERIC) {
double d = cell.getNumericCellValue();
return df.format(d);
} else if (cell.getCellType() == CellType.STRING) {
return cell.getStringCellValue();
} else if (cell.getCellType() == CellType.FORMULA) {
return cell.getCellFormula();
} else if (cell.getCellType() == CellType.BLANK) {
return "";
} else if (cell.getCellType() == CellType.BOOLEAN) {
boolean b = cell.getBooleanCellValue();
if (b) {
return "true";
} else {
return "false";
}
}
return "";
}
public static void main(String[] args) throws Exception {
UploadUtil<Map<String, Object>> uploadUtil = new UploadUtil<Map<String, Object>>() {
@Override
public Map<String, Object> buiderEntity(Row row) {
Map<String, Object> map = new HashMap<>();
map.put("param1", UploadUtil.getCellContent(row.getCell(0)));
map.put("param2", UploadUtil.getCellContent(row.getCell(1)));
return map;
}
};
File file = new File("/Users/qishi/Desktop/aaaa.xlsx");
InputStream inputStream = new FileInputStream(file);
List<Map<String, Object>> list = uploadUtil.getListByExcel(inputStream, "aaaa.xlsx");
}
}