Apache POI操作批次匯入MySQL資料庫

2022-06-20 22:03:48

HSSF － 提供讀寫Microsoft Excel XLS格式檔案的功能
XSSF － 提供讀寫Microsoft Excel OOXML XLSX格式檔案的功能
HWPF － 提供讀寫Microsoft Word DOC格式檔案的功能
HSLF － 提供讀寫Microsoft PowerPoint格式檔案的功能
HDGF － 提供讀Microsoft Visio格式檔案的功能
HPBF － 提供讀Microsoft Publisher格式檔案的功能
HSMF － 提供讀Microsoft Outlook格式檔案的功能

1、POI操作入門案例

1.1、從Excel檔案讀取資料1

使用POI可以從一個已經存在的Excel檔案中讀取資料

前提需要建立一個需要讀取的表格資料進行讀取

/**
 * 使用poi讀取表格資料
 * @throws Exception
 */
@Test
public void test1() throws Exception {
    // 1、載入指定的檔案進行讀取
    XSSFWorkbook excel = new XSSFWorkbook(new FileInputStream("C:\Users\zhong\Desktop\poi.xlsx"));
    // 2、讀取表格中的Sheet頁，通過索引決定
    XSSFSheet sheetAt = excel.getSheetAt(0);
    // 3、讀取Sheet頁中的行資料
    for (Row row : sheetAt) {
        // 4、讀取每一行資料的單元格資料（如果涉及到型別裝轉換的可能出現報錯訊息，後期通過程式碼進行判斷即可）
        for (Cell cell : row) {
            System.out.print(cell+"   ");
        }
        System.out.println();
    }
    // 5、關閉讀取檔案的流
    excel.close();
}

輸出結果如下：

姓名省份城市
張三廣東高州
李四四川成都

POI操作Excel表格封裝了幾個核心物件：

XSSFWorkbook：工作簿
XSSFSheet：工作表
Row：行
Cell：單元格

1.2、從Excel檔案讀取資料2

還有一種方式就是獲取工作表最後一個行號，從而根據行號獲得行物件，通過行獲取最後一個單元格索引，從而根據單元格索引獲取每行的一個單元格物件，程式碼如下：

/**
 * 使用poi讀取檔案的第二種方式
 * @throws Exception
 */
@Test
public void test2() throws Exception {
    // 1、載入指定的檔案進行讀取
    XSSFWorkbook excel = new XSSFWorkbook(new FileInputStream("C:\Users\zhong\Desktop\poi.xlsx"));
    // 2、讀取表格中的Sheet頁，通過索引決定
    XSSFSheet sheetAt = excel.getSheetAt(0);
    // 3、獲取當前工作表中最後一個行號,注意行號是從0開啟的
    int lastRowNum = sheetAt.getLastRowNum();
    // 4、遍歷獲取到的行號
    for (int i = 0; i <= lastRowNum; i++) {
        // 5、根據行號獲取到每一行的資料
        XSSFRow row = sheetAt.getRow(i);
        // 6、獲取到當前最後一個單元格索引
        short lastCellNum = row.getLastCellNum();
        // 7、遍歷當前的單元格獲取到具體的資料
        for (int j = 0; j < lastCellNum; j++) {
            // 獲取到單元格的物件
            XSSFCell cell = row.getCell(j);
            // 輸出獲取到的資料
            System.out.print(cell + "   ");
        }
        System.out.println();
    }
    // 8、釋放資源
    excel.close();
}

1.3、向Excel檔案寫入資料

使用POI可以在記憶體中建立一個Excel檔案並將資料寫入到這個檔案，最後通過輸出流將記憶體中的Excel檔案下載到磁碟

/**
 * poi寫出資料到磁碟
 */
@Test
public void test3() throws Exception {
    // 1、建立工作簿
    XSSFWorkbook excel = new XSSFWorkbook();
    // 2、建立工作簿中的表物件
    XSSFSheet sheet = excel.createSheet("建立表");
    // 3、建立第一行（表頭）
    XSSFRow row1 = sheet.createRow(0);
    // 4、在行中建立單元格資料
    row1.createCell(0).setCellValue("姓名");
    row1.createCell(1).setCellValue("省份");
    row1.createCell(2).setCellValue("城市");
    row1.createCell(3).setCellValue("年齡");
    // 5、建立第二行
    XSSFRow row2 = sheet.createRow(1);
    // 6、建立第6行的資料
    row2.createCell(0).setCellValue("張三");
    row2.createCell(1).setCellValue("遼寧");
    row2.createCell(2).setCellValue("上海");
    row2.createCell(3).setCellValue("50");
    // 7、建立一個位元組輸出流，將資料儲存到本地
    FileOutputStream fileOutputStream = new FileOutputStream(new File("C:\Users\zhong\Desktop\aaa.xlsx"));
    excel.write(fileOutputStream);
    fileOutputStream.flush();
    // 8、關閉輸出
    excel.close();
    System.out.println("資料匯出成功");
}

