当前位置:   article > 正文

Java实现Excel数据导入数据库_java读取excel文件并导入数据库

java读取excel文件并导入数据库

首先

在生成Excel前,我们需要理解一下Excel文件的组织形式。在POI中,是这样理解的:一个Excel文件对应一个workbook,一个workerbook是有若干个sheet组成的。一个sheet有多个row,一个row一般存在多个cell。

1、根据业务需求设计数据库表

2、根据数据库表设计一个Excel模板

模板的每列属性必须与表字段一一对应

3、环境准备

我这里项目环境是基于SpringBoot单体式架构,持久层用的公司框架,内置了基于MyBatis-Plus的各种单表操作的方法。

导入依赖

  1. <!--使用POI读取文件-->
  2. <dependency>
  3. <groupId>org.apache.poi</groupId>
  4. <artifactId>poi</artifactId>
  5. <version>3.17</version>
  6. </dependency>
  7. <dependency>
  8. <groupId>org.apache.poi</groupId>
  9. <artifactId>poi-ooxml</artifactId>
  10. <version>3.17</version>
  11. </dependency>

4、通过插件生成表对应的实体类

  1. /**
  2. * TbZbzs: 值班值守表
  3. * @author zs
  4. * @date 2021-12-17 08:46:31
  5. **/
  6. @Data
  7. @ApiModel(value="值班值守表,对应表tb_zbzs",description="适用于新增和修改页面传参")
  8. public class TbZbzs extends ProBaseEntity<TbZbzs> {
  9. private static final long serialVersionUID = 1L;
  10. @ApiModelProperty(value="id")
  11. private String id; // id
  12. @ApiModelProperty(value="部门")
  13. private String bm; // 部门
  14. @ApiModelProperty(value="值班上报")
  15. private String zbsb; // 值班上报
  16. @ApiModelProperty(value="值班人员")
  17. private String zbry; // 值班人员
  18. @ApiModelProperty(value="上报时间")
  19. @DateTimeFormat(pattern = "yyyy-MM-dd HH:mm:ss")
  20. private java.util.Date sbsj; // 上报时间
  21. @ApiModelProperty(value="结束时间")
  22. @DateTimeFormat(pattern = "yyyy-MM-dd HH:mm:ss")
  23. private java.util.Date jssj; // 结束时间
  24. @ApiModelProperty(value="联系方式")
  25. private String lxfs; // 联系方式
  26. @ApiModelProperty(value="状态")
  27. private String zt; // 状态
  28. /**
  29. * 逻辑删除
  30. */
  31. @ApiModelProperty(value="逻辑删除")
  32. private String delFlag;
  33. /**
  34. * 创建时间
  35. */
  36. @ApiModelProperty(value="创建时间")
  37. @DateTimeFormat(pattern = "yyyy-MM-dd HH:mm:ss")
  38. private Date createDate;
  39. }

5、自定义编写工具类

