【Java】【46】导入Excel到数据库
前言：
业务场景：用户提供Excel表，在页面上点击“导入按钮”，系统读取Excel中的数据，存到对应的数据库
注：
1，目前仅提供导入简单的Excel表，没有合并单元格，只能读取单页sheet
2，方法里用到了具体业务场景的实体类，所以适用性不是很强，其他场景要用的话，还需要修改代码。但是也具有一定的参考性，所以先记录下来。我的后续思路是：在 工具类（ReadExcelUtil） 里根据导入的Excel表头确定要处理的字段名，然后拼成json格式的数据，传到Service层，再做处理。之后再做更新 //TODO
3，用的jar包：POI
正文：
html：
<div> <span><input type="file" id="upfile" name="upfile" placeholder=""/></span> <button onclick="importExp();">导入</button> <span>格式：.xls</span> </div>
JS：
function importExp() { var name = $("#upfile").val(); var file = $("#upfile")[0].files[0]; // ajax... }
Java：
Controller层
@ApiOperation(value = "导入Excel") @RequestMapping(value="excel/import", method = RequestMethod.POST) public void importExcel(MultipartFile file) { return this.theService.importExcel(file); }
Service层
public void importExcel(MultipartFile file) { if(file.isEmpty()){ //请先选择Excel文件。是否把报错等信息返回到前端，视情况而定。 return; } Result readResult = ReadExcelUtil.readExcel(file); //Result是封装了返回值的类，相当于实体类 if(readResult.getCode() != 0){ //错误:readResult.msg return; } List<Award> list = (List<Award>) readResult.getData(); //Award是数据库表对应的实体类 //获取了list之后，insert到数据库... }
工具类：
ReadExcelUtil
package com.bf.base.utils; import com.bf.base.entity.DripAward; import com.bf.base.params.Result; import org.apache.poi.hssf.usermodel.HSSFWorkbook; import org.apache.poi.ss.usermodel.*; import org.apache.poi.xssf.usermodel.XSSFWorkbook; import org.springframework.web.multipart.MultipartFile; import java.io.FileNotFoundException; import java.io.IOException; import java.io.InputStream; import java.text.DecimalFormat; import java.util.ArrayList; import java.util.Calendar; import java.util.List; public class ReadExcelUtil { /** * 读取 单页sheet，返回一个集合 * @return */ public static Result<?> readDripAwardExcel(MultipartFile file) { Result result = new Result<>(); InputStream is = null; Workbook wb = null; String fileName = file.getOriginalFilename(); String fileType = fileName.substring(fileName.lastIndexOf(".") + 1, fileName.length()); try { is = file.getInputStream(); if (fileType.equals("xls")) { wb = new HSSFWorkbook(is); } else if (fileType.equals("xlsx")) { wb = new XSSFWorkbook(is); } else { return new Result<>(Result.FAIL, "读取的不是excel文件", null, null); } int sheetSize = wb.getNumberOfSheets();//有多少sheet页 if(sheetSize >= 2){ return new Result<>(Result.FAIL, "请核对Excel的页数", null, null); } Sheet sheet = wb.getSheetAt(0); result = sheetData2List(sheet); //关键，sheet表数据 转 集合 } catch (FileNotFoundException e) { e.printStackTrace(); } catch (IOException e) { e.printStackTrace(); } catch (Exception e) { e.printStackTrace(); } return result; } //获取数据 private static Result sheetData2List(Sheet sheet) { Result result = new Result<>(); List<Award> awards = new ArrayList<>(); int CELL_SIZE = 3; //excel固定三列(编码、金额) int rowSize = sheet.getLastRowNum() + 1; for (int j = 0; j < rowSize; j++) { //读取每一行 Row row = sheet.getRow(j); if (row == null) { continue; } if(row.getLastCellNum() != CELL_SIZE){ result.setCode(2); result.setMsg("第"+ j +"行数据异常，请核对后再上传！"); return result; } if (j == 0) { continue; } else { DripAward rowObj = new DripAward(); for (int k = 0; k < CELL_SIZE; k++) { //获取列的数据 Cell cell = row.getCell(k); String value = getCellFormatValue(cell); switch (k) { case 0: rowObj.setCode(AESUtil.encrypt(value)); break; case 1: rowObj.setAmount(Double.parseDouble(value)); break; case 2: rowObj.setClassify(Integer.parseInt(value)); break; default: break; } } awards.add(rowObj); } } result.setCode(0); result.setData(awards); return result; } //获取列的数据 private static String getCellFormatValue(Cell cell) { String cellvalue = ""; if (cell != null) { CellType cellType = cell.getCellTypeEnum(); switch (cellType) { case NUMERIC: { if(String.valueOf(cell.getNumericCellValue()).indexOf("E") == -1){ cellvalue = String.valueOf(cell.getNumericCellValue()); }else { cellvalue = new DecimalFormat("#").format(cell.getNumericCellValue()); } break; } case STRING: cellvalue = cell.getRichStringCellValue().getString(); break; default: cellvalue = "-"; } } return cellvalue; } }
其他：
Result （封装了返回值的类，相当于实体类）
package com.bf.base.params; import org.apache.commons.lang.StringUtils; /** * 传给前端数据的统一格式 * code = 0时，表示调用成功 * 其余code均表示调用接口异常，异常时，标明异常码，并给出msg和detail注释，同步文档 */ public class Result<T> { private int code;//状态返回码 private String msg;//返回码描述 private String detail;//错误详细描述或返回码对应处理方案 private T data;//返回的主体数据 public static final int FAIL = -1; public static final int SUCCESS = 0; @Override public String toString() { return "Result [code=" + code + ", msg=" + msg + ", detail=" + detail + ", data=" + data + "]"; } public int getCode() { return code; } public void setCode(int code) { this.code = code; } public String getMsg() { return msg; } public void setMsg(String msg) { this.msg = msg; } public String getDetail() { return detail; } public void setDetail(String detail) { this.detail = detail; } public T getData() { return data; } public void setData(T data) { this.data = data; } }
Award（数据库表对应的实体类）
public class Award { private String code; private Double amount; private int classify; public String getCode() { return code; } public void setCode(String code) { this.code = code; } public Double getAmount() { return amount; } public void setAmount(Double amount) { this.amount = amount; } public int getClassify() { return classify; } public void setClassify(int classify) { this.classify = classify; } }
