将TOML反序列化为枚举向量,并包含值

10
我正在尝试读取一个TOML文件,以创建一个包含关联值枚举向量的结构体。以下是示例代码:
extern crate serde;
#[macro_use]
extern crate serde_derive;
extern crate toml;

use std::fs::File;
use std::io::Read;

#[derive(Debug, Deserialize, PartialEq)]
struct Actor {
    name: String,
    actions: Vec<Actions>,
}

#[derive(Debug, Deserialize, PartialEq)]
enum Actions {
    Wait(usize),
    Move { x: usize, y: usize },
}

fn main() {
    let input_file = "./sample_actor.toml";
    let mut file = File::open(input_file).unwrap();
    let mut file_content = String::new();
    let _bytes_read = file.read_to_string(&mut file_content).unwrap();
    let actor: Actor = toml::from_str(&file_content).unwrap();
    println!("Read actor {:?}", actor);
}

Cargo.toml

[dependencies]
serde_derive = "1.0.10"
serde = "1.0.10"
toml = "0.4.2"

sample_actor.toml

name = "actor1"
actions = [move 3 4, wait 20, move 5 6]

我知道该文件看起来有些“不对劲”,但是我不知道应该如何编写TOML文件中的操作,使得crate能够将它们识别为具有X个值的枚举。

当我使用cargo run运行此示例时,我会收到以下错误:

thread 'main' panicked at 'called `Result::unwrap()` on an `Err` value: Error { inner: ErrorInner { kind: NumberInvalid, line: Some(1), col: 11, message: "", key: [] } }', /checkout/src/libcore/result.rs:906:4
note: Run with `RUST_BACKTRACE=1` for a backtrace.

我知道我可能需要为我的枚举实现FromStr方法,将字符串转换为枚举类型。我大致了解可以实现自定义反序列化器以特定方式反序列化,但我不确定这些部分如何配合使用。
使用serde_json而不是TOML的等效示例似乎可以直接运行(当然要使用JSON文件代替TOML)。
代码的JSON版本:
extern crate serde;
extern crate serde_json;

#[macro_use]
extern crate serde_derive;

use serde_json::Error;
use std::fs::File;
use std::io::Read;

#[derive(Debug, Serialize, Deserialize)]
enum Foo {
    bar(u32),
    baz { x: u32, y: u32 },
}

#[derive(Debug, Serialize, Deserialize)]
struct Address {
    street: String,
    city: String,
    nums: Vec<Foo>,
}

fn main() {
    /*
        let address = Address {
            street: "10 Downing Street".to_owned(),
            city: "London".to_owned(),
            nums : vec![Foo::bar(1), Foo::baz{x : 2, y : 3}],
        };

        // Serialize it to a JSON string.
        let j = serde_json::to_string(&address).unwrap();

        // Print, write to a file, or send to an HTTP server.
        println!("{}", j);
    */
    let input_file = "./sample_address.json";
    let mut file = File::open(input_file).unwrap();
    let mut file_content = String::new();
    let _bytes_read = file.read_to_string(&mut file_content).unwrap();
    let address: Address = serde_json::from_str(&file_content).unwrap();
    println!("{:?}", address);
}

这个例子中读取/写入的JSON数据如下:
Address { street: "10 Downing Street", city: "London", nums: [bar(1), baz { x: 2, y: 3 }] }

也许 TOML 库无法支持我的用例?

4
找出你的 toml 长什么样子最快捷的方法是派生 Serialize 并为你的结构体生成 toml。 - oli_obk
@oli_obk-ker,这是一个非常好的想法,谢谢! - Dash83
1
现在你提到了它,它似乎如此明显,我感觉自己很傻没有一开始尝试这个方法。哈哈。 - Dash83
好像 toml crate 也有和我一样的问题:如何序列化枚举?当我创建一个 actor 并在它上面使用 toml::to_string() 时,它会产生输出 Err(UnsupportedType)。如果我更改 actor 的定义,使 actions 成为 Vec<String> 而不是 Vec<Actions>,则会产生有效的 TOML。 - Dash83
1个回答

8

Serde拥有许多用于枚举序列化的选项。其中一个适用于您的情况:

use serde::{Deserialize, Serialize}; // 1.0.117
use toml; // 0.5.7

#[derive(Debug, Serialize, Deserialize, PartialEq)]
#[serde(tag = "type", content = "args")]
enum Actions {
    Wait(usize),
    Move { x: usize, y: usize },
}

fn main() {
    let a_wait = Actions::Wait(5);
    println!("{}", toml::to_string(&a_wait).unwrap());

    let a_move = Actions::Move { x: 1, y: 1 };
    println!("{}", toml::to_string(&a_move).unwrap());
}

type = "Wait"
args = 5

type = "Move"

[args]
x = 1
y = 1

谢谢,Shep。那个serde枚举文档正是我在寻找的。干杯! - Dash83

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接