Java Excel資料匯入資料庫的方法

2022-03-18 13:01:08

1、根據業務需求設計資料庫表
2、根據資料庫表設計一個Excel模板
3、環境準備
4、通過外掛生成表對應的實體類
5、自定義編寫工具類
6、編寫具體業務邏輯Service
7、在dao層對應的xml檔案中，編寫批次上傳的方法
8、Controller實現業務的控制
9、通過Swagger測試介面
10、在資料和控制檯中檢視匯入效果
總結

1、根據業務需求設計資料庫表

2、根據資料庫表設計一個Excel模板

模板的每列屬性必須與表欄位一一對應

3、環境準備

我這裡專案環境是基於SpringBoot單體式架構，持久層用的公司框架，內建了基於MyBatis-Plus的各種單表操作的方法。

匯入依賴

        <!--使用POI讀取檔案-->
        <dependency>
            <groupId>org.apache.poi</groupId>
            <artifactId>poi</artifactId>
            <version>3.17</version>
        </dependency>
        <dependency>
            <groupId>org.apache.poi</groupId>
            <artifactId>poi-ooxml</artifactId>
            <version>3.17</version>
        </dependency>

4、通過外掛生成表對應的實體類

/**
 * TbZbzs: 值班值守表
 * @author zs
 * @date 2021-12-17 08:46:31
 **/
@Data
@ApiModel(value="值班值守表,對應表tb_zbzs",description="適用於新增和修改頁面傳參")
public class TbZbzs extends ProBaseEntity<TbZbzs> {
    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<List<Object>>物件
     * @param in,fileName
     * @return
     * @throws Exception
     */
    public static List<List<Object>> getListByExcel(InputStream in, String fileName) throws Exception {
        List<List<Object>> 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<List<Object>>();
        //遍歷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<Object> li = new ArrayList<Object>();
                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<String,Object> importTprkxx(MultipartFile file){
        Map<String,Object> resultMap = new HashMap<>();
        SimpleDateFormat simpleDateFormat = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
        List<TbZbzs> tbZbzsList = new ArrayList<>();
        try {
            //獲取資料
            List<List<Object>> olist = ImportExcelUtil.getListByExcel(file.getInputStream(), file.getOriginalFilename());
            resultMap.put("匯入成功",200);
            //封裝資料
            for (int i = 0; i < olist.size(); i++) {
                List<Object> 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) {
                    resultMap.put("state", "部門不能為空");
                    continue;
                }
                tbZbzs.setBm(String.valueOf(list.get(0)));
                if (String.valueOf(list.get(1))==null) {
                    resultMap.put("state", "值班上報不能為空");
                    continue;
                }
                tbZbzs.setZbsb(String.valueOf(list.get(1)));
                if (String.valueOf(list.get(2))==null) {
                    resultMap.put("state", "值班人員不能為空");
                    continue;
                }
                tbZbzs.setZbry(String.valueOf(list.get(2)));
                if (String.valueOf(list.get(3))==null) {
                    resultMap.put("state", "匯入失敗,上報時間不能為空");
                    continue;
                }
                String dateStr3 = String.valueOf(list.get(3));
                Date date3 = simpleDateFormat.parse(dateStr3);
                tbZbzs.setSbsj(date3);
                if (String.valueOf(list.get(4))==null) {
                    resultMap.put("state", "匯入失敗,結束時間不能為空");
                    continue;
                }
                String dateStr4 = String.valueOf(list.get(4));
                Date date4 = simpleDateFormat.parse(dateStr4);
                tbZbzs.setJssj(date4);
                if (String.valueOf(list.get(5))==null) {
                    resultMap.put("state", "聯絡方式不能為空");
                    continue;
                }
                tbZbzs.setLxfs(String.valueOf(list.get(5)));
                if (String.valueOf(list.get(6))==null) {
                    resultMap.put("state", "狀態不能為空");
                    continue;
                }
                tbZbzs.setZt(String.valueOf(list.get(6)));
                if (String.valueOf(list.get(7))==null) {
                    resultMap.put("state", "邏輯刪除不能為空");
                    continue;
                }
                tbZbzs.setDelFlag(String.valueOf(list.get(7)));
                if (String.valueOf(list.get(8))==null) {
                    resultMap.put("state", "匯入失敗,建立時間不能為空");
                    continue;
                }
                String dateStr8 = String.valueOf(list.get(8));
                Date date8 = simpleDateFormat.parse(dateStr8);
                tbZbzs.setCreateDate(date8);
                tbZbzsList.add(tbZbzs);
            }
            int i = tbZbzsDao.insertTbZbzsList(tbZbzsList);
            if (i != 0) {
                resultMap.put("state", "匯入成功");
            }else {
                resultMap.put("state", "匯入失敗");
            }
        } catch (Exception e) {
            e.printStackTrace();
            resultMap.put("state", "匯入失敗");
        }
        return resultMap;
    }

7、在dao層對應的xml檔案中，編寫批次上傳的方法

<insert id="insertTbZbzsList"  parameterType="java.util.List">
        insert into tb_zbzs (
        id,
        bm,
        zbsb,
        zbry,
        sbsj,
        jssj,
        lxfs,
        zt,
        del_flag,
        create_date
        ) VALUES
        <foreach collection="list" item="item" separator=",">
            (
            #{item.id},
            #{item.bm},
            #{item.zbsb},
            #{item.zbry},
            #{item.sbsj},
            #{item.jssj},
            #{item.lxfs},
            #{item.zt},
            #{item.delFlag},
            #{item.createDate}
            )
        </foreach>
        </insert>

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("匯入成功") ? buildSuccessResultData() : buildErrorResultData(service.importTprkxx(file).get("state").toString()));
        }catch (RuntimeException e){
            logError(log, e);
            return buildResultStr(buildErrorResultData(e));
        }
    }