这里提供的是一个基础模板,根据业务的需求可以增加转换条件

  1. import org.apache.poi.hssf.usermodel.HSSFWorkbook;
  2. import org.apache.poi.ss.usermodel.Cell;
  3. import org.apache.poi.ss.usermodel.Row;
  4. import org.apache.poi.ss.usermodel.Sheet;
  5. import org.apache.poi.ss.usermodel.Workbook;
  6. import org.apache.poi.xssf.usermodel.XSSFWorkbook;
  7. import java.io.IOException;
  8. import java.io.InputStream;
  9. import java.text.DecimalFormat;
  10. import java.text.SimpleDateFormat;
  11. import java.util.ArrayList;
  12. import java.util.List;
  13. /**
  14. * 新增值班排班表导入Excel表工具类
  15. * zyw
  16. */
  17. public class ImportExcelUtil {
  18. private final static String excel2003L =".xls"; //2003- 版本的excel
  19. private final static String excel2007U =".xlsx"; //2007+ 版本的excel
  20. /**
  21. * 描述:获取IO流中的数据,组装成List<List<Object>>对象
  22. * @param in,fileName
  23. * @return
  24. * @throws Exception
  25. */
  26. public static List<List<Object>> getListByExcel(InputStream in, String fileName) throws Exception {
  27. List<List<Object>> list = null;
  28. //创建Excel工作薄
  29. Workbook work = ImportExcelUtil.getWorkbook(in,fileName);
  30. if(null == work){
  31. throw new Exception("创建Excel工作薄为空!");
  32. }
  33. Sheet sheet = null;
  34. Row row = null;
  35. Cell cell = null;
  36. list = new ArrayList<List<Object>>();
  37. //遍历Excel中所有的sheet
  38. for (int i = 0; i < work.getNumberOfSheets(); i++) {
  39. sheet = work.getSheetAt(i);
  40. if(sheet==null){continue;}
  41. //遍历当前sheet中的所有行
  42. for (int j = sheet.getFirstRowNum(); j < sheet.getLastRowNum()+1; j++) {
  43. row = sheet.getRow(j);
  44. if(row==null||row.getFirstCellNum()==j){continue;}
  45. //遍历所有的列
  46. List<Object> li = new ArrayList<Object>();
  47. for (int y = row.getFirstCellNum(); y < row.getLastCellNum(); y++) {
  48. cell = row.getCell(y);
  49. li.add(ImportExcelUtil.getCellValue(cell));
  50. }
  51. list.add(li);
  52. }
  53. }
  54. // work.close();
  55. return list;
  56. }
  57. /**
  58. * 描述:根据文件后缀,自适应上传文件的版本
  59. * @param inStr,fileName
  60. * @return
  61. * @throws Exception
  62. */
  63. public static Workbook getWorkbook(InputStream inStr, String fileName) throws Exception{
  64. Workbook wb = null;
  65. String fileType = fileName.substring(fileName.lastIndexOf("."));
  66. if(excel2003L.equals(fileType)){
  67. wb = new HSSFWorkbook(inStr); //2003-
  68. }else if(excel2007U.equals(fileType)){
  69. wb = new XSSFWorkbook(inStr); //2007+
  70. }else{
  71. throw new Exception("解析的文件格式有误!");
  72. }
  73. return wb;
  74. }
  75. /**
  76. * 描述:对表格中数值进行格式化
  77. * @param cell
  78. * @return
  79. */
  80. public static Object getCellValue(Cell cell){
  81. Object value = null;
  82. DecimalFormat df = new DecimalFormat("0"); //格式化number String字符
  83. SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
  84. // SimpleDateFormat sdf = new SimpleDateFormat("yyy-MM-dd"); //日期格式化
  85. // DecimalFormat df2 = new DecimalFormat("0.00"); //格式化数字
  86. if (cell!=null){
  87. switch (cell.getCellType()) {
  88. case Cell.CELL_TYPE_STRING:
  89. value = cell.getRichStringCellValue().getString();
  90. break;
  91. case Cell.CELL_TYPE_NUMERIC:
  92. if("General".equals(cell.getCellStyle().getDataFormatString())){
  93. value = df.format(cell.getNumericCellValue());
  94. }
  95. else if("m/d/yy".equals(cell.getCellStyle().getDataFormatString())){
  96. value = sdf.format(cell.getDateCellValue());
  97. }
  98. else{
  99. value = sdf.format(cell.getDateCellValue());
  100. }
  101. break;
  102. case Cell.CELL_TYPE_BOOLEAN:
  103. value = cell.getBooleanCellValue();
  104. break;
  105. case Cell.CELL_TYPE_BLANK:
  106. value = "";
  107. break;
  108. default:
  109. break;
  110. }
  111. }
  112. return value;
  113. }
  114. }

6、编写具体业务逻辑Service

