业务记录:处理动态表头的CSV/EXCEL文件

业务描述:传入一个动态表头的CSV文件,解析CSV,并保存入库。

CSV文件的表头是不确定的,即顺序和字段个数不确定,以及表头是中文字段。

例如:

为了顺利解析CSV文件,前端需要传入对应的字段名和顺序名

{
  "age": "1",
  "sex": "2",
  "name": "3"
}

思路和步骤

  1. 接收并解析前端传递的参数
  2. 读取并解析CSV文件
  3. 建立表头与字段名的映射
  4. 动态赋值实体类字段
  5. 保存数据到数据库

用main函数来演示第三步和第四步的代码思路:

  public static void main(String[] args) throws IOException, NoSuchFieldException, IllegalAccessException {
        // 创建一个列表来存储解析后的用户对象
        List<User> users = new ArrayList<>();
        String csvFile = "C:\\Users\\Administrator\\Desktop\\CSV测试.csv";
        FileReader fileReader = new FileReader(csvFile);
        // headerMapping用于模拟前端传入的参数  例如:{"age": "1", "sex": "2", "name": "3"}
        Map<String,String> headerMapping = new HashMap<String,String>();
        headerMapping.put("age","1");
        headerMapping.put("sex","2");
        headerMapping.put("name","3");
        CSVParser csvParser = new CSVParser(fileReader, CSVFormat.DEFAULT.withFirstRecordAsHeader());
        // 遍历解析后的CSV记录,每次迭代处理一行CSV数据
            for (CSVRecord record : csvParser) {
                // 创建一个新的 User 对象
                User user = new User();
                // 遍历前端传递的字段映射,每次迭代处理一个字段映射
                for (Map.Entry<String, String> entry : headerMapping.entrySet()) {
                    // 获取实体类的字段名 如:sex
                    String fieldName = entry.getKey();
                    // 获取CSV列索引,并将1基索引转换为0基索引 如:性别(sex) 是顺序2 对应CSV下标为1 
                    int columnIndex = Integer.parseInt(entry.getValue()) - 1;
                    // 从当前记录中获取对应列的值 
                    String value = record.get(columnIndex);
                    // 获取实体类中对应的字段对象 
                    Field field = User.class.getDeclaredField(fieldName);
                    // 设置字段可访问,以便后续进行赋值操作
                    field.setAccessible(true);
                    // 将字符串值转换为字段的类型,并将其赋值给实体类对象的字段
                    field.set(user, convertValue(field, value));
                }
                // 将填充后的用户对象添加到结果列表中
                users.add(user);
            }
    }

private Object convertValue(Field field, String value) {
    // 获取字段的类型
    Class<?> type = field.getType();
    
    // 根据字段的类型将字符串值转换为相应的类型
    if (type == Integer.class) {
        return Integer.parseInt(value);
    } else if (type == Long.class) {
        return Long.parseLong(value);
    } else if (type == Double.class) {
        return Double.parseDouble(value);
    } else if (type == Boolean.class) {
        return Boolean.parseBoolean(value);
    }
    
    // 如果字段类型是字符串或其他类型,直接返回原值
    return value;
}

  • 3
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值