如何基于键名反序列化为枚举变量? [英] How to deserialize into a enum variant based on a key name?

问题描述

我有采用两种形式的JSON:

I have JSON which takes two forms:

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

我想将其反序列化为这样的结构:

I want to deserialize it into a struct like this:

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

其中

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

如果不为Example编写自定义解串器，我不知道该怎么做.

I don't know how to do it without writing a custom deserializer for Example.

这有效:

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

或者，在YAML中:

Example:
    field: 42
    an_enum:
        A: 76

an_enum是多余的并且令人讨厌.如何将第一种形式反序列化为struct?或者，或者，如何声明可以成功反序列化所需语法的结构?

The an_enum is superfluous and annoying to write. How can I deserialize the first form into the struct? Or, alternatively, how can I declare a struct that will successfully deserialize the syntax I want?

推荐答案

您在寻找 #[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) })

---

在此功能可用之前，我将使用自定义反序列化:

---

Before this was available, I'd use custom deserialization:

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);
}

这篇关于如何基于键名反序列化为枚举变量?的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！