package com.example.example1.Controller;
1 import com.example.example1.Entity.User; 2 import com.example.example1.Service.UserRepository; 3 import org.apache.poi.hssf.usermodel.*; 4 import org.apache.poi.ss.usermodel.Cell; 5 import org.apache.poi.ss.usermodel.Row; 6 import org.apache.poi.ss.usermodel.Sheet; 7 import org.apache.poi.ss.usermodel.Workbook; 8 import org.apache.poi.xssf.usermodel.XSSFWorkbook; 9 import org.springframework.beans.factory.annotation.Autowired; 10 import org.springframework.stereotype.Controller; 11 import org.springframework.web.bind.annotation.GetMapping; 12 import org.springframework.web.bind.annotation.RequestMapping; 13 import org.springframework.web.bind.annotation.ResponseBody; 14 15 import java.io.File; 16 import java.io.FileInputStream; 17 import java.io.IOException;
/** * Created by BLIT on 2018/11/30. */ @Controller @RequestMapping(path = "/excel") public class ImportController { @Autowired private UserRepository userRepository; @GetMapping(path = "/import") @ResponseBody public void Import() throws IOException{ File file = new File("C:\\Users\\zxg\\Desktop\\測試2003.xls"); //實際這個路徑由前端傳後臺 FileInputStream fis = new FileInputStream(file); Workbook wb = null; try { if(isExcel2003(file.getPath())){ System.out.println("2003版本Excel: .xls結尾"); wb = new HSSFWorkbook(fis); //建立2003版本Excel工做簿對象 }else if (isExcel2007(file.getPath())){ System.out.println("2007版本Excel: .xlsx結尾"); wb = new XSSFWorkbook(fis); //建立2007版本Excel工做簿對象 }else { System.out.println("未知版本的Excel !!!"); } Sheet sheet = wb.getSheetAt(0); //獲取第1個工做表 for(int i=1;i<=sheet.getLastRowNum();i++){//循環Excel文件的i=1行開始 User user = new User(); Row row = sheet.getRow(i); //獲取第i行 Cell cell1 = row.getCell(0); //獲取第1個單元格的數據 cell1.setCellType(Cell.CELL_TYPE_STRING); //設置Cell的類型爲String類型 Cell cell2 = row.getCell(1); //獲取第2個單元格的數據 cell2.setCellType(Cell.CELL_TYPE_STRING); user.setName(cell1.getStringCellValue()); user.setEmail(cell2.getStringCellValue()); System.out.println("第一單元格: " + cell1.getStringCellValue()); System.out.println("第二單元格: " + cell2.getStringCellValue()); userRepository.save(user); //保存 } } catch (Exception e) { e.printStackTrace(); }finally { fis.close(); } } public static boolean isExcel2003(String filePath) { return filePath.matches("^.+\\.(?i)(xls)$"); } public static boolean isExcel2007(String filePath) { return filePath.matches("^.+\\.(?i)(xlsx)$"); } }