POI导入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插入，直接将数据集合传过去就

巴特西