SpringBoot自定义工具类实现Excel数据存入MySQL数据库

/**
 * @Version: 1.0.0
 * @Author: Dragon_王
 * @ClassName: ExcelUtils
 * @Description: Excel相关工具类
 * @Date: 2024/3/9 11:24
 */

import cn.hutool.core.collection.CollUtil;
import com.alibaba.excel.EasyExcel;
import com.alibaba.excel.support.ExcelTypeEnum;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.ObjectUtils;
import org.apache.commons.lang3.StringUtils;
import org.springframework.web.multipart.MultipartFile;

import java.io.IOException;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

/**
 * Excel 相关工具类
 */
@Slf4j
public class ExcelUtils {

    /**
     * excel 转 csv
     *
     * @param multipartFile
     * @return
     */
    public static String excelToCsv(MultipartFile multipartFile) {
        // 读取数据
        List<Map<Integer, String>> list = null;
        try {
            list = EasyExcel.read(multipartFile.getInputStream()).excelType(ExcelTypeEnum.XLSX).sheet().headRowNumber(0).doReadSync();
        } catch (IOException e) {
            log.error("表格处理错误", e);
        }
        if (CollUtil.isEmpty(list)) {
            return "";
        }
        // 转换为 csv
        StringBuilder stringBuilder = new StringBuilder();
        // 读取表头
        LinkedHashMap<Integer, String> headerMap = (LinkedHashMap) list.get(0);
        List<String> headerList = headerMap.values().stream().filter(ObjectUtils::isNotEmpty).collect(Collectors.toList());
        stringBuilder.append(StringUtils.join(headerList, ",")).append("\n");
        // 读取数据
        for (int i = 1; i < list.size(); i++) {
            LinkedHashMap<Integer, String> dataMap = (LinkedHashMap) list.get(i);
            List<String> dataList = dataMap.values().stream().filter(ObjectUtils::isNotEmpty).collect(Collectors.toList());
            stringBuilder.append(StringUtils.join(dataList, ",")).append("\n");
        }
        return stringBuilder.toString();
    }
}

//传入的是Excel文件，不是路径哦
ExcelUtils.excelToCsv(Excel文件);

        //定义表名
        String tableName = "test";
        //获取第一个换行符的索引
        int index = csvData.indexOf("\n");
        //分割出属性名字符串，如"日期,阅读量"
        String colum = csvData.substring(0,index);
        //得到属性名数组，如{"日期","阅读量"}
        String[] colums = colum.split(",");
        //得到全部属性值字符串，如"3,253\n4,408\n5,363\n6,955\n7,496\n8,1310\n9,748"
        String data = csvData.substring(index).trim();
        //得到全部属性值数组，如：{"3,253","4,408","5,363""6,955".......}
        String[] split_data = data.split("\n");
        //调用建表
        tableCreationUtils.createTable(tableName,colums);
        //调用插入
        tableCreationUtils.Dynamicinsert(tableName,colums,split_data);

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

import javax.sql.DataSource;
import java.sql.SQLException;


/**
 * 构建生成表sql
 *
 * @version 1.0.0
 * @Author: dragon_王
 * @Date: 2024/3/11 20:45:16
 */
@Component
public class TableCreationUtils {

    private final JdbcTemplate jdbcTemplate;
    @Autowired
    public TableCreationUtils(DataSource dataSource){
        this.jdbcTemplate = new JdbcTemplate(dataSource);
    }

    /**
     * 动态构建创建表的sql语句并执行
     *
     * @param tableName 表名字符串
     * @param columnNames 属性名数组
     * @return: void
     * @Date: 2024-03-13 23:23:36
     */
    public  void createTable(String tableName, String[] columnNames) {
        String sql = "CREATE TABLE " + tableName + " (";
        for (int i = 0; i < columnNames.length; i++) {
            sql += columnNames[i] + " VARCHAR(255)";
            if (i < columnNames.length - 1) {
                sql += ", ";
            }
        }
        sql += ");";
        jdbcTemplate.execute(sql);
    }

    /**
     * 动态构建插入sql语句并执行
     *
     * @param tableName 表名字符串
     * @param columnName 属性名数组，如{"日期"，"阅读"}
     * @param rowData 属性值数组，如{"3,253","4,408","5,363""6,955".......}
     * @return: void
     * @Date: 2024-03-13 23:24:09
     */
    public void Dynamicinsert(String tableName, String[] columnName,String[] rowData) throws SQLException {
        //属性值为空就抛出异常
        if (rowData == null || rowData.length == 0) {
            throw new IllegalArgumentException("Row data must not be null or empty");
        }

        for (int i = 0;i < rowData.length;i++){
            //传进来的是所有属性值数组，如：{"3,253","4,408","5,363""6,955".......}
            //所以以将每个元素取出以英文逗号分割，得到插入的每行元素如["3","253"]
            String[] row = rowData[i].split(",");
            // 构建占位符
            StringBuilder columnNames = new StringBuilder();
            StringBuilder placeholders = new StringBuilder();
            for (int j = 0; j < row.length; j++) {
                if (j > 0) {
                    columnNames.append(", ");
                    placeholders.append(", ");
                }
                columnNames.append(columnName[j]);
                placeholders.append("?"); // 使用?作为PreparedStatement的占位符
            }

            // 构建完整的SQL语句
            String sql = "INSERT INTO " + tableName + " (" + columnNames + ") VALUES (" + placeholders + ")";

            // 使用JdbcTemplate执行插入操作，如第一次循环：insert into tablename (日期,阅读) values (?,?)
            //row：["3","253"]
            jdbcTemplate.update(sql,row);
        }
    }

}