POI导入excel
在做后台管理的时候经常会用到excel导入的问题,就是将excel中的内容批量导入到数据库中,正好在新项目中我也做了excel导入的功能,来分享给大家,也给自己做个记录。
前言
在做后台管理的时候经常会用到excel导入的问题,就是将excel中的内容批量导入到数据库中,正好在新项目中我也做了excel导入的功能,来分享给大家,也给自己做个记录。
核心思想
excel导入的核心思想很简单,就是读取单元格里的内容然后塞到JavaBean
中,把每一个JavaBean
再添加到list
中再循环insert
就可以了。
具体实现
前台怎么上传文件我就不在这里赘述了,我只讲后台的实现。这里我们使用poi
来进行导入操作
文件导入
导入判断
上传到后台肯定得需要一系列的判断,例如文件或者其内容是否为空,上传文件格式是否正确等等
/**
* 导入判断
*
* @param mfile
* @return map
*/
public Map<String, Object> lockExcelImport(MultipartFile mfile) {
Map<String, Object> result = new HashMap<String, Object>();
//判断文件是否为空
if (mfile == null) {
result.put("result", "error");
result.put("msg", "文件不能为空!");
return result;
}
//获取文件名
String fileName = mfile.getOriginalFilename();
//验证文件名是否合格
if (!ExcelImportUtils.validateExcel(fileName)) {
result.put("result", "error");
result.put("msg", "文件必须是excel格式!");
return result;
}
//进一步判断文件内容是否为空(即判断其大小是否为0或其名称是否为null)
long size = mfile.getSize();
if (StringUtils.isEmpty(fileName) || size == 0) {
result.put("result", "error");
result.put("msg", "文件不能为空!");
return result;
}
try {
File uploadDir = new File("E:\\fileupload");
//创建一个目录 (它的路径名由当前 File 对象指定,包括任一必须的父路径。)
if (!uploadDir.exists()) uploadDir.mkdirs();
//新建一个文件
File tempFile = new File("E:\\fileupload\\" + new Date().getTime() + ".xlsx");
//初始化输入流
InputStream is = null;
//将上传的文件写入新建的文件中
mfile.transferTo(tempFile);
//根据新建的文件实例化输入流
is = new FileInputStream(tempFile);
//根据版本选择创建Workbook的方式
Workbook wb = null;
//根据文件名判断文件是2003版本还是2007版本
if (ExcelImportUtils.isExcel2007(fileName)) {
wb = new XSSFWorkbook(is);
} else {
wb = new HSSFWorkbook(is);
}
//根据excel里面的内容读取知识库信息
result.put("result", "success");
result.put("msg", readExcelValue(wb, tempFile));
} catch (IOException e) {
e.printStackTrace();
}
return result;
}
读取excel内容
进行完文件判断之后我们来进行读取excel中的内容,并将数据塞到JavaBean
中,里面的每行代码我都注释的很详细,大概就是先获取到当前sheet页,之后获取行数来循环行,因为一般第一行是标题所以我们从第二行开始循环,之后获取每行的单元格,因为这里每个单元格内容都是特定的,所以我们直接获取指定单元格的值塞到对象中,再将每一个对象添加到list中去,这就是我们用来导入的数据。这里我返回的是封装好的map集合,大家直接返回map就可以
/**
* 解析Excel里面的数据
*
* @param wb
* @param tempFile
* @param userId
* @return
*/
private Map<String, Object> readExcelValue(Workbook wb, File tempFile, Integer userId) {
//返回结果
int reslutNum = 0;
//对象
List<LockHistoryEntity> lockImportList = new ArrayList<>();
//获取sheet页数
int pages = wb.getNumberOfSheets();
// 循环Excel的sheet页数,获取每个sheet表
for (int i = 0; i < pages; i++) {
// 取得工作表
Sheet sheet = wb.getSheetAt(i);
// 获取行数
int rows = sheet.getPhysicalNumberOfRows();
//循环Excel的行数,从第二行开始
for (int r = 1; r < rows; r++) {
LockHistoryEntity lockImport = new LockHistoryEntity();
//获得每一行
Row row = sheet.getRow(r);
//判断每个单元格是否为空
if (row.getCell(0) != null || row.getCell(1) != null || row.getCell(2) != null || row.getCell(3) != null) {
// 客户卡号
lockImport.setCustCode(row.getCell(0).getStringCellValue().trim());
// 客户姓名
lockImport.setCustName(row.getCell(1).getStringCellValue().trim());
}
//进行一下判断,防止将空对象添加到了list中,当然每个业务的判断条件肯定不同,因情景而仪
if (!StringUtils.isEmpty(lockImport.getCustCode())) {
// 加入数据队列
lockImportList.add(lockImport);
}
// 没有明细数据时,直接结束程序
if (lockImportList.size() == 0) {
return ServiceUtil.generateResponseMap(null, Constants.STATE_FAILED);
}
}
reslutNum = impLockHistory(lockImportList);
}
//删除上传的临时文件
if (tempFile.exists()) {
tempFile.delete();
}
return ServiceUtil.generateResponseMap(null, reslutNum > 0 ? Constants.STATE_SUCCESS : Constants.STATE_FAILED);
}
数据DB插入更新
已经到了最后一步数据插入,这里我用的是mybatis
的foreach插入,直接将数据集合传过去就