脚本之家 服务器常用软件
快捷导航
您的位置：首页 → 软件编程 → Rust语言 → Rust CSV文件读取器

Rust使用csv crate构建CSV文件读取器的全过程

 更新时间：2024年05月28日 09:50:59   作者：代号0408  
这篇文章主要学习如何基于Rust使用csv这个crate构建一个CSV文件读取器的过程,学习了csv相关的用法以及一些往期学过的crate的复习,兼顾了实用性和Rust的学习,需要的朋友可以参考下

效果演示

csv

快速上手

依赖导入：

cargo add csv

读取实现：

use std::error::Error;
use std::fs::File;
use std::path::Path;

fn read_csv<P: AsRef<Path>>(filename: P) -> Result<(), Box<dyn Error>> {
    let file = File::open(filename)?;
    let mut rdr = csv::Reader::from_reader(file);

    for result in rdr.records() {
        let record = result?;
        println!("{:?}", record);
    }

    Ok(())
}

fn main() -> Result<(), Box<dyn Error>> {
    let filename = "src/email.csv";
    read_csv(filename)
}

这是一段简单的Rust程序，演示了如何使用csv``crate中的读取API,通过指定csv路径进行csv数据的读取。

  • use std::error::Error;, use std::fs::File;, use std::path::Path;：
    • 这些是Rust语言中用于导入标准库中的错误处理、文件操作和路径相关模块的语句。
  • fn read_csv<P: AsRef<Path>>(filename: P) -> Result<(), Box<dyn Error>>：
    • 这是一个函数定义，名为read_csv，它接受一个实现了AsRef<Path> trait 的泛型参数P，表示文件名。函数返回一个Result枚举类型，其中Ok(())表示成功，Err包含一个实现了Error trait 的错误对象的Box指针。
    • 函数打开指定的CSV文件，创建一个CSV读取器（csv::Reader），然后遍历文件中的每一行记录并打印出来。
  • fn main() -> Result<(), Box<dyn Error>>：
    • 这是程序的入口点，也是主函数。它也返回一个Result枚举类型，用于处理可能出现的错误。
    • 在main函数中，指定了要读取的CSV文件的文件名为"src/email.csv"，然后调用read_csv函数来处理这个文件。
  • let file = File::open(filename)?;：
    • 在read_csv函数中，这行代码尝试打开指定的文件，?操作符用于处理可能出现的错误，如果出现错误，则会将错误传播到调用方。
  • let mut rdr = csv::Reader::from_reader(file);：
    • 创建一个CSV读取器rdr，并从打开的文件中读取数据。
  • for result in rdr.records() { ... }：
    • 使用for循环遍历CSV文件中的每一行记录。
  • let record = result?;：
    • 在循环中，尝试将每一行记录解析为csv::StringRecord类型的record，?操作符用于处理可能的解析错误。
  • println!("{:?}", record);：
    • 打印每一行记录的内容。
  • Ok(())：
    • 在函数末尾，返回一个Ok(())表示函数执行成功。

读取结果：

image-20240526192615414

csv文件的读取功能基本实现了，但是每次读取需要我们手动修改代码，指定要读取的csv文件路径，相对还是不够实用和灵活，特别是对于非程序猿来说。下面将对代码进行进一步提取和优化，将读取的功能封装为命令行程序，提升使用体验。

命令行程序封装

关于命令行，Rust的crate中有很多不错的库，在之前我的文章中也提及了部分，这里选择使用clap这个crate来实现。

[dependencies]
ansi_term = "0.12.1"
clap = { version = "4.5.4", features = ["derive"] }
csv = "1.3.0"
prettytable-rs = "0.10.0"

结构分离，为了利于维护，将读取CSV文件的方法独立在lib.rs中，命令行参数处理等内容依旧在main.rs

lib.rs

pub fn read_csv<P: AsRef<Path>>(filename: P) -> Result<(), Box<dyn Error>> {
    let file = File::open(filename)?;
    let mut rdr = csv::Reader::from_reader(file);

    let mut table = Table::new();

    // 添加表头
    let headers = rdr
        .headers()?
        .iter()
        .map(|h| Cell::new(h).style_spec("Fg=green"))
        .collect();
    table.add_row(Row::new(headers));

    // 添加记录
    for result in rdr.records() {
        let record = result?;
        let cells: Vec<Cell> = record.iter().map(|field| Cell::new(field)).collect();
        table.add_row(Row::new(cells));
    }

    table.printstd();
    Ok(())
}

感觉没啥新的东西可以讲的，这个方法的主要逻辑在上面已经说过，至于内容的打印，还是使用之前在X-SCAN端口扫描器中使用的Table进行美化。

main.rs

