excel批量数据导入时用poi将数据转化成指定实体工具类

1.实现目标

excel进行批量数据导入时,将批量数据转化成指定的实体集合用于数据操作,实现思路:使用注解将属性与表格中的标题进行同名绑定来赋值。

2.代码实现

2.1 目录截图如下

在这里插入图片描述

2.2 代码实现
package poi.constants;

/**
 * @description: 用来定义一些通用的常量数据
 * @author: zengwenbo
 * @date: 2024/3/10 13:08
 */
public class Constant {

    public final static String POINT = ".";
    public final static String SPACE = " ";
}

package poi.exception;

/**
 * @description: 用于解析excel报错提供的异常
 * @author: zengwenbo
 * @date: 2024/3/10 12:47
 */
public class ExcelException extends RuntimeException {
    public ExcelException() {
        super();
    }

    public ExcelException(String message) {
        super(message);
    }

    public ExcelException(String message, Throwable cause) {
        super(message, cause);
    }
}

package poi.annotation;

import org.apache.poi.ss.usermodel.Cell;
import org.apache.poi.ss.usermodel.DateUtil;
import org.springframework.util.StringUtils;

import java.lang.annotation.*;
import java.util.List;

import static poi.constants.Constant.SPACE;



/**
 * @description: 用于绑定excel和实体字段属性的注解
 * @author: zengwenbo
 * @date: 2024/3/10 12:51
 */
@Documented
@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.FIELD)
public @interface ExcelDescription {
    String desc(); // excel描述的标题的内容
    DataType type() default DataType.String; // 指定当前title数据的数据类型
    enum DataType {
        String {
            @Override
            public Object evaluateDataByType(Cell dataCell, List<List<java.lang.String>> validateList) {
                String cellValue = dataCell.getStringCellValue();
                if (StringUtils.hasLength(cellValue) && StringUtils.hasLength(cellValue.trim())) {
                    // 判断当前值是否在序列中,在的话用SPACE进行划分取前面的值,不在的话原值返回
                    return validateList.stream()
                            .filter(item -> item.contains(cellValue))
                            .map(item -> cellValue.split(SPACE)[0])
                            .findFirst()
                            .orElse(cellValue);
                }
                return cellValue;
            }
        }, Date {
            @Override
            public Object evaluateDataByType(Cell dataCell, List<List<java.lang.String>> validateList) {
                double cellValue = dataCell.getNumericCellValue();
                return cellValue != 0 ? DateUtil.getJavaDate(cellValue) : null;
            }
        }, BigDecimal {
            @Override
            public Object evaluateDataByType(Cell dataCell, List<List<java.lang.String>> validateList) {
                return java.math.BigDecimal.valueOf(dataCell.getNumericCellValue());
            }
        };

        /**
         * 根据数据类型来获取excel的数据值
         *
         * @param dataCell excel单元格对象
         * @param validateList excel当前的序列数据有效性
         * @return
         */
        public abstract Object evaluateDataByType(Cell dataCell, List<List<String>> validateList);
    }
}

package poi.utils;

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.util.StringUtil;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
import org.springframework.util.StringUtils;
import org.springframework.web.multipart.MultipartFile;
import poi.annotation.ExcelDescription;
import poi.exception.ExcelException;

import java.io.IOException;
import java.io.InputStream;
import java.lang.reflect.Field;
import java.nio.file.OpenOption;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.Optional;

import static poi.constants.Constant.POINT;

/**
 * @description: 提供解析excel的工具类
 * @author: zengwenbo
 * @date: 2024/3/10 12:42
 */
public class ExcelUtil {
    private final static String EXCEL_XLS = "xls";
    private final static String EXCEL_XLSX = "xlsx";

