如何根據鍵名反序列化為枚舉變體？

Question

我有兩種形式的 JSON：

"Example:" { "field": 42, "A": 76 }
"Example:" { "field": 42, "B": 110 }

我想將它反序列化為這樣的結構：

struct Example {
    field: i32,
    an_enum: AnEnum,
}

在哪里

enum AnEnum {
    A(i32),
    B(i32),
}

如果不為Example編寫自定義反序列化器，我不知道該怎么做。

這有效：

"Example:" { "field": 42, "an_enum": {"A": 76} }

或者，在 YAML 中：

Example:
    field: 42
    an_enum:
        A: 76

an_enum是多余的而且寫起來很煩人。 如何將第一種形式反序列化為結構？ 或者，或者，我如何聲明一個能夠成功反序列化我想要的語法的結構？

Answer 1

您正在尋找#[serde(flatten)] ：

use serde::Deserialize; // 1.0.104
use serde_json; // 1.0.48

#[derive(Debug, Deserialize)]
struct Example {
    field: i32,
    #[serde(flatten)]
    an_enum: AnEnum,
}

#[derive(Debug, Deserialize)]
enum AnEnum {
    A(i32),
    B(i32),
}

fn main() {
    let a = r#"{ "field": 42, "A": 76 }"#;
    let b = r#"{ "field": 42, "B": 110 }"#;

    let a = serde_json::from_str::<Example>(a);
    let b = serde_json::from_str::<Example>(b);

    println!("{:?}", a);
    println!("{:?}", b);
}

Ok(Example { field: 42, an_enum: A(76) })
Ok(Example { field: 42, an_enum: B(110) })

---

在這可用之前，我會使用自定義反序列化：

extern crate serde;
extern crate serde_json;
#[macro_use]
extern crate serde_derive;

use serde::{Deserialize, Deserializer};
use serde::de::Error;

#[derive(Debug)]
struct Example {
    field: i32,
    an_enum: AnEnum,
}

#[derive(Debug)]
enum AnEnum {
    A(i32),
    B(i32),
}

impl<'de> Deserialize<'de> for Example {
    fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
    where
        D: Deserializer<'de>,
    {
        #[derive(Debug, Deserialize)]
        struct Mapping {
            field: i32,
            #[serde(rename = "A")]
            a: Option<i32>,
            #[serde(rename = "B")]
            b: Option<i32>,
        }

        let Mapping { field, a, b } = Mapping::deserialize(deserializer)?;

        match (a, b) {
            (Some(_), Some(_)) => 
                Err(D::Error::custom("multiple variants specified")),
            (Some(a), None) =>
                Ok(Example { field, an_enum: AnEnum::A(a) }),
            (None, Some(b)) => 
                Ok(Example { field, an_enum: AnEnum::B(b) }),
            (None, None) =>
                Err(D::Error::custom("no variants specified")),
        }
    }
}

fn main() {
    let a = r#"{ "field": 42, "A": 76 }"#;
    let b = r#"{ "field": 42, "B": 110 }"#;

    let a: Result<Example, _> = serde_json::from_str(a);
    let b: Result<Example, _> = serde_json::from_str(b);

    println!("{:?}", a);
    println!("{:?}", b);
}