2、使用POI批次匯入資料到MySQL資料庫

建立一個資料庫的表t_ordersetting

-- auto-generated definition
create table t_ordersetting
(
    id           int auto_increment
        primary key,
    orderDate    date null comment '約預日期',
    number       int  null comment '可預約人數',
    reservations int  null comment '已預約人數'
)
    charset = utf8;

批次匯入預約設定資訊操作過程：

1、點選模板下載按鈕下載Excel模板檔案

2、將預約設定資訊錄入到模板檔案中

3、點選上傳檔案按鈕將錄入完資訊的模板檔案上傳到伺服器

4、通過POI讀取上傳檔案的資料並儲存到資料庫

建立對應的實體類

public class OrderSetting implements Serializable{
    private Integer id ;
    private Date orderDate;//預約設定日期
    private int number;//可預約人數
    private int reservations ;//已預約人數

    public OrderSetting() {
    }

    public OrderSetting(Date orderDate, int number) {
        this.orderDate = orderDate;
        this.number = number;
    }
}

2.1、建立匯入資料的Excel模板檔案

目的是為了統一管理匯入的資料格式

修改頁面提供一個下載批次匯入資料的模板按鈕

//下載模板檔案
downloadTemplate(){
    window.location.href="../../template/ordersetting_template.xlsx" rel="external nofollow" ;
}

檔案上傳前端程式碼實現

<el-upload action="/ordersetting/upload.do"
           name="excelFile"
           :show-file-list="false"
           :on-success="handleSuccess"
           :before-upload="beforeUpload">
    <el-button type="primary">上傳檔案</el-button>
</el-upload>

提交函數

// 上傳之前進行檔案格式校驗
beforeUpload(file){
    const isXLS = file.type === 'application/vnd.ms-excel';
    if(isXLS){
        return true;
    }
    const isXLSX = file.type === 'application/vnd.openxmlformats-officedocument.spreadsheetml.sheet';
    if (isXLSX) {
        return true;
    }
    this.$message.error('上傳檔案只能是xls或者xlsx格式!');
    return false;
},
// 上傳成功提示
handleSuccess(response, file) {
    if(response.flag){
        this.$message({
            message: response.message,
            type: 'success'
        });
    }else{
        this.$message.error(response.message);
    }
    console.log(response, file, fileList);
},

2.2、批次上傳檔案的後端程式碼編寫

關於上傳檔案一般網上都會有很多的工具類可以下載使用，也就是別人將已經封裝好的程式碼拿出來分享給大家使用的

package com.zcl.utils;

import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.InputStream;
import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.List;
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;

public class POIUtils {
    private final static String xls = "xls";
    private final static String xlsx = "xlsx";
    private final static String DATE_FORMAT = "yyyy/MM/dd";
    /**
     * 讀入excel檔案，解析後返回
     * @param file
     * @throws IOException
     */
    public static List<String[]> readExcel(MultipartFile file) throws IOException {
        //檢查檔案
        checkFile(file);
        //獲得Workbook工作薄物件
        Workbook workbook = getWorkBook(file);
        //建立返回物件，把每行中的值作為一個陣列，所有行作為一個集合返回
        List<String[]> list = new ArrayList<String[]>();
        if(workbook != null){
            for(int sheetNum = 0;sheetNum < workbook.getNumberOfSheets();sheetNum++){
                //獲得當前sheet工作表
                Sheet sheet = workbook.getSheetAt(sheetNum);
                if(sheet == null){
                    continue;
                }
                //獲得當前sheet的開始行
                int firstRowNum  = sheet.getFirstRowNum();
                //獲得當前sheet的結束行
                int lastRowNum = sheet.getLastRowNum();
                //迴圈除了第一行的所有行
                for(int rowNum = firstRowNum+1;rowNum <= lastRowNum;rowNum++){
                    //獲得當前行
                    Row row = sheet.getRow(rowNum);
                    if(row == null){
                        continue;
                    }
                    //獲得當前行的開始列
                    int firstCellNum = row.getFirstCellNum();
                    //獲得當前行的列數
                    int lastCellNum = row.getPhysicalNumberOfCells();
                    String[] cells = new String[row.getPhysicalNumberOfCells()];
                    //迴圈當前行
                    for(int cellNum = firstCellNum; cellNum < lastCellNum;cellNum++){
                        Cell cell = row.getCell(cellNum);
                        cells[cellNum] = getCellValue(cell);
                    }
                    list.add(cells);
                }
            }
            workbook.close();
        }
        return list;
    }