主要思想:通过工具类将Excel文件解析成Object泛型的集合,再将集合循环遍历,在遍历中,将每行数据一次填入对象中,再每次循环中,将赋值后的对象存入一个list集合,最后统一将集合执行批量上传的方法,存入数据库。

  1. public Map<String,Object> importTprkxx(MultipartFile file){
  2. Map<String,Object> resultMap = new HashMap<>();
  3. SimpleDateFormat simpleDateFormat = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
  4. List<TbZbzs> tbZbzsList = new ArrayList<>();
  5. try {
  6. //验证文件类型
  7. if (!file.getOriginalFilename().substring(file.getOriginalFilename().lastIndexOf(".")).equals(".xls")&&!file.getOriginalFilename().substring(file.getOriginalFilename().lastIndexOf(".")).equals(".xlsx")){
  8. resultMap.put("mete", "文件类型有误!请上传Excle文件");
  9. throw new Exception("文件类型有误!请上传Excle文件");
  10. }
  11. //获取数据
  12. List<List<Object>> olist = ImportExcelUtil.getListByExcel(file.getInputStream(), file.getOriginalFilename());
  13. resultMap.put("导入成功",200);
  14. //封装数据
  15. for (int i = 0; i < olist.size(); i++) {
  16. List<Object> list = olist.get(i);
  17. if (list.get(0) == "" || ("序号").equals(list.get(0))) {
  18. continue;
  19. }
  20. TbZbzs tbZbzs = new TbZbzs();
  21. tbZbzs.setId(UUID.randomUUID().toString().replace("-", "").substring(0, 20));
  22. //根据下标获取每一行的每一条数据
  23. if (String.valueOf(list.get(0))==null||String.valueOf(list.get(0)).equals("")) {
  24. resultMap.put("mete", "部门不能为空");
  25. throw new Exception("部门不能为空");
  26. }
  27. tbZbzs.setBm(String.valueOf(list.get(0)));
  28. if (String.valueOf(list.get(1))==null||String.valueOf(list.get(1)).equals("")) {
  29. resultMap.put("mete", "值班上报不能为空");
  30. throw new Exception("值班上报不能为空");
  31. }
  32. tbZbzs.setZbsb(String.valueOf(list.get(1)));
  33. if (String.valueOf(list.get(2))==null||String.valueOf(list.get(2)).equals("")) {
  34. resultMap.put("mete", "值班人员不能为空");
  35. throw new Exception("值班人员不能为空");
  36. }
  37. tbZbzs.setZbry(String.valueOf(list.get(2)));
  38. if (String.valueOf(list.get(3))==null||String.valueOf(list.get(3)).equals("")) {
  39. resultMap.put("mete", "导入失败,上报时间不能为空");
  40. throw new Exception("导入失败,上报时间不能为空");
  41. }
  42. try {
  43. tbZbzs.setSbsj(simpleDateFormat.parse(String.valueOf(list.get(3))));
  44. }catch (ParseException e){
  45. resultMap.put("mete","请填写正确的时间格式!");
  46. throw new Exception("请填写正确的时间格式!");
  47. }
  48. if (String.valueOf(list.get(4))==null||String.valueOf(list.get(4)).equals("")) {
  49. resultMap.put("mete", "导入失败,结束时间不能为空");
  50. throw new Exception("导入失败,结束时间不能为空");
  51. }
  52. try {
  53. tbZbzs.setJssj(simpleDateFormat.parse(String.valueOf(list.get(4))));
  54. }catch (ParseException e){
  55. resultMap.put("mete","请填写正确的时间格式!");
  56. throw new Exception("请填写正确的时间格式!");
  57. }
  58. if (String.valueOf(list.get(5))==null||String.valueOf(list.get(5)).equals("")) {
  59. resultMap.put("mete", "联系方式不能为空");
  60. throw new Exception("导入失败,联系方式不能为空");
  61. }
  62. tbZbzs.setLxfs(String.valueOf(list.get(5)));
  63. if (String.valueOf(list.get(6))==null||String.valueOf(list.get(6)).equals("")) {
  64. resultMap.put("mete", "状态不能为空");
  65. throw new Exception("导入失败,状态不能为空");
  66. }
  67. tbZbzs.setZt(String.valueOf(list.get(6)));
  68. if (String.valueOf(list.get(7))==null||String.valueOf(list.get(7)).equals("")) {
  69. resultMap.put("mete", "逻辑删除不能为空");
  70. continue;
  71. }
  72. tbZbzs.setDelFlag(String.valueOf(list.get(7)));
  73. if (String.valueOf(list.get(8))==null||String.valueOf(list.get(8)).equals("")) {
  74. resultMap.put("mete", "导入失败,创建时间不能为空");
  75. throw new Exception("导入失败,创建时间不能为空");
  76. }
  77. try {
  78. tbZbzs.setCreateDate(simpleDateFormat.parse(String.valueOf(list.get(8))));
  79. }catch (ParseException e){
  80. resultMap.put("mete","请填写正确的时间格式!");
  81. throw new Exception("请填写正确的时间格式!");
  82. }
  83. tbZbzsList.add(tbZbzs);
  84. }
  85. int i = tbZbzsDao.insertTbZbzsList(tbZbzsList);
  86. if (i != 0) {
  87. resultMap.put("state", 200);
  88. }else {
  89. resultMap.put("mete","文档内无数据,请重新导入");
  90. resultMap.put("state", 500);
  91. }
  92. } catch (Exception e) {
  93. e.printStackTrace();
  94. }finally {
  95. return resultMap;
  96. }
  97. }

7、在dao层对应的xml文件中,编写批量上传的方法

  1. <insert id="insertTbZbzsList" parameterType="java.util.List">
  2. insert into tb_zbzs (
  3. id,
  4. bm,
  5. zbsb,
  6. zbry,
  7. sbsj,
  8. jssj,
  9. lxfs,
  10. zt,
  11. del_flag,
  12. create_date
  13. ) VALUES
  14. <foreach collection="list" item="item" separator=",">
  15. (
  16. #{item.id},
  17. #{item.bm},
  18. #{item.zbsb},
  19. #{item.zbry},
  20. #{item.sbsj},
  21. #{item.jssj},
  22. #{item.lxfs},
  23. #{item.zt},
  24. #{item.delFlag},
  25. #{item.createDate}
  26. )
  27. </foreach>
  28. </insert>

8、Controller实现业务的控制

  1. /**
  2. * @方法名称: excelProTbZbzs
  3. * @实现功能: 导入值班值守表Excel TODO: 方法入参根据页面对象设置
  4. * @param file
  5. * @return java.lang.String
  6. * @create by zyw at 2022-03-17 16:49:31
  7. **/
  8. @ApiOperation(value="导入值班值守表Excel",notes="返回导入情况接口",response = TbZbzs.class)
  9. @PostMapping(value = "/excelProTbZbzs")
  10. public String excelProTbZbzs(@RequestParam("file") MultipartFile file){
  11. try {
  12. return buildResultStr(service.importTprkxx(file).get("state").equals(200) ? buildSuccessResultData() : buildErrorResultData(service.importTprkxx(file).get("mete").toString()));
  13. }catch (RuntimeException e){
  14. logError(log, e);
  15. return buildResultStr(buildErrorResultData(service.importTprkxx(file).get("mete").toString()));
  16. }
  17. }

9、通过Swagger测试接口

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/繁依Fanyi0/article/detail/726015
推荐阅读
相关标签
  

闽ICP备14008679号