阅读量:0
业务描述:传入一个动态表头的CSV文件,解析CSV,并保存入库。
CSV文件的表头是不确定的,即顺序和字段个数不确定,以及表头是中文字段。
例如:
为了顺利解析CSV文件,前端需要传入对应的字段名和顺序名
{ "age": "1", "sex": "2", "name": "3" }
思路和步骤
- 接收并解析前端传递的参数
- 读取并解析CSV文件
- 建立表头与字段名的映射
- 动态赋值实体类字段
- 保存数据到数据库
用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; }