当前位置: 首页 > news >正文

springboot读取和写入csv文件数据

前言

csv格式的表格,和xls以及xlsx格式的表格有一些不同,不能够直接用处理xls的方式处理csv;
以下我将介绍如何读取并写入csv数据

准备工作

要处理csv格式的表格数据,我们首先需要引入pom.xml的依赖

    <dependency><artifactId>commons-csv</artifactId><groupId>org.apache.commons</groupId><version>1.8</version></dependency><dependency>

读取

前端

一般情况下,我们都是使用前后端交互获取到csv文件,那么前端应该是写一个file类型的input接收文件,为演示方便使用原生的组件

// 上传文件的组件
<input type="file" id="fileInput" name="file">
// 点击上传文件
<button type="button" onclick="uploadFile()">上传</button>

其中js逻辑如下:

// 上传文件方法function uploadFile() {var fileInput = document.getElementById("fileInput");var file = fileInput.files[0];var formData = new FormData();formData.append("file", file);
// 此处调用后端接口上传文件}

ps:需要注意的是,为了正确上传成功文件,我们的请求应该为:multipart/form-data
如果你使用ajax上传,示例如下:

export function uploadFileInterface(obj) { return service.post('/xxx', obj, {headers: { 'Content-Type': 'multipart/form-data' }
})

其中service为我封装的ajax的js
代码如下:

import axios from 'axios'// 创建axios实例
// eslint-disable-next-line no-unused-vars
const service = axios.create({baseURL: '/', // api的base_Urltimeout: 50000 // 请求超时时间
});// 请求拦截器
axios.interceptors.request.use(function (config) {// 在发送请求之前做些什么return config},function (error) {// 对请求错误做些什么return Promise.reject(error)}
)// 响应拦截器
axios.interceptors.response.use(function (config) {// 对响应数据做点什么return config},function (error) {// 对响应错误做点什么return Promise.reject(error)}
)export default service

后端

我的csv文件如下图所示:
在这里插入图片描述

读取到的数据为:
在这里插入图片描述

后端接收文件的逻辑如下:

  @POST@Path("/getFile")public String submitOriginalFile(FormDataMultiPart form) {// 拿取前端标识的文件// 获取文件流InputStream fileIns = form.getField("file").getValueAs(InputStream.class);// 获取文件名,中文不乱码写法String name = new String(form.getField("file").getContentDisposition().getFileName().getBytes("iso-8859-1"), "UTF-8");// 将文件流转byte,可作为其他情况存储起来处理byte[] fileBytes = new InputStreamToByteArray(fileIns);// 将byte数组转csv数据List<CSVRecord> csvData = new FormDataMultiPartJob().byteToCsvFileData(fileBytes);// 遍历每一行的数据for (int i = 0; i < csvData.size(); i++) {System.out.println("每一行数据:" + csvData.get(i));for(int j = 0; j < csvData.get(i).size(); j++) {System.out.println("每一行下每一个单元格数据:" + csvData.get(i).get(j));}}return "读取成功!";}

文件流转byte数组

  /*** 文件流转byte*/public byte[] InputStreamToByteArray(InputStream inputStream) {try {BufferedInputStream bufferedInputStream = new BufferedInputStream(inputStream);ByteArrayOutputStream byteArrayOutputStream = new ByteArrayOutputStream();byte[] buffer;int len;byte[] buf = new byte[2048];while ((len = bufferedInputStream.read(buf)) != -1) {byteArrayOutputStream.write(buf, 0, len);}byteArrayOutputStream.flush();buffer = byteArrayOutputStream.toByteArray();inputStream.close();bufferedInputStream.close();byteArrayOutputStream.close();return buffer;} catch (Exception e) {return null;}}

byte数组转csv数据

/*** byte转csv数据*/public List<CSVRecord> byteToCsvFileData(byte[] bytes) {List<CSVRecord> records = new ArrayList<>();try {// 将 byte 数组转为 InputStreamReader 对象InputStreamReader in = new InputStreamReader(new ByteArrayInputStream(bytes), "UTF-8");// 创建csvParser对象CSVFormat csvFormat = CSVFormat.EXCEL;CSVParser parser = new CSVParser(in, csvFormat);// 读取CSV数据行records = parser.getRecords();parser.close();in.close();} catch (Exception e) {return null;}return records;}

写入

如果需要新建csv文件并往里面写数据,参考如下代码:

      try (CSVPrinter printer = new CSVPrinter(new FileWriter("data.csv"), CSVFormat.DEFAULT)) {List<String> header = Arrays.asList("Name", "Age", "Email");printer.printRecord(header);List<List<String>> data = Arrays.asList(Arrays.asList("John", "25", "john@example.com"),Arrays.asList("Alice", "30", "alice@example.com"),Arrays.asList("Bob", "40", "bob@example.com"));for (List<String> rowData : data) {printer.printRecord(rowData);}} catch (IOException e) {e.printStackTrace();}

上面的代码中,我们创建了一个名为 data.csv 的新 CSV 文件,并打开了一个 CSVPrinter 对象。printRecord方法用于打印数据行。

首先,我们打印了一个标题行。标题行包含列的名称。我们将标题行表示为 List 对象 header,并将其传递给 printRecord 方法以打印到文件中

其次,我们打印了一些数据行。数据行包含实际的数据。我们将所有数据行表示为、嵌套的 List 对象 data,并使用 for 循环将每行数据打印到文件中。

需要注意的是,这里使用到了 try-with-resources 语句块来确保 CSVPrinter 能够正确关闭。如果在使用 CSVPrinter 时发生任何异常,它都会在 try-with-resources 语句块结束时被捕获并关闭。


如果你需要使用不同的分隔符或者引号字符来定制 CSV 文件格式,你可以在创建 CSVPrinter 时指定对应的 CSVFormat 对象。例如,如果你要使用制表符作为分隔符,你可以使用以下代码创建 CSVPrinter:

CSVFormat format = CSVFormat.TDF.withHeader("Name", "Age", "Email");
try (CSVPrinter printer = new CSVPrinter(new FileWriter("data.tsv"), format)) {// 逻辑同上
} catch (IOException e) {e.printStackTrace();
}

以上代码会写成一个csv文件保存在你的工程文件里,假如你不想生成文件,而只想拿到byte数组数据做其他操作,参考如下:

try (ByteArrayOutputStream stream = new ByteArrayOutputStream();CSVPrinter printer = new CSVPrinter(new OutputStreamWriter(stream), CSVFormat.DEFAULT)) {List<String> header = Arrays.asList("Name", "Age", "Email");printer.printRecord(header);List<List<String>> data = Arrays.asList(Arrays.asList("John", "25", "john@example.com"),Arrays.asList("Alice", "30", "alice@example.com"),Arrays.asList("Bob", "40", "bob@example.com"));for (List<String> rowData : data) {printer.printRecord(rowData);}// byte数组byte[] bytes = stream.toByteArray();} catch (IOException e) {e.printStackTrace();
}

结语

以上就是如何获取并写入csv文件的过程

http://www.lryc.cn/news/69364.html

相关文章:

  • 【产品经理】工作交接
  • Springer期刊 latex投稿经验分享
  • Python 文件读取的练习
  • Redis:主从复制_通过此功能实现对内存上的数据更好的保护
  • LoRA:大模型的低秩自适应微调模型
  • 拼多多买家如何导出“个人中心”订单信息
  • 11.计算机基础-计算机网络面试题—基础知识
  • cs109-energy+哈佛大学能源探索项目 Part-1(项目背景)
  • ARM Linux摄像头传感器数据处理全景视野:从板端编码视频到高级应用
  • Fixed Function Shader
  • HTML- 标签学习之- 列表、表格
  • Canal搭建 idea设置及采集数据到kafka
  • CentOS7搭建伪分布式Hadoop(全过程2023)
  • Linux中文件描述符fd和文件指针filp的理解
  • CSS color中常用英文色值
  • Springboot idea 中 maven配置问题,找不到依赖:Could not find artifact xxxx
  • 编译原理笔记(一)引论
  • C++ 类和对象下 [补充]
  • [CTF/网络安全] 攻防世界 PHP2 解题详析
  • 图神经网络:(节点分类)在Cora数据集上动手实现图神经网络
  • RabbitMQ应用问题——消息补偿机制以及代码示例
  • 量化特征贡献度函数:feature_importances_函数/LGBMClassifier/XGBClassifier
  • 总结JVM重要知识点
  • 奇技淫巧第8期
  • 这个 归并排序详解过程 我能吹一辈子!!!
  • docker版jxTMS使用指南:自动生成代码
  • 聚观早报 | 小冰启动GPT克隆人计划;ofo创始人在美创业改做咖啡
  • 面试造航母,入职拧螺丝,工资离了个大谱...
  • Python+selenium自动化元素定位防踩坑
  • 【计算机组成原理】实验一