    /**
     * 将excel中table里面的数据转换成对应的实体
     *
     * @param clazz       需要转化实体的类对象
     * @param titleRowNum table表格的标题行
     * @param sheetIndex  sheet的下标
     * @return 返回转换后对象的list集合
     */
    public static <T> List<T> transTableToEntity(MultipartFile file, Class<T> clazz, int titleRowNum, int sheetIndex) {
        // 1.获取文件的后缀
        String filename = file.getOriginalFilename();
        String suffix = filename.substring(filename.lastIndexOf(POINT) + 1);

        // 2.根据获取的后缀名获取操作excel的对象
        Workbook workbook;
        try (InputStream inputStream = file.getInputStream()) {
            switch (suffix) {
                case EXCEL_XLS:
                    workbook = new HSSFWorkbook(inputStream);
                    break;
                case EXCEL_XLSX:
                    workbook = new XSSFWorkbook(inputStream);
                    break;
                default:
                    throw new ExcelException("后缀名不符");
            }
        } catch (IOException e) {
            throw new ExcelException("文件解析失败", e);
        }

        // 3.获取要操作的sheet
        Sheet sheet = workbook.getSheetAt(sheetIndex);

        // 4.通过表格标题获取操作的开始列和结束列
        Row titleRow = sheet.getRow(titleRowNum);
        short firstCellNum = titleRow.getFirstCellNum();
        short lastCellNum = titleRow.getLastCellNum();

        // 5.获取表格中序列的数据有效性
        List<List<String>> validateList = new ArrayList<>();
        if (null != sheet.getDataValidations()) {
            sheet.getDataValidations().forEach(item -> {
                // 筛选出有效性数据时序列的进行添加
                if (null != item.getValidationConstraint().getExplicitListValues()) {
                    validateList.add(Arrays.asList(item.getValidationConstraint().getExplicitListValues()));
                }
            });
        }

        // 6.遍历数据进行解析
        ArrayList<T> list = new ArrayList<>();
        Field[] fields = clazz.getDeclaredFields();
        for (int i = titleRowNum + 1; i < sheet.getLastRowNum(); i++) {
            Row row = sheet.getRow(i);
            try {
                // 获取实例对每个绑定的属性进行赋值
                T t = clazz.newInstance();
                for (int j = firstCellNum; j < lastCellNum; j++) {
                    Cell titleCell = titleRow.getCell(j);
                    if (null == titleCell) {
                        throw new ExcelException("标题缺失,请检查导入模板是否正常");
                    }
                    String title = titleCell.getStringCellValue();
                    if (!StringUtils.hasLength(title)) {
                        throw new ExcelException("标题内容为空,请检查导入模板是否正常");
                    }
                    Optional.ofNullable(row.getCell(j))
                            .ifPresent(value -> evaluateField(t, fields, value, title, validateList));
                }
                list.add(t);
            } catch (InstantiationException e) {
                throw new ExcelException("创建实例异常,该类缺失无参构造方法");
            } catch (IllegalAccessException e) {
                throw new ExcelException("创建实例异常,权限不足");
            }
        }
        return list;
    }


    /**
     * 通过单元格的值给对象的属性进行赋值
     *
     * @param t 对象实体
     * @param fields 对象对应的属性数组
     * @param dataCell 单元格对象
     * @param title 单元格对象对应的title
     * @param validateList 数据有效性列表
     */
    private static <T> void evaluateField(T t, Field[] fields, Cell dataCell,
                                       String title, List<List<String>> validateList) {
        for (Field field : fields) {
            // 处理属性上有ExcelDescription注解的数据进行赋值
            if (field.isAnnotationPresent(ExcelDescription.class)) {
                ExcelDescription annotation = field.getAnnotation(ExcelDescription.class);
                // 获取注解的描述
                String desc = annotation.desc();
                // 获取注解的数据类型
                ExcelDescription.DataType type = annotation.type();
                // 如果title和描述desc一致,则将cell里面的值赋值给该属性
                if (title.equals(desc)) {
                    // 获取value的值
                    Object value = type.evaluateDataByType(dataCell, validateList);
                    field.setAccessible(true);
                    try {
                        field.set(t, value);
                    } catch (IllegalAccessException e) {
                        throw new ExcelException("对象属性赋值权限异常");
                    }
                }
            }
        }
    }
}

3.测试数据

测试实体

package poi.bean;

import lombok.Data;
import poi.annotation.ExcelDescription;

import java.math.BigDecimal;
import java.util.Date;

/**
 * @description:
 * @author: zengwenbo
 * @date: 2024/3/10 14:07
 */
@Data
public class Person {
    @ExcelDescription(desc = "名称")
    private String name;

    @ExcelDescription(desc = "年龄", type = ExcelDescription.DataType.BigDecimal)
    private BigDecimal age;

    @ExcelDescription(desc = "生日", type = ExcelDescription.DataType.Date)
    private Date birth;

    @ExcelDescription(desc = "国籍")
    private String country;
}

测试的excel文件数据截图
在这里插入图片描述
对国籍数据进行了有效性填充
在这里插入图片描述
测试代码:将excel文件放在resource目录下

package com.example.demo;

import com.example.demo.redis.User;

