excel表中数据导入到数据库中

一、引入依赖

<dependency>
            <groupId>org.apache.poi</groupId>
            <artifactId>poi</artifactId>
            <version>3.17</version>
        </dependency>

        <dependency>
            <groupId>org.apache.poi</groupId>
            <artifactId>poi-ooxml</artifactId>
            <version>3.17</version>
        </dependency>

二、工具类

直接上代码。

package com.wb.srpingboot.demoz.utils;

import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.ss.usermodel.*;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;

import java.io.InputStream;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * <p>
 * Description: poi将excle文件中的信息导入到数据库中工具类
 * </p>
 *
 * @author 阿萨
 * @version v1.0.0
 * @see com.wb.srpingboot.demoz.utils
 * @since 2020-08-25 19:42:28
 */
public class ImportExcel {
    private static final String XLS = "xls";
    private static final String XLSX = "xlsx";

    /**
     * TODO: 判断文件是否符合标准
     *
     * @param fileName 文件命
     * @return boolean
     */
    private static Result isXls(String fileName) {
        // (?i)忽略大小写
        if (fileName.matches("^.+\\.(?i)(xls)$")) {
            return Result.ok("xls");
        } else if (fileName.matches("^.+\\.(?i)(xlsx)$")) {
            return Result.ok("xlsx");
        } else {
            return Result.fail("文件格式不对");
        }
    }

    /**
     * TODO: 读取信息方法
     *
     * @param fileName    文件命
     * @param inputStream 文件
     * @return List<Map < String, Object>>
     */
    public static List<Map<String, Object>> readExcel(String fileName, InputStream inputStream) {

        Result ret = isXls(fileName);
        //System.out.println(ret);
        Workbook workbook = null;
        // 根据后缀创建不同的对象
        try {
            if (XLS.equals(ret.getMessage())) {
                workbook = new HSSFWorkbook(inputStream);
            } else if (XLSX.equals(ret.getMessage())) {
                workbook = new XSSFWorkbook(inputStream);
            } else {
                return null;
            }
            Sheet sheet = workbook.getSheetAt(0);
            // 得到标题行
            Row titleRow = sheet.getRow(0);

            int lastRowNum = sheet.getLastRowNum();
            int lastCellNum = titleRow.getLastCellNum();


            List<Map<String, Object>> list = new ArrayList<>();

            for (int i = 1; i <= lastRowNum; i++) {
                Map<String, Object> map = new HashMap<>();
                Row row = sheet.getRow(i);
                for (int j = 0; j < lastCellNum; j++) {
                    // 得到列名
                    String key = titleRow.getCell(j).getStringCellValue();
                    Cell cell = row.getCell(j);
                    cell.setCellType(CellType.STRING);

                    map.put(key, cell.getStringCellValue());
                }
                list.add(map);
            }
            return list;
        } catch (Exception e) {
            e.printStackTrace();
            return null;
        } finally {
            if (workbook != null) {
                try {
                    workbook.close();
                } catch (Exception e) {
                    e.printStackTrace();
                }
            }
        }
    }
}

三、controller代码

/**
     * TODO: 将上传的excel文件导入到数据库中(仅支持xls和xlsx文件)
     * @param mFile 上传的文件
     * @return com.wb.srpingboot.demoz.utils.Result
     */
    @RequestMapping("/uploadExcel")
    @ResponseBody
    public Result importExcel(@RequestParam(value = "mFile")MultipartFile mFile) {

        InputStream inputStream = null;
        try {
            String fileName = mFile.getOriginalFilename();
            // 获取上传文件的输入流
            inputStream = mFile.getInputStream();
            // 调用工具类中方法,读取excel文件中数据
            List<Map<String, Object>> sourceList = ImportExcel.readExcel(fileName, inputStream);

            // 将对象先转为json格式字符串,然后再转为List<SysUser> 对象
            ObjectMapper objMapper = new ObjectMapper();
            String infos = objMapper.writeValueAsString(sourceList);

            // json字符串转对象
            List<User> list = objMapper.readValue(infos, new TypeReference<List<User>>() {});

            if (list != null && list.size() >0) {
                // 批量添加
                return userService.userAdds(list);
            }
            return Result.fail("文件格式错误,仅支持xls和xlsx格式的文件");

        } catch (Exception e) {
            e.printStackTrace();
            return Result.fail("文件格式错误");
        }finally {
            if (inputStream != null ){
                try {
                    inputStream.close();
                }catch (Exception e){
                    e.printStackTrace();
                }
            }
        }
    }

