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> { }
即可實現檔案上傳:
以上就是本文的全部內容,希望對大家的學習有所幫助,也希望大家多多支援我們。