import org.junit.jupiter.api.Test;
import org.junit.runner.RunWith;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.core.io.ClassPathResource;
import org.springframework.core.io.Resource;
import org.springframework.core.io.ResourceLoader;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.mock.web.MockMultipartFile;
import org.springframework.test.context.junit4.SpringRunner;
import org.springframework.util.ResourceUtils;
import org.springframework.web.multipart.MultipartFile;
import poi.bean.Person;
import poi.utils.ExcelUtil;

import java.io.File;
import java.io.FileNotFoundException;
import java.io.InputStream;
import java.nio.file.Files;
import java.util.List;

@RunWith(SpringRunner.class)
@SpringBootTest
class DemoApplicationTests {

    @Autowired
    private ResourceLoader resourceLoader;
    
    @Test
    void TestExcel() throws Exception {
        Resource resource = resourceLoader.getResource("classpath:test.xls" );
        String fileName = resource.getFilename();
        byte[] fileBytes = Files.readAllBytes(resource.getFile().toPath());

        MultipartFile multipartFile = new MockMultipartFile(fileName, fileName, "text/plain", fileBytes);
        List<Person> list = ExcelUtil.transTableToEntity(multipartFile, Person.class, 0, 0);
    }

}

最终结果
在这里插入图片描述

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:/a/445385.html

如若内容造成侵权/违法违规/事实不符,请联系我们进行投诉反馈qq邮箱809451989@qq.com,一经查实,立即删除!

相关文章

一键部署Tesseract-OCR环境C++版本(Windows)

环境&#xff1a;Windows 10 工具&#xff1a;git vcpkg vscode cmake 库&#xff1a;Tesseract 一键部署Tesseract-OCR环境C版本&#xff08;Windows&#xff09; 分享这篇文章的原因很简单&#xff0c;就是为了让后续的朋友少走弯路。自己在搜索相关C版本的tesseract部署时…

【python量化】基于okex API开发的海龟策略

介绍 基于okex api开发的海龟策略&#xff0c;okex海龟策略python实现方式。该程序目前只支持单品种&#xff0c;比如设置ETH后&#xff0c;只对ETH进行做多做空。该程序运行需要两样东西&#xff1a;apikey 和 标的 运行该程序之前&#xff0c;用户需要到okex网站去申请apiK…

虚函数与纯虚函数有什么区别?

总的来说有两点区别&#xff1a; 1.虚函数的作用主要是矫正指针&#xff08;口语化的说法&#xff09; 2.虚函数不一定要重新定义&#xff0c;纯虚函数一定要定义&#xff08;口语化的说法&#xff09; 1&#xff09;. 虚函数的作用主要是矫正指针&#xff0c;使得基类的指针…

【Python数据结构与判断1/7】复杂的多向选择

目录 导入 举个栗子 代码优化 elif 栗子 执行顺序 情况一 情况二 情况三 if-elif-else特性 三种判断语句小结 if if-else if-elif-else 嵌套语句 if嵌套 栗子 执行顺序 相互嵌套 Tips Debug 总结 导入 在前面&#xff0c;我们学习了单向选择的if语句和多项…

Decontam去污染:一个尝试

为了程序运行的便利性&#xff0c;不想将Decontam放到windows的Rstudio里面运行&#xff0c;需要直接在Ubuntu中运行&#xff0c;并且为了在Decontam时进行其他操作&#xff0c;使用python去运行R 首先你需要有一个conda环境&#xff0c;安装了R&#xff0c;Decontam&#xff0…

day 49 动态规划 part 10● 121. 买卖股票的最佳时机 ● 122.买卖股票的最佳时机II

看了题解&#xff0c;第一种暴力&#xff0c;两个for循环。 class Solution { public:int maxProfit(vector<int>& prices) {int result 0;for (int i 0; i < prices.size(); i) {for (int j i 1; j < prices.size(); j){result max(result, prices[j] -…

使用scrapy爬取蜻蜓FM

创建框架和项目 ### 1. 创建虚拟环境 conda create -n spiderScrapy python3.9 ### 2. 安装scrapy pip install scrapy2.8.0 -i https://pypi.tuna.tsinghua.edu.cn/simple### 3. 生成一个框架并进入框架 scrapy startproject my_spider cd my_spider### 4. 生成项目 scrapy …

LeetCode:143.重排链表