    //校驗檔案是否合法
    public static void checkFile(MultipartFile file) throws IOException{
        //判斷檔案是否存在
        if(null == file){
            throw new FileNotFoundException("檔案不存在！");
        }
        //獲得檔名
        String fileName = file.getOriginalFilename();
        //判斷檔案是否是excel檔案
        if(!fileName.endsWith(xls) && !fileName.endsWith(xlsx)){
            throw new IOException(fileName + "不是excel檔案");
        }
    }
    public static Workbook getWorkBook(MultipartFile file) {
        //獲得檔名
        String fileName = file.getOriginalFilename();
        //建立Workbook工作薄物件，表示整個excel
        Workbook workbook = null;
        try {
            //獲取excel檔案的io流
            InputStream is = file.getInputStream();
            //根據檔案字尾名不同(xls和xlsx)獲得不同的Workbook實現類物件
            if(fileName.endsWith(xls)){
                //2003
                workbook = new HSSFWorkbook(is);
            }else if(fileName.endsWith(xlsx)){
                //2007
                workbook = new XSSFWorkbook(is);
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
        return workbook;
    }
    public static String getCellValue(Cell cell){
        String cellValue = "";
        if(cell == null){
            return cellValue;
        }
        //如果當前單元格內容為日期型別，需要特殊處理
        String dataFormatString = cell.getCellStyle().getDataFormatString();
        if(dataFormatString.equals("m/d/yy")){
            cellValue = new SimpleDateFormat(DATE_FORMAT).format(cell.getDateCellValue());
            return cellValue;
        }
        //把數位當成String來讀，避免出現1讀成1.0的情況
        if(cell.getCellType() == Cell.CELL_TYPE_NUMERIC){
            cell.setCellType(Cell.CELL_TYPE_STRING);
        }
        //判斷資料的型別
        switch (cell.getCellType()){
            case Cell.CELL_TYPE_NUMERIC: //數位
                cellValue = String.valueOf(cell.getNumericCellValue());
                break;
            case Cell.CELL_TYPE_STRING: //字串
                cellValue = String.valueOf(cell.getStringCellValue());
                break;
            case Cell.CELL_TYPE_BOOLEAN: //Boolean
                cellValue = String.valueOf(cell.getBooleanCellValue());
                break;
            case Cell.CELL_TYPE_FORMULA: //公式
                cellValue = String.valueOf(cell.getCellFormula());
                break;
            case Cell.CELL_TYPE_BLANK: //空值
                cellValue = "";
                break;
            case Cell.CELL_TYPE_ERROR: //故障
                cellValue = "非法字元";
                break;
            default:
                cellValue = "未知型別";
                break;
        }
        return cellValue;
    }
}

2.2.1、建立批次上傳控制器

這裡使用的dubbo遠端呼叫了介面方法

/**
 * 專案名稱：health_parent
 * 描述：預約管理控制器
 *
 * @author zhong
 * @date 2022-06-18 14:50
 */
@RestController
@RequestMapping("/ordersetting")
public class OrderSettingController {
    /**
     * 遠端呼叫預約管理的服務介面
     */
    @Reference
    private OrderSettingService orderSettingService;

    /**
     * 檔案上傳，實現預約管理的批次匯入
     * @param excelFile
     * @return
     */
    @RequestMapping("/upload")
    public Result upload(@RequestParam("excelFile") MultipartFile excelFile){
        try {
            // 1、使用工具類解析上傳的資料
            List<String[]> list = POIUtils.readExcel(excelFile);
            // 將讀取的表格資料轉換為表格預約物件資料
            List<OrderSetting> data = new ArrayList<>();
            // 2、處理解析的資料
            for (String[] strings : list) {
                // 獲取到匯入資料行的第一個單元格(預約時間)
                String orderSate = strings[0];
                // 獲取到匯入資料行的第二個單元格（預約姓名）
                String number = strings[1];
                // 將重新獲取到的資料新增到預約集合中
                OrderSetting orderSetting = new OrderSetting(new Date(orderSate), Integer.parseInt(number));
                data.add(orderSetting);
            }
            // 2、通過dubbo遠端呼叫服務實現資料批次匯入到資料庫
            orderSettingService.add(data);

        } catch (IOException e) {
            e.printStackTrace();
            return new Result(false, MessageConstant.IMPORT_ORDERSETTING_FAIL);
        }
        return new Result(true, MessageConstant.IMPORT_ORDERSETTING_SUCCESS);
    }
}

2.2.2、建立批次上傳的介面實現類

package com.zcl.service.impl;

import com.alibaba.dubbo.config.annotation.Service;
import com.itheima.pojo.OrderSetting;
import com.zcl.dao.OrderSettingDao;
import com.zcl.service.OrderSettingService;
import org.springframework.transaction.annotation.Transactional;

import java.util.List;

/**
 * 專案名稱：health_parent
 * 描述：預約管理介面實現類
 *
 * @author zhong
 * @date 2022-06-18 15:09
 */
@Service(interfaceClass = OrderSettingService.class)
@Transactional
public class OrderSettingImpl implements OrderSettingService {
    /**
     * 注入儲存資料
     */
    @Autowired
    private OrderSettingDao orderSettingDao;

    /**
     * 批次上傳業務實現
     * @param data
     */
    @Override
    public void add(List<OrderSetting> data) {
        // 判斷集合資料是否為空
        if(data != null && data.size() > 0){
            // 判斷當前日期是否已經進行了預約設定
            for (OrderSetting datum : data) {
                // 根據日期查詢預約訊息
                long count = orderSettingDao.findCountByOrderDate(datum.getOrderDate());
                if(count > 0){
                    // 已進行預約管理，執行更新操作
                    orderSettingDao.editNumberByOrderDate(datum);
                }else{
                    // 新增預約設定
                    orderSettingDao.add(datum);
                }
            }
        }else{

        }
    }
}

2.2.3、建立資料存取層介面和對映檔案

package com.zcl.dao;

import com.itheima.pojo.OrderSetting;

import java.util.Date;

/**
 * 專案名稱：health_parent
 * 描述：預約管理資料存取層
 *
 * @author zhong
 * @date 2022-06-18 15:13
 */
public interface OrderSettingDao {
    /**
     * 新增預約資料
     * @param orderSetting
     */
    void add(OrderSetting orderSetting);

    /**
     * 修改預約資料
     * @param orderSetting
     */
    void editNumberByOrderDate(OrderSetting orderSetting);

    /**
     * 查詢預約資料的總數
     * @param orderDate
     * @return
     */
    long findCountByOrderDate(Date orderDate);
}

資料存取層對映檔案編寫SQL語句

<?xml version="1.0" encoding="UTF-8" ?>
<!DOCTYPE mapper
        PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN"
        "http://mybatis.org/dtd/mybatis-3-mapper.dtd">
<mapper namespace="com.zcl.dao.OrderSettingDao">
    <!--新增預約嘻嘻嘻-->
    <insert id="add" parameterType="com.itheima.pojo.OrderSetting">
        insert into t_ordersetting (orderDate,number,reservations)
        values (#{orderDate},#{number},#{reservations});
    </insert>

    <!--根據日期修改預約資訊-->
    <update id="editNumberByOrderDate" parameterType="com.itheima.pojo.OrderSetting">
        update t_ordersetting
        set number = #{number}
        where orderDate = #{orderDate}
    </update>

    <!--根據日期查詢資料-->
    <select id="findCountByOrderDate" parameterType="date" resultType="java.lang.Long">
        select count(id)
        from t_ordersetting
        where orderDate = #{orderDate}
    </select>
</mapper>

3、批次匯入資料測試

在上傳模板上製作資料,然後進行匯入，先匯入新的資料再匯入修改後日期不變的資料再次上傳資料

到此這篇關於Apache POI操作批次匯入MySQL資料庫的文章就介紹到這了,更多相關Apache POI操作批次匯入MySQL內容請搜尋it145.com以前的文章或繼續瀏覽下面的相關文章希望大家以後多多支援it145.com！