Есть ли способ сказать Серде использовать поле структуры в качестве ключа карты? - PullRequest
Новая
       61

Есть ли способ сказать Серде использовать поле структуры в качестве ключа карты?

0 голосов
/ 02 ноября 2018

У меня есть карта элементов, которые я хотел бы сериализовать в список структур, каждый из которых имеет поле для соответствующего ключа.

Представьте себе файл YAML, подобный этому: 

name_a:
    some_field: 0
name_b:
    some_field: 0
name_c:
    some_field: 0

И соответствующая структура, подобная этой: 

struct Item {
    name: String,
    some_field: usize,
}

Я бы хотел десериализовать именованные элементы в Vec<Item> вместо Map<String, Item>. Имена элементов (name_a, ...) помещаются в поле name объектов Item.

Я попытался сделать следующее: 

extern crate serde_yaml;
use std::fs::read_to_string;

let contents = read_to_string("file.yml").unwrap();
let items: Vec<Item> = serde_yaml::from_str(&contents).unwrap();

Однако это не работает и выдает ошибку invalid type: map, expected a sequence.

Я бы предпочел избегать создания переходного процесса Map<String, PartialItem>, который преобразуется в Vec, и я бы также предпочел не реализовывать дополнительную структуру PartialItem. Использование Option<String> в качестве name было бы возможно, хотя я не думаю, что это оптимально.

Ответы [ 2 ]

0 голосов
/ 03 ноября 2018

Один из способов - десериализовать карту самостоятельно: 

use std::fmt;

use serde::de::{Deserialize, Deserializer, MapAccess, Visitor};
use serde_derive::Deserialize;

struct ItemMapVisitor {}

impl ItemMapVisitor {
    fn new() -> Self {
        Self {}
    }
}

#[derive(Debug, Deserialize)]
struct SomeField {
    some_field: u32,
}

#[derive(Debug)]
struct Item {
    name: String,
    some_field: u32,
}

#[derive(Debug)]
struct VecItem(Vec<Item>);

impl Item {
    fn new(name: String, some_field: u32) -> Self {
        Self { name, some_field }
    }
}

impl<'de> Visitor<'de> for ItemMapVisitor {
    type Value = VecItem;

    fn expecting(&self, formatter: &mut fmt::Formatter) -> fmt::Result {
        formatter.write_str("name: somefield:")
    }

    fn visit_map<M>(self, mut access: M) -> Result<Self::Value, M::Error>
    where
        M: MapAccess<'de>,
    {
        let mut items = Vec::with_capacity(access.size_hint().unwrap_or(0));
        while let Some((key, value)) = access.next_entry::<String, SomeField>()? {
            items.push(Item::new(key, value.some_field));
        }
        Ok(VecItem(items))
    }
}

impl<'de> Deserialize<'de> for VecItem {
    fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
    where
        D: Deserializer<'de>,
    {
        deserializer.deserialize_map(ItemMapVisitor::new())
    }
}

fn main() {
    let contents = r#"
name_a:
    some_field: 0
name_b:
    some_field: 1
name_c:
    some_field: 2
"#;

    let items: VecItem = serde_yaml::from_str(&contents).unwrap();
    println!("{:#?}", items);
}

Выход: 

VecItem(
    [
        Item {
            name: "name_a",
            some_field: 0
        },
        Item {
            name: "name_b",
            some_field: 1
        },
        Item {
            name: "name_c",
            some_field: 2
        }
    ]
)

Если вам не нужна структура Somefield. Вы также можете использовать это: 

#[derive(Debug, Deserialize)]
struct Item {
    #[serde(skip)]
    name: String,
    some_field: u32,
}

while let Some((key, value)) = access.next_entry::<String, Item>()? {
    items.push(Item::new(key, value.some_field));
}

Но это может добавить бесполезную копию.

0 голосов
/ 02 ноября 2018

Определить значение по умолчанию для Item::name поле 

#[derive(Debug, Serialize, Deserialize)]
struct Item {
    #[serde(default)]
    name: String,
    some_field: usize,
}

С помощью этого трюка Item можно использовать как для десериализации, так и для преобразования в Vec из Item s: 

let contents = read_to_string("file.yml").unwrap();

let items: HashMap<String, Item> = serde_yaml::from_str(&contents).unwrap();

let slist: Vec<Item> = items
    .into_iter()
    .map(|(k, v)| Item {
        name: k,
        some_field: v.some_field,
    })
    .collect();
...