143. 重排链表 解题过程 /*** Definition for singly-linked list.* public class ListNode {* int val;* ListNode next;* ListNode() {}* ListNode(int val) { this.val val; }* ListNode(int val, ListNode next) { this.val val; this.next next; …

数据结构——堆的应用 Topk问题

&#x1f49e;&#x1f49e; 前言 hello hello~ &#xff0c;这里是大耳朵土土垚~&#x1f496;&#x1f496; &#xff0c;欢迎大家点赞&#x1f973;&#x1f973;关注&#x1f4a5;&#x1f4a5;收藏&#x1f339;&#x1f339;&#x1f339; &#x1f4a5;个人主页&#x…

实验一:华为VRP系统的基本操作

1.1实验介绍 1.1.1关于本实验 本实验通过配置华为设备&#xff0c;了解并熟悉华为VRP系统的基本操作 1.1.2实验目的 理解命令行视图的含义以及进入离开命令行视图的方法 掌握一些常见的命令 掌握命令行在线帮助的方法 掌握如何撤销命令 掌握如何使用命令快捷键 1.1.3实验组网 …

挑战杯 基于设深度学习的人脸性别年龄识别系统

文章目录 0 前言1 课题描述2 实现效果3 算法实现原理3.1 数据集3.2 深度学习识别算法3.3 特征提取主干网络3.4 总体实现流程 4 具体实现4.1 预训练数据格式4.2 部分实现代码 5 最后 0 前言 &#x1f525; 优质竞赛项目系列&#xff0c;今天要分享的是 基于深度学习机器视觉的…

React组件(函数式组件,类式组件)

函数式组件 <!DOCTYPE html> <html lang"en"> <head><meta charset"UTF-8"><meta name"viewport" content"widthdevice-width, initial-scale1.0"><title>React Demo</title> <!-- 引…

阿里云服务器9元1个月优惠价格表

阿里云服务器9元1个月优惠价格表&#xff0c;用不上9元&#xff0c;又降价了&#xff0c;只要5元。阿里云服务器一个月多少钱&#xff1f;最便宜5元1个月。阿里云轻量应用服务器2核2G3M配置61元一年&#xff0c;折合5元一个月&#xff0c;2核4G服务器30元3个月&#xff0c;2核2…

深入探讨AI团队的角色分工

目录 前言1. 软件工程师&#xff1a;构建系统基石的关键执行者2. 机器学习工程师&#xff1a;数据与模型的塑造专家3. 机器学习研究员&#xff1a;引领算法创新的智囊4. 机器学习应用科学家&#xff1a;理论与实践的巧妙连接5. 数据分析师&#xff1a;洞察数据&#xff0c;智慧…

C语言初学10:typedef

一、作用 为用户定义的数据类型取一个新名字 二、对结构体使用typedef定义新的数据类型名字 #include <stdio.h> #include <string.h>typedef struct Books //使用 typedef 来定义一个新的数据类型名字 {char title[50];} book;int main( ) {//book是typedef定…

图片在div完全显示

效果图&#xff1a; html代码&#xff1a; <div class"container" style" display: flex;width: 550px;height: 180px;"><div class"box" style" color: red; background-color:blue; width: 50%;"></div><div …

基于冠豪猪优化算法(Crested Porcupine Optimizer,CPO)的无人机三维路径规划(MATLAB)

一、无人机路径规划模型介绍 无人机三维路径规划是指在三维空间中为无人机规划一条合理的飞行路径&#xff0c;使其能够安全、高效地完成任务。路径规划是无人机自主飞行的关键技术之一&#xff0c;它可以通过算法和模型来确定无人机的航迹&#xff0c;以避开障碍物、优化飞行…

企业如何安全参与开源项目?

【开源三句半】 企业参与开源潮&#xff0c; 安全创新都重要&#xff0c; 持续投入不可少&#xff0c; 眼光独到。 开源已经成为构建现代软件的常见方式&#xff0c;这不仅局限于IT技术本身&#xff0c;更推动了多个行业的数字化发展。企业决定引入开源项目打造商业软件时&…

NeRF模型NeRF模型

参考视频&#xff1a;https://www.youtube.com/watch?vHfJpQCBTqZs&ab_channelVision%26GraphicsSeminaratMIT NeRF模型的输入输出: 输入: (x, y, z): 一个三维空间坐标,代表场景中的一个位置点(θ, φ): 视线方向,θ表示与y轴的夹角,φ表示与x轴的夹角,用两个角度可以…

random标准模块

一、概述 在 Python 中&#xff0c;random 是一个内置模块&#xff0c;用于生成随机数。它提供了各种用于生成随机数的函数&#xff0c;包括伪随机数生成器、随机序列操作等。 1、需要导包 不会自动导入&#xff0c;需要显示的将random模块导入 import random2、源码分析&…