Springboot2模塊系列：poi（上傳excel保存至數據庫）

1 簡介

當數據量較大時，向數據庫一條條添加數據費時，易出錯，使用Excel導入數據較方便，省時省事。

2 Usage

2.1 pom.xml

<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>
		<dependency>
			<groupId>commons-io</groupId>
			<artifactId>commons-io</artifactId>
			<version>2.4</version>
    </dependency>
    <dependency>
			<groupId>commons-fileupload</groupId>
			<artifactId>commons-fileupload</artifactId>
			<version>1.3.1</version>
		</dependency>

2.2 數據類

package com.prjname.po;

import java.io.Serializable;

public class Database implements Serializable {
    private String name;
    private String address;
    private String sex;

    public void setName(String name) {
        this.name = name;
    }

    public String getName() {
        return name;
    }

    public void setAddress(String address) {
        this.address = address;
    }

    public String getAddress() {
        return address;
    }

    public void setSex(String sex) {
        this.sex = sex;
    }

    public String getSex() {
        return sex;
    }
}

2.3 讀取Excel內容工具類

package com.prjname.util;

import java.util.List;
import java.util.ArrayList;
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 org.springframework.web.multipart.MultipartFile;
import java.io.InputStream;
import com.prjname.po.Database;

public class ReadExcelUtil {
    public static List readExcel(MultipartFile file) throws Exception {
        List<Database> excelDatas = new ArrayList<>();
        // 獲取文件名
        String fileName = file.getOriginalFilename();
        // 獲取文件後綴
        String suffix = fileName.substring(fileName.lastIndexOf(".") + 1);
        // 獲取文件流
        InputStream ins = file.getInputStream();
        // 文件類
        Workbook wb = null;
        // 文件流讀取
        if (suffix.equals("xlsx")) {
            wb = new XSSFWorkbook(ins);
        } else {
            wb = new HSSFWorkbook(ins);
        }
        // 讀取文件流中sheet
        Sheet sheet = wb.getSheetAt(0);
        if (null != sheet) {
        	// 讀取sheet單元內容，line爲行，從第幾行開始讀取，行編號從0開始
            for (int line = 1; line <= sheet.getLastRowNum(); line++) {
                Database datas = new Database();
                // 每行行內容
                Row row = sheet.getRow(line);
                // 讀取行中列內容，列從0開始編號
                String name = row.getCell(0).getStringCellValue();
                String address = row.getCell(1).getStringCellValue();
                String sex = row.getCell(2).getStringCellValue();
                datas.setName(name);
                datas.setAddress(address);
                datas.setSex(sex);
                excelDatas.add(datas);
            }
        }
        return excelDatas;
    }
}

2.4 路由層

package com.prjname.controller;

import com.prjname.service.DatabaseService;
import com.prjname.po.Database;
import com.prjname.util.ReadExcelUtil;

import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
import com.xhwl.util.httpUtil.HttpClientUtil;
import com.xhwl.util.httpsUtil.*;
import java.text.DecimalFormat;
import java.util.HashMap;
import java.util.Map;
import java.util.List;
import java.util.Arrays;
import java.util.ArrayList;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.web.bind.annotation.CrossOrigin;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestMethod;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.multipart.MultipartFile;

@CrossOrigin(origins = "*", maxAge = 3600)
@RestController
@RequestMapping("/api/v1/databases")
public class DatasController {
    static Logger logger = LoggerFactory.getLogger(DatasController.class);
    @Autowired
    private DatabaseService databaseService;

    @RequestMapping(value = "/import-excel", method = RequestMethod.POST)
    public Map importExcel(@RequestParam(value = "filename") MultipartFile file) {
        Map returnMap = new HashMap();
        try {
            List<Database> returnLi = ReadExcelUtil.readExcel(file);
            for (Database datas : returnLi) {
                databaseService.addDatas(datas);
            }
            returnMap.put("code", 200);
            returnMap.put("msg", "SUCCESS");
            returnMap.put("infos", returnLi);
            return returnMap;
        } catch (Exception e) {
            e.printStackTrace();
        }
        returnMap.put("code", 400);
        returnMap.put("msg", "ERROR");
        returnMap.put("infos", "檢查參數");
        return returnMap;
    } 
}

2.5 Postman測試

圖2.1 導入數據請求方式

---

【參考文獻】
[1]https://blog.csdn.net/qq_37231511/article/details/93756623