时间戳

首页 > 解决方案 > CSV from_writer 适用于 stdout()，但在 from_path 上失败

rust - CSV from_writer 适用于 stdout()，但在 from_path 上失败

问题描述

这里是 Rust 初学者。

我一直在尝试学习 CSV 板条箱，但陷入了以下情况。

我的目标是：

  1. 解析嵌套数组
  2. 将列名设置为数组值
  3. 写入 CSV

首先，这里是输出我想要的代码的代码。

use serde::Serialize;
use serde::Deserialize;
use csv;
use serde_json;
use std::io;

#[derive(Debug,Serialize,Deserialize)]
#[serde(transparent)]
struct Parent {
    arr_field: Vec<Row>
}

#[derive(Debug,Serialize,Deserialize)]
struct Row {
    a: u8,
    b: u8,
    c: u8,
}

fn main() {
    let resp = r#" [[1,2,3],[3,2,1],[4,5,6]] "#;
    let mut wtr = csv::WriterBuilder::new().from_writer(io::stdout());

    let v: Parent = serde_json::from_str(resp).unwrap();

    for row in v.arr_field{
        wtr.serialize(row);
    }
}

这段代码的输出是：

a,b,c
1,2,3
3,2,1
4,5,6

但是当我想将输出保存到本地文件而不是标准输出时，如下所示：

    let mut wtr = csv::WriterBuilder::new().from_path("./foo.csv");

我收到以下错误wtr.serialize

error[E0599]: no method named `serialize` found for enum `std::result::Result<Writer<File>, csv::Error>` in the current scope

谢谢您的帮助。

标签： rustserde

解决方案

错误消息告诉您您需要知道的所有信息 -from_path返回 aResult而不是 a WriterBuilder，因为打开该文件可能并不总是有效。这与from_writer- 不需要打开文件不同，因此不可能遇到错误。

要解决此问题，您可以使用.unwrap()，就像使用serde_json::from_str下面的行一样。当遇到错误时，这将导致恐慌，立即终止您的程序。

let mut wtr = csv::WriterBuilder::new().from_path("./foo.csv").unwrap();

请注意，它serialize也会返回结果，因此您还应该添加.unwrap()或其他一些逻辑来处理 for 循环中的错误。Rust 可能会显示一个警告，指出有一个未使用的结果。

推荐阅读