germanubis/rust-csv: A CSV parser for Rust, with Serde support.

examples/tutorial-perf-alloc-03.rs

26 строк568 B

Andrew Gallant

*: fix all warnings and update code

14 фев 2023, 05:31

14 фев 2023, 05:31ea0273c

use std::{error::Error, io, process};

fn run() -> Result<u64, Box<dyn Error>> {
    let mut rdr = csv::Reader::from_reader(io::stdin());
    let mut record = csv::ByteRecord::new();

    let mut count = 0;
    while rdr.read_byte_record(&mut record)? {
        if &record[0] == b"us" && &record[3] == b"MA" {
            count += 1;
        }
    }
    Ok(count)
}

fn main() {
    match run() {
        Ok(count) => {
            println!("{}", count);
        }
        Err(err) => {
            println!("{}", err);
            process::exit(1);
        }
    }
}