java实现批量导入Excel表格数据到数据库

Z时代
2024-01-10
分类：综合

本文是基于Apache poi类实现的批量导入读取Excel文件，所以要先引入Apache poi的依赖


<dependency>
   <groupId>org.apache.poi</groupId>
   <artifactId>poi</artifactId>
   <version>4.1.1</version>
  </dependency>
  <dependency>
   <groupId>org.apache.poi</groupId>
   <artifactId>poi-ooxml</artifactId>
   <version>4.1.1</version>
</dependency>

在引入依赖之后，我们就可以开始进行操作了，首先，导入Excel数据，我们要先能够读取Excel每一行每一列的内容，只有读取到内容了，才可以将内容存入数组，最后实现插入数据库。所以我们要先读取Excel表格的数据，我的项目是springboot，我在service定义了一个读取方法，然后在impl里面进行实现，具体代码为


public class ImportOrderDTO {
 private String filePath;
 public String getFilePath() {
  return filePath;
 }
 public void setFilePath(String filePath) {
  this.filePath = filePath;
 }
} 
/**
  * 读取导入数据 excel 内容
  *
  * @param importOrderDTO 导入参数
  * @param rootPath  根路径
  * @return result
  */
 public static final String SEPA = File.separator;//这是下面用到的常量，自己放好位置
 private List<User> readExcel(ImportOrderDTO importOrderDTO, String rootPath) {
  List<User> excelContent = new ArrayList<>();
  try {
   InputStream inputStream = new FileInputStream(rootPath + SEPA + importOrderDTO.getFilePath());
   XSSFWorkbook wb = new XSSFWorkbook(inputStream);
   //遍历所有表，只支持xlsx，xls的是H的类
   XSSFSheet xssfSheet = wb.getSheetAt(0);
   int lastRowNum = xssfSheet.getLastRowNum();
   for (int i = 0; i <= lastRowNum; i++) {
//   通过下标获取行
    XSSFRow row = xssfSheet.getRow(i);
//   从行中获取数据
    if (row.getRowNum() == 0) {
     continue;
    }
    //第一列为空就跳出
    if (row.getCell(0) == null) {
     continue;
    }
    /**
     * getNumericCellValue() 获取数字
     * getStringCellValue 获取String，设置表格类型为String，可以避免很多问题
     */
    row.getCell(0).setCellType(CellType.STRING);
    row.getCell(1).setCellType(CellType.STRING);
    row.getCell(3).setCellType(CellType.STRING);
    row.getCell(5).setCellType(CellType.STRING);
    //UserInformPO是我自己定义的数据类，你们导入需要哪些数据就封装哪些，这就不用多讲了吧
    UserInformPO userInformPO = new UserInformPO();
    userInformPO.setAccount(row.getCell(0).getStringCellValue());
    userInformPO.setIdCard(row.getCell(1).getStringCellValue());
    userInformPO.setAvatar(row.getCell(2).getStringCellValue());
    userInformPO.setNickname(row.getCell(3).getStringCellValue());
    userInformPO.setSex(row.getCell(4).getStringCellValue());
    String salt = EncryptUtils.createSalt();
    userInformPO.setSalt(salt);
    //职业类别
    userInformPO.setIntegral(0);
    userInformPO.setVipLevel(0);
    userInformPO.setIsEnabled(0);
    userInformPO.setDelFlag(0);
    //然后将po转到我的实体类entity里面并将实体类加入到数组，方便正式执行批量导入的时候可以用，一些没有的类是我自己定义的加密的(规范)
    User user = new User();
    if (userInformPO.getSex().equals(SexEnum.MAN.getText())) {
     user.setSex(SexEnum.MAN.getValue());
    } else {
     user.setSex(SexEnum.WOMAN.getValue());
    }
    user.setAccount(userInformPO.getAccount());
    user.setIdCard(userInformPO.getIdCard());
    user.setAvatar(userInformPO.getAvatar());
    user.setNickname(userInformPO.getNickname());
    user.setPassword(userInformPO.getPassword());
    user.setSalt(userInformPO.getSalt());
    user.setIntegral(userInformPO.getIntegral());
    user.setVipLevel(userInformPO.getVipLevel());
    user.setDepartmentId(31);
    user.setIsEnabled(userInformPO.getIsEnabled());
    user.setDelFlag(userInformPO.getDelFlag());
    user.setUpdateTime(new Timestamp(System.currentTimeMillis()));
    user.setCreateTime(new Timestamp(System.currentTimeMillis()));
    //加入到数组中并且该方法返回该数组
    excelContent.add(user);
   }
  } catch (FileNotFoundException e) {
   throw new ServerException("文件不存在");
  } catch (IOException e) {
   System.out.println(e);
   throw new ServerException("读取文件失败");
  }
  return excelContent;
}

