SpringBoot實現Excel檔案批量上傳匯入資料庫
阿新 • • 發佈：2020-01-07
Spring boot + Spring data jpa + Thymeleaf
批量插入 + POI讀取 + 檔案上傳
pom.xml：
<!-- https://mvnrepository.com/artifact/org.apache.poi/poi --> <dependency> <groupId>org.apache.poi</groupId> <artifactId>poi</artifactId> <version>3.17</version> </dependency> <!-- https://mvnrepository.com/artifact/org.apache.poi/poi-ooxml --> <dependency> <groupId>org.apache.poi</groupId> <artifactId>poi-ooxml</artifactId> <version>3.17</version> </dependency>
upload.html：
<form enctype="multipart/form-data" method="post" action="/upload/excel"> 檔案 <input type="file" name="file" /> <input type="submit" value="上傳" /> </form>
如果自己的專案中使用了Spring security，頁面提交檔案之後，會出現403的錯誤，最快的解決辦法，如下：
http.csrf().ignoringAntMatchers("/upload/**").
在security的配置檔案中，加入上邊的程式碼即可。當然還有其他的辦法，大家可在網上查詢。
Controller：
package org.meng.project.controller;
import org.meng.project.service.ExcelService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Controller;
import org.springframework.ui.Model;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;
import javax.servlet.http.HttpServletRequest;
/**
*<p>上傳Controller類</p>
*<p> 上傳檔案的Controller</p>
* @Author MengMeng
* @Date 2018/10/6 </p>
* @version: 0.1
* @since JDK 1.80_144
*/
@Controller
@RequestMapping("/upload")
public class UploadController {
@Autowired
private HttpServletRequest request;
@Autowired
private ExcelService excelService;
//跳轉到上傳檔案的頁面
@RequestMapping(value = "",method = RequestMethod.GET)
public String goUpload() {
//跳轉到 templates 中tools目錄下的 upload.html
return "tools/upload";
}
@RequestMapping(value = "/excel",method = RequestMethod.POST)
public String upload(MultipartFile file,Model model) throws Exception {
boolean flag = excelService.getExcel(file);
if(flag){
model.addAttribute("Message","上傳成功");
}else{
model.addAttribute("Message","上傳失敗");
}
return "tools/upload";
}
}Excel實體：
package org.meng.project.entity;
import javax.persistence.*;
import java.io.Serializable;
import java.util.Objects;
/**
* <p>使用者實體類</p>
* @ClassName User
* @Author MengMeng
* @Date 2018/10/6 </p>
* @Version: 0.1
* @Since JDK 1.80_171
*/
@Entity
@Table(name = "test",schema = "project")
public class Excel implements Serializable {
private static final long serialVersionUID = 1L;
@Id
@Column(length=36)
private String id;
@Column(length=45,nullable=false,unique=true)
private String username;
@Column(length=100,unique=true)
private String email;
@Column(length=45,nullable=false)
private String password;
@Column(length=45)
private String role;
public Excel() {
}
public Excel(Excel user){
this.id = user.getId();
this.username = user.getUsername();
this.role = user.getRole();
this.email = user.getEmail();
this.password = user.getPassword();
}
//get 和 set
}
Service：
package org.meng.project.service;
import com.alibaba.fastjson.JSON;
import org.meng.project.entity.*;
import org.springframework.data.jpa.repository.Modifying;
import org.springframework.data.jpa.repository.Query;
import org.springframework.web.multipart.MultipartFile;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.transaction.Transactional;
/**
*<p>Excel的Service類介面</p>
*<p> Excel的Service類介面，與Excel有關的業務邏輯方法</p>
* @Author MengMeng
* @Date 2018/10/6 </p>
* @version: 0.1
* @since JDK 1.80_144
*/
public interface ExcelService {
boolean getExcel(MultipartFile file) throws Exception;
}
ServiceImpl：
package org.meng.project.service;
import org.apache.poi.EncryptedDocumentException;
import org.apache.poi.openxml4j.exceptions.InvalidFormatException;
import org.apache.poi.ss.usermodel.*;
import org.apache.poi.ss.usermodel.WorkbookFactory;
import org.meng.project.entity.*;
import org.meng.project.repository.ExcelRepository;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.web.multipart.MultipartFile;
import java.io.IOException;
import java.util.*;
/**
* <p>Excel的Service類</p>
* <p>Excel的Service類，與User有關的業務邏輯方法</p>
* @Author MengMeng
* @Date 2018/10/6
* @version: 0.1
* @since JDK 1.80_144
*/
@Service
public class ExcelServiceImpl implements ExcelService {
@Autowired
private ExcelRepository excelRepository;
@Override
public boolean getExcel(MultipartFile file) throws Exception {
// TODO Auto-generated method stub
List<Excel> list = new ArrayList<Excel>();
//1.得到上傳的表
Workbook workbook2 = WorkbookFactory.create(file.getInputStream());
//2、獲取test工作表
Sheet sheet2 = workbook2.getSheet("test");
//獲取表的總行數
int num = sheet2.getLastRowNum();
//System.out.println(num);
//總列數
int col = sheet2.getRow(0).getLastCellNum();
//遍歷excel每一行
for (int j = 0; j <= num; j++) {
Row row1 = sheet2.getRow(j);
//如果單元格中有數字或者其他格式的資料，則呼叫setCellType()轉換為string型別
Cell cell1 = row1.getCell(0);
cell1.setCellType(CellType.STRING);
//獲取表中第i行，第2列的單元格
Cell cell2 = row1.getCell(1);
//excel表的第i行，第3列的單元格
Cell cell3 = row1.getCell(2);
cell3.setCellType(CellType.STRING);
Cell cell4 = row1.getCell(3);
Cell cell5 = row1.getCell(4);
//這裡new 一個物件，用來裝填從頁面上傳的Excel資料，欄位根據上傳的excel決定
Excel excel= new Excel();
excel.setId(cell1.getStringCellValue());
excel.setEmail(cell2.getStringCellValue());
excel.setPassword(cell3.getStringCellValue());
excel.setRole(cell4.getStringCellValue());
excel.setUsername(cell5.getStringCellValue());
list.add(excel);
}
excelRepository.saveAll(list);
return true;
}
}
注意：需統一資料庫和Excel表中資料值的型別，不然會出現Cannot get a STRING value from a NUMERIC cell等錯誤。
Repository：
package org.meng.project.repository;
import org.meng.project.entity.Excel;
import org.meng.project.repository.base.BaseRepository;
import org.springframework.data.domain.Page;
import org.springframework.data.domain.Pageable;
import org.springframework.data.jpa.repository.Query;
import org.springframework.stereotype.Repository;
/**
*<p>Excel的DAO類介面</p>
*<p> Excel的DAO類介面，與Excel有關的持久化操作方法</p>
* @Author MengMeng
* @Date 2018/10/6 </p>
* @version: 0.1
* @since JDK 1.80_144
*/
@Repository
public interface ExcelRepository extends BaseRepository<Excel,String> {
}
即可實現檔案上傳：
以上就是本文的全部內容，希望對大家的學習有所幫助，也希望大家多多支援我們。