四、给出Result返回值类

我还是自定义了一个返回值类型,可以自己根据自己公司的规范进行更改。

package com.wb.srpingboot.demoz.utils;

import java.io.Serializable;

/**
 * Controller中处理方法的返回值类型
 * @author 阿萨
 * @since 1.0
 */
public class Result implements Serializable {

    // 状态码
    private int code;
    // 消息
    private String message;
    // 数据
    private Object data;

    public Result() { }

    public Result(int code, String message, Object data) {
        this.code = code;
        this.message = message;
        this.data = data;
    }

    public Result(int code, String message) {
        this(code, message, null);
    }

    public Result(int code, Object data) {
        this(code, null, data);
    }

    public static Result ok(String message, Object data) {
        return new Result(0, message, data);
    }

    public static Result ok(String message) {
        return new Result(0, message, null);
    }

    public static Result ok(Object data) {
        return new Result(0, null, data);
    }

    public static Result fail(String message) {
        return new Result(-1, message);
    }

    public static Result fail(String message, Object data) {
        return new Result(-1, message, data);
    }

    public static Result fail(Object data) {
        return new Result(-1, null, data);
    }

    public int getCode() {
        return code;
    }

    public void setCode(int code) {
        this.code = code;
    }

    public String getMessage() {
        return message;
    }

    public void setMessage(String message) {
        this.message = message;
    }

    public Object getData() {
        return data;
    }

    public void setData(Object data) {
        this.data = data;
    }

    @Override
    public String toString() {
        return "Result{" +
                "code=" + code +
                ", message='" + message + '\'' +
                ", data=" + data +
                '}';
    }
}

感谢观看。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
你可以使用 Apache POI 库来解析 Excel 表中数据,并使用 JDBC 将数据导入数据库。是一个简单的 Java 代码示: 首先,确保你已经将 POI 和 JDBC 相关的 JAR 文件添加到你的项目。 ```java import java.FileInputStream; import java.sql; import java.sql.DriverManager; import java.sql.PreparedStatement; import java.sql.SQLException; 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; public class ExcelImporter { public static void main(String[] args) { String excelFilePath = "path/to/your/excel/file.xlsx"; try (FileInputStream inputStream = new FileInputStream(excelFilePath); Workbook workbook = new XSSFWorkbook(inputStream)) { Sheet sheet = workbook.getSheetAt(0); int rowCount = 0; for (Row row : sheet) { if (rowCount == 0) { rowCount++; continue; } String column1 = row.getCell(0).getStringCellValue(); int column2 = (int) row.getCell(1).getNumericCellValue(); // 获取其他列的数据 // 将数据插入数据库 insertToDatabase(column1, column2); rowCount++; } System.out.println("数据导入成功!"); } catch (Exception e) { e.printStackTrace(); } } private static void insertToDatabase(String column1, int column2) { String url = "jdbc:mysql://localhost:3306/your_database"; String username = "your_username"; String password = "your_password"; try (Connection connection = DriverManager.getConnection(url, username, password)) { String sql = "INSERT INTO your_table (column1, column2) VALUES (?, ?)"; PreparedStatement statement = connection.prepareStatement(sql); statement.setString(1, column1); statement.setInt(2, column2); statement.executeUpdate(); } catch (SQLException e) { e.printStackTrace(); } } } ``` 在上面的示例,你需要将 `path/to/your/excel/file.xlsx` 替换为你实际的 Excel 文件路径。此外,你还需要根据你的数据库设置更新 `url`、`username`、`password`、`your_database` 和 `your_table`。 这个示例假设 Excel 表的第一行为标题,从第二行开始是数据。你可以根据需要调整代码以适应不同的表格结构。同时,你可以根据你的数据库表结构调整 `insertToDatabase()` 方法的 SQL 语句和参数设置。 请确保你已正确引入所需的库和驱动程序,并在执行代码之前设置好数据库连接信息。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值