在上面写完读取Excel表单数据后，就可以开始写插入数据库的方法了，我用的是mybatis plus ，方法直接写到下面，大家不会陌生，返回方法是我封装的类，大家用自己项目的稍加修改就可以


public Result importUserWithExcel(ImportOrderDTO importOrderDTO, String rootPath) {
  try {
   //调用上面的方法，读取前端传过来的参数和文件路径
   List<User> excelContent = readExcel(importOrderDTO, rootPath);
   if (excelContent.isEmpty()) {
    return ResultUtil.error("数据为空");
   }
   //下面都是一些逻辑处理，大家一定看得懂就不多说了
   List<User> userList = userMapper.selectList(new EntityWrapper<User>());
   List<User> sameList = new ArrayList<>();
   List<User> differentList = new ArrayList<>();
   for (User excelStudent : excelContent) {
    //数据不同
    boolean flag = true;
    for (User user : userList) {
     if (user.getAccount().equals(excelStudent.getAccount())) {
      //相同的数据
      flag = false;
      sameList.add(excelStudent);
     }
    }
    if (flag) {
     //如果导入的数据与上面判断的条件相等了，就执行插入操作
     differentList.add(excelStudent);
     userMapper.insert(excelStudent);
    }
   }
   if (differentList.size() == 0) {
    //判断如果导入的数据跟当前的数据一致的话
    return ResultUtil.error("导入数据与当前数据一致！");
   }
   return ResultUtil.successWithMessage("数据导入成功");
  } catch (Exception e) {
   System.out.println(e);
   return ResultUtil.error("数据导入失败，请检查导入文件格式与模板文件是否相同！");
  }
 }

service层方法写完了，接着就是controller了，直接上代码，工具类在下面


@PostMapping(value = "/importExcel")
 public Result importExcel(HttpServletRequest request, @RequestBody ImportOrderDTO importOrderDTO) {
  String rootPath = ExcelUtils.getRootPath(request);
  System.out.println(importOrderDTO.getFilePath());
  return userService.importUserWithExcel(importOrderDTO, rootPath);
 }

这是获取根路径的工具类


public class ExcelUtils {
 /**
  * 获取上传根路径
  *
  * @param request 请求信息
  * @return string
  */
 public static String getRootPath(HttpServletRequest request) {
  return request.getSession().getServletContext().getRealPath(UtilConstant.UPLOAD_PATH);
 }
 public static Workbook getWorkbook(String excelType) {
  try {
   return WorkbookFactory.create(!ExcelTypeEnum.XLS.getVal().equals(excelType));
  } catch (IOException e) {
   throw new ServerException("创建excel文件失败", e);
  }
 }
 /**
  * 创建表
  *
  * @param wb   目标文件
  * @param sheetName 表名
  * @param sheetTitle 表头
  * @return sheet
  */
 public static Sheet createSheet(Workbook wb, String sheetName, String[] sheetTitle) {
  Sheet sheet = wb.createSheet(sheetName);
  Row row = sheet.createRow(0);
  for (int i = 0; i < sheetTitle.length; i++) {
   row.createCell(i).setCellValue(sheetTitle[i]);
  }
  return sheet;
 }
 /**
  * 创建单元格.
  *
  * @param row 行
  * @param column 列
  * @param value 值
  */
 public static void createCell(Row row, int column, String value) {
  Cell cell = row.createCell(column);
  cell.setCellValue(value);
 }
}