本文是基于Apache poi类实现的批量导入读取Excel文件，所以要先引入Apache poi的依赖
|
1
2
3
4
5
6
7
8
9
10
|
<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里面进行实现，具体代码为
java" id="highlighter_278316">
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
|
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 ，方法直接写到下面，大家不会陌生，返回方法是我封装的类，大家用自己项目的稍加修改就可以
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
|
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了，直接上代码，工具类在下面
|
1
2
3
4
5
6
|
@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); } |
这是获取根路径的工具类
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
|
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); }} |
好了，一个简单的Excel批量导入数据就完成了，感谢支持，大家有什么问题都可以评论留言哦
以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持服务器之家。
原文链接：https://blog.csdn.net/weixin_43250266/article/details/107684036