use x_csvreader::read_csv;
#[derive(Parser, Debug)]
struct Args {
    #[clap(short, long, help = "The path to the CSV file.")]
    path: String,
}
fn print_infos() {
    println!(
        "{}",
        Blue.paint(
            r#"
            __   __      _____  _______      __     _____                _
            \ \ / /     / ____|/ ____\ \    / /    |  __ \              | |
             \ V /_____| |    | (___  \ \  / /_____| |__) |___  __ _  __| | ___ _ __
              > <______| |     \___ \  \ \/ /______|  _  // _ \/ _` |/ _` |/ _ \ '__|
             / . \     | |____ ____) |  \  /       | | \ \  __/ (_| | (_| |  __/ |
            /_/ \_\     \_____|_____/    \/        |_|  \_\___|\__,_|\__,_|\___|_|
        author:代号0408
        version:0.1.0
        "#
        )
    );
}
fn main() {
    print_infos();
    let args = Args::parse();
    // 调用lib.rs中定义的read_csv函数
    match read_csv(&args.path) {
        Ok(_) => {
            println!("=============================");
            println!("CSV 文件读取成功！");
        }
        Err(e) => {
            eprintln!("读取 CSV 文件时出现错误：{}", e);
        }
    }
}

逻辑简单，就不赘述了。如果不了解字符打印美化和表格美化这两个lib基本使用的，建议翻下我往期的文章，都是有写的。

那么如何使用呢？

cargo run -- --path <csv文件路径>

不妨将开头的效果复现一下：

cargo run -- --path C:\RustProjects\x-csvreader\src\email.csv

当然，为了演示的效果，这里选择的CSV文件数据量并不大，处理大数据量的文件也是可以的，只不过打印出来的表格数据可能会出现终端 霸屏的情况，纸上得来终觉浅！建议你自己试试，这里就不截图了。

总结

这篇文章主要学习如何基于Rust使用csv这个crate构建一个CSV文件读取器的过程。学习了csv相关的用法以及一些往期学过的crate的复习，兼顾了实用性和Rust的学习，是个很不错的练手小项目。

以上就是Rust使用csv crate构建CSV文件读取器的全过程的详细内容，更多关于Rust CSV文件读取器的资料请关注脚本之家其它相关文章！

您可能感兴趣的文章:

相关文章

  • 探索Rust切片与Go有何区别

    探索Rust切片与Go有何区别

    这篇文章主要为大家介绍了Rust切片与Go的区别探索,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪
    2024-01-01
  • Rust 中 Mutex 的基本用法

    Rust 中 Mutex 的基本用法

    Rust 标准库中的 Mutex 结构体位于 std::sync::Mutex 中,它提供了线程安全的数据访问,Mutex 保证了在同一时间只有一个线程可以访问被锁定的数据,这篇文章主要介绍了Rust 中 Mutex 的基本用法,需要的朋友可以参考下
    2024-05-05
  • Rust常用特型之ToOwned特型示例详解

    Rust常用特型之ToOwned特型示例详解

    在Rust中,假定某类型实现了Clone特型,如果给你一个对它引用,那我们得到它指向内容的备份的最常见方式是调用其clone()函数,这篇文章主要介绍了Rust常用特型之ToOwned特型,需要的朋友可以参考下
    2024-04-04
  • 从迷你todo 命令行入门Rust示例详解

    从迷你todo 命令行入门Rust示例详解

    这篇文章主要为大家介绍了从一个迷你todo命令行入门Rust的示例详解，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪
    2023-01-01
  • Rust中字符串类型&str和String的使用

    Rust中字符串类型&str和String的使用

    在Rust中,字符串是一种非常重要的数据类型,&str和String是Rust中两种主要的字符串类型,本文主要介绍了Rust中字符串类型&str和String的使用,感兴趣的可以了解一下
    2024-03-03
  • Go调用Rust方法及外部函数接口前置

    Go调用Rust方法及外部函数接口前置

    这篇文章主要为大家介绍了Go调用Rust方法及外部函数接口前置示例实现，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪
    2022-06-06
  • 解析rust中的struct

    解析rust中的struct

    自定义的数据类型，为相关联的值命名，打包成有意义的组合，类似python的dict，但是赋值的时候可以不按顺序，本文给大家介绍下rust中的struct知识，感兴趣的朋友一起看看吧
    2022-10-10
  • rust智能指针的具体使用

    rust智能指针的具体使用

    智能指针是一些数据结构,它们的行为类似于指针但拥有额外的元数据和附加功能,本文就来介绍一下rust智能指针的具体使用,感兴趣的可以了解一下
    2023-12-12
  • Rust如何使用线程同时运行代码

    Rust如何使用线程同时运行代码

    Rust使用1:1线程模型,通过std::thread::spawn创建线程,返回JoinHandle用于等待线程完成,闭包默认借用外部变量,使用move关键字转移所有权,多线程共享数据时需使用并发原语,如Mutex、RwLock、Arc等,以避免竞态条件
    2025-02-02
  • Rust指南枚举类与模式匹配详解

    Rust指南枚举类与模式匹配详解

    这篇文章主要介绍了Rust指南枚举类与模式匹配精讲,枚举允许我们列举所有可能的值来定义一个类型，枚举中的值也叫变体，今天通过一个例子给大家详细讲解，需要的朋友可以参考下
    2022-09-09

最新评论