SpringBoot導入Excel數據到MySQL數據庫

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;
View Code
/**
 * 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)$");
    }

}
相關文章
相關標籤/搜索