springboot 根据excel动态在mysql创建表

黄啸宇

于 2024-08-28 04:08:35 发布

阅读量51

点赞数

我整理的一些关于【mysql】的项目学习资料（附讲解～～）和大家一起分享、学习一下：

https://d.51cto.com/yOSbkR

使用 Spring Boot 根据 Excel 动态在 MySQL 创建表的完整指南

在现代应用开发中，Excel 文件常用于数据导入。如何在 Spring Boot 项目中根据 Excel 文件动态创建 MySQL 数据表，成为了一个较为有趣的挑战。本文章将带你一步一步完成这个任务，具体分为几个步骤：

流程概述

步骤	说明
1	创建 Spring Boot 项目
2	导入所需依赖
3	读取 Excel 文件
4	解析 Excel 数据
5	生成 SQL 创建表语句
6	执行 SQL 语句创建表
7	完成并测试

步骤1: 创建 Spring Boot 项目

在开始之前，你需要创建一个新的 Spring Boot 项目。可以使用 [Spring Initializr]( 创建项目，选择需要的依赖，如 Spring Web 和 Spring Data JPA。

步骤2: 导入所需依赖

在 pom.xml 文件中添加以下依赖：

<dependency>
    <groupId>org.apache.poi</groupId>
    <artifactId>poi-ooxml</artifactId>
    <version>5.0.0</version>
</dependency>
<dependency>
    <groupId>mysql</groupId>
    <artifactId>mysql-connector-java</artifactId>
    <version>8.0.26</version>
</dependency>

这里，我们使用 Apache POI 库来处理 Excel 文件，MySQL Connector 作为数据库连接器。

步骤3: 读取 Excel 文件

使用 Apache POI 读取 Excel 文件。创建一个服务类并添加读取 Excel 的逻辑。

import org.apache.poi.ss.usermodel.*;
import org.springframework.stereotype.Service;

import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;

@Service
public class ExcelService {

    public List<String[]> readExcel(String filePath) throws IOException {
        List<String[]> data = new ArrayList<>();
        FileInputStream fis = new FileInputStream(new File(filePath));
        Workbook workbook = WorkbookFactory.create(fis);
        Sheet sheet = workbook.getSheetAt(0);
        
        for (Row row : sheet) {
            String[] rowData = new String[row.getPhysicalNumberOfCells()];
            for (int i = 0; i < row.getPhysicalNumberOfCells(); i++) {
                Cell cell = row.getCell(i);
                rowData[i] = cell.toString(); // 读取单元格数据
            }
            data.add(rowData);
        }
        workbook.close();
        return data; // 返回读取到的数据
    }
}

步骤4: 解析 Excel 数据

在读取数据后，你需要解析数据以生成 SQL 语句。假设 Excel 的第一行是表头。

public String generateCreateTableSQL(String tableName, List<String[]> data) {
    StringBuilder sql = new StringBuilder("CREATE TABLE " + tableName + " (");
    String[] headers = data.get(0); // 取第一行作为表头
    
    for (String header : headers) {
        sql.append(header).append(" VARCHAR(255), "); // 假设所有字段都是 VARCHAR
    }
    
    sql.setLength(sql.length() - 2); // 删除最后的逗号和空格
    sql.append(");");
    
    return sql.toString(); // 返回创建表的 SQL 语句
}

步骤5: 执行 SQL 语句创建表

使用 Spring Data JPA 来执行生成的 SQL 语句。

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.jdbc.core.JdbcTemplate;
import org.springframework.stereotype.Service;

@Service
public class TableService {

    @Autowired
    private JdbcTemplate jdbcTemplate;

    public void createTable(String sql) {
        jdbcTemplate.execute(sql); // 执行 SQL 语句创建表
    }
}

步骤6: 完成并测试

在你的控制器中，结合以上所有步骤，完成整个流程。

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;

import java.io.IOException;
import java.util.List;

@RestController
public class ExcelController {

    @Autowired
    private ExcelService excelService;

    @Autowired
    private TableService tableService;

    @PostMapping("/createTable")
    public String createTableFromExcel(@RequestParam String filePath) throws IOException {
        List<String[]> data = excelService.readExcel(filePath); // 读取 Excel
        String tableName = "DynamicTable"; // 可以根据需要动态生成表名
        String sql = excelService.generateCreateTableSQL(tableName, data); // 生成 SQL
        tableService.createTable(sql); // 创建表
        return "Table created successfully!";
    }
}