| Java实现Excel数据导入数据库 | 您所在的位置：网站首页 › 写入数据库 › Java实现Excel数据导入数据库 |
Java实现Excel数据导入数据库
|
首先
在生成Excel前，我们需要理解一下Excel文件的组织形式。在POI中，是这样理解的：一个Excel文件对应一个workbook，一个workerbook是有若干个sheet组成的。一个sheet有多个row，一个row一般存在多个cell。
模板的每列属性必须与表字段一一对应
我这里项目环境是基于SpringBoot单体式架构，持久层用的公司框架，内置了基于MyBatis-Plus的各种单表操作的方法。 导入依赖 org.apache.poi poi 3.17 org.apache.poi poi-ooxml 3.17 4、通过插件生成表对应的实体类 /** * TbZbzs: 值班值守表 * @author zs * @date 2021-12-17 08:46:31 **/ @Data @ApiModel(value="值班值守表,对应表tb_zbzs",description="适用于新增和修改页面传参") public class TbZbzs extends ProBaseEntity { private static final long serialVersionUID = 1L; @ApiModelProperty(value="id") private String id; // id @ApiModelProperty(value="部门") private String bm; // 部门 @ApiModelProperty(value="值班上报") private String zbsb; // 值班上报 @ApiModelProperty(value="值班人员") private String zbry; // 值班人员 @ApiModelProperty(value="上报时间") @DateTimeFormat(pattern = "yyyy-MM-dd HH:mm:ss") private java.util.Date sbsj; // 上报时间 @ApiModelProperty(value="结束时间") @DateTimeFormat(pattern = "yyyy-MM-dd HH:mm:ss") private java.util.Date jssj; // 结束时间 @ApiModelProperty(value="联系方式") private String lxfs; // 联系方式 @ApiModelProperty(value="状态") private String zt; // 状态 /** * 逻辑删除 */ @ApiModelProperty(value="逻辑删除") private String delFlag; /** * 创建时间 */ @ApiModelProperty(value="创建时间") @DateTimeFormat(pattern = "yyyy-MM-dd HH:mm:ss") private Date createDate; } 5、自定义编写工具类这里提供的是一个基础模板，根据业务的需求可以增加转换条件 import org.apache.poi.hssf.usermodel.HSSFWorkbook; import org.apache.poi.ss.usermodel.Cell; import org.apache.poi.ss.usermodel.Row; import org.apache.poi.ss.usermodel.Sheet; import org.apache.poi.ss.usermodel.Workbook; import org.apache.poi.xssf.usermodel.XSSFWorkbook; import java.io.IOException; import java.io.InputStream; import java.text.DecimalFormat; import java.text.SimpleDateFormat; import java.util.ArrayList; import java.util.List; /** * 新增值班排班表导入Excel表工具类 * zyw */ public class ImportExcelUtil { private final static String excel2003L =".xls"; //2003- 版本的excel private final static String excel2007U =".xlsx"; //2007+ 版本的excel /** * 描述：获取IO流中的数据，组装成List对象 * @param in,fileName * @return * @throws Exception */ public static List getListByExcel(InputStream in, String fileName) throws Exception { List list = null; //创建Excel工作薄 Workbook work = ImportExcelUtil.getWorkbook(in,fileName); if(null == work){ throw new Exception("创建Excel工作薄为空！"); } Sheet sheet = null; Row row = null; Cell cell = null; list = new ArrayList(); //遍历Excel中所有的sheet for (int i = 0; i < work.getNumberOfSheets(); i++) { sheet = work.getSheetAt(i); if(sheet==null){continue;} //遍历当前sheet中的所有行 for (int j = sheet.getFirstRowNum(); j < sheet.getLastRowNum()+1; j++) { row = sheet.getRow(j); if(row==null||row.getFirstCellNum()==j){continue;} //遍历所有的列 List li = new ArrayList(); for (int y = row.getFirstCellNum(); y < row.getLastCellNum(); y++) { cell = row.getCell(y); li.add(ImportExcelUtil.getCellValue(cell)); } list.add(li); } } // work.close(); return list; } /** * 描述：根据文件后缀，自适应上传文件的版本 * @param inStr,fileName * @return * @throws Exception */ public static Workbook getWorkbook(InputStream inStr, String fileName) throws Exception{ Workbook wb = null; String fileType = fileName.substring(fileName.lastIndexOf(".")); if(excel2003L.equals(fileType)){ wb = new HSSFWorkbook(inStr); //2003- }else if(excel2007U.equals(fileType)){ wb = new XSSFWorkbook(inStr); //2007+ }else{ throw new Exception("解析的文件格式有误！"); } return wb; } /** * 描述：对表格中数值进行格式化 * @param cell * @return */ public static Object getCellValue(Cell cell){ Object value = null; DecimalFormat df = new DecimalFormat("0"); //格式化number String字符 SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss"); // SimpleDateFormat sdf = new SimpleDateFormat("yyy-MM-dd"); //日期格式化 // DecimalFormat df2 = new DecimalFormat("0.00"); //格式化数字 if (cell!=null){ switch (cell.getCellType()) { case Cell.CELL_TYPE_STRING: value = cell.getRichStringCellValue().getString(); break; case Cell.CELL_TYPE_NUMERIC: if("General".equals(cell.getCellStyle().getDataFormatString())){ value = df.format(cell.getNumericCellValue()); } else if("m/d/yy".equals(cell.getCellStyle().getDataFormatString())){ value = sdf.format(cell.getDateCellValue()); } else{ value = sdf.format(cell.getDateCellValue()); } break; case Cell.CELL_TYPE_BOOLEAN: value = cell.getBooleanCellValue(); break; case Cell.CELL_TYPE_BLANK: value = ""; break; default: break; } } return value; } } 6、编写具体业务逻辑Service主要思想：通过工具类将Excel文件解析成Object泛型的集合，再将集合循环遍历，在遍历中，将每行数据一次填入对象中，再每次循环中，将赋值后的对象存入一个list集合，最后统一将集合执行批量上传的方法，存入数据库。 public Map importTprkxx(MultipartFile file){ Map resultMap = new HashMap(); SimpleDateFormat simpleDateFormat = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss"); List tbZbzsList = new ArrayList(); try { //验证文件类型 if (!file.getOriginalFilename().substring(file.getOriginalFilename().lastIndexOf(".")).equals(".xls")&&!file.getOriginalFilename().substring(file.getOriginalFilename().lastIndexOf(".")).equals(".xlsx")){ resultMap.put("mete", "文件类型有误！请上传Excle文件"); throw new Exception("文件类型有误！请上传Excle文件"); } //获取数据 List olist = ImportExcelUtil.getListByExcel(file.getInputStream(), file.getOriginalFilename()); resultMap.put("导入成功",200); //封装数据 for (int i = 0; i < olist.size(); i++) { List list = olist.get(i); if (list.get(0) == "" || ("序号").equals(list.get(0))) { continue; } TbZbzs tbZbzs = new TbZbzs(); tbZbzs.setId(UUID.randomUUID().toString().replace("-", "").substring(0, 20)); //根据下标获取每一行的每一条数据 if (String.valueOf(list.get(0))==null||String.valueOf(list.get(0)).equals("")) { resultMap.put("mete", "部门不能为空"); throw new Exception("部门不能为空"); } tbZbzs.setBm(String.valueOf(list.get(0))); if (String.valueOf(list.get(1))==null||String.valueOf(list.get(1)).equals("")) { resultMap.put("mete", "值班上报不能为空"); throw new Exception("值班上报不能为空"); } tbZbzs.setZbsb(String.valueOf(list.get(1))); if (String.valueOf(list.get(2))==null||String.valueOf(list.get(2)).equals("")) { resultMap.put("mete", "值班人员不能为空"); throw new Exception("值班人员不能为空"); } tbZbzs.setZbry(String.valueOf(list.get(2))); if (String.valueOf(list.get(3))==null||String.valueOf(list.get(3)).equals("")) { resultMap.put("mete", "导入失败,上报时间不能为空"); throw new Exception("导入失败,上报时间不能为空"); } try { tbZbzs.setSbsj(simpleDateFormat.parse(String.valueOf(list.get(3)))); }catch (ParseException e){ resultMap.put("mete","请填写正确的时间格式！"); throw new Exception("请填写正确的时间格式！"); } if (String.valueOf(list.get(4))==null||String.valueOf(list.get(4)).equals("")) { resultMap.put("mete", "导入失败,结束时间不能为空"); throw new Exception("导入失败,结束时间不能为空"); } try { tbZbzs.setJssj(simpleDateFormat.parse(String.valueOf(list.get(4)))); }catch (ParseException e){ resultMap.put("mete","请填写正确的时间格式！"); throw new Exception("请填写正确的时间格式！"); } if (String.valueOf(list.get(5))==null||String.valueOf(list.get(5)).equals("")) { resultMap.put("mete", "联系方式不能为空"); throw new Exception("导入失败,联系方式不能为空"); } tbZbzs.setLxfs(String.valueOf(list.get(5))); if (String.valueOf(list.get(6))==null||String.valueOf(list.get(6)).equals("")) { resultMap.put("mete", "状态不能为空"); throw new Exception("导入失败,状态不能为空"); } tbZbzs.setZt(String.valueOf(list.get(6))); if (String.valueOf(list.get(7))==null||String.valueOf(list.get(7)).equals("")) { resultMap.put("mete", "逻辑删除不能为空"); continue; } tbZbzs.setDelFlag(String.valueOf(list.get(7))); if (String.valueOf(list.get(8))==null||String.valueOf(list.get(8)).equals("")) { resultMap.put("mete", "导入失败,创建时间不能为空"); throw new Exception("导入失败,创建时间不能为空"); } try { tbZbzs.setCreateDate(simpleDateFormat.parse(String.valueOf(list.get(8)))); }catch (ParseException e){ resultMap.put("mete","请填写正确的时间格式！"); throw new Exception("请填写正确的时间格式！"); } tbZbzsList.add(tbZbzs); } int i = tbZbzsDao.insertTbZbzsList(tbZbzsList); if (i != 0) { resultMap.put("state", 200); }else { resultMap.put("mete","文档内无数据，请重新导入"); resultMap.put("state", 500); } } catch (Exception e) { e.printStackTrace(); }finally { return resultMap; } } 7、在dao层对应的xml文件中，编写批量上传的方法 insert into tb_zbzs ( id, bm, zbsb, zbry, sbsj, jssj, lxfs, zt, del_flag, create_date ) VALUES ( #{item.id}, #{item.bm}, #{item.zbsb}, #{item.zbry}, #{item.sbsj}, #{item.jssj}, #{item.lxfs}, #{item.zt}, #{item.delFlag}, #{item.createDate} ) 8、Controller实现业务的控制 /** * @方法名称: excelProTbZbzs * @实现功能: 导入值班值守表Excel TODO: 方法入参根据页面对象设置 * @param file * @return java.lang.String * @create by zyw at 2022-03-17 16:49:31 **/ @ApiOperation(value="导入值班值守表Excel",notes="返回导入情况接口",response = TbZbzs.class) @PostMapping(value = "/excelProTbZbzs") public String excelProTbZbzs(@RequestParam("file") MultipartFile file){ try { return buildResultStr(service.importTprkxx(file).get("state").equals(200) ? buildSuccessResultData() : buildErrorResultData(service.importTprkxx(file).get("mete").toString())); }catch (RuntimeException e){ logError(log, e); return buildResultStr(buildErrorResultData(service.importTprkxx(file).get("mete").toString())); } } 9、通过Swagger测试接口
|
【本文地址】
公司简介
联系我们