Question

我正在使用Serde将BSON对象反序列化为Rust结构实例。我可以将对象反序列化为具体的结构实例，但是如何通用地反序列化呢？

我在MongoDB中有“国家”和“城市”集合。在Rust程序中，我有一个Country和City的结构。当我从Mongo拉出一个国家或城市时，可以使用Serde将其反序列化为Country或City结构。请参阅下面main()的第二行。

我想将BSON对象反序列化为通用Location对象。根据我在Rust本书中所读到的有关泛型的知识，我创建了特征LocationTrait并将其实现为Country和City。（请参见main()中的第3行）。它无法编译，说编译时无法知道类型dyn LocationTrait的值的大小。

#[derive(Serialize, Deserialize)]
pub struct Country {
    pub name: String,
}

#[derive(Serialize, Deserialize)]
pub struct City {
    pub name: String,
}

pub trait LocationTrait {}
impl LocationTrait for Country {}
impl LocationTrait for City {}

fn main() {
    let item = mongo_coll
        .find_one(Some(doc! {"name": "usa"}), None)
        .unwrap()
        .unwrap();
    let country: Country = bson::from_bson(bson::Bson::Document(item)).unwrap();
    // fails -> let gen_location: LocationTrait = bson::from_bson(bson::Bson::Document(item)).unwrap();
}

最终，我想创建一个表示Country或City的通用对象。但是，我不确定起点是什么？我需要专注于特征还是需要创建新的特征绑定结构？

Answer 1

有两个问题阻止您的代码编译。

您看到的第一个错误：the size for values of type dyn LocationTrait cannot be known at compilation time，是由于bson::from_bson需要按值返回反序列化的结果。编译器需要知道需要在调用堆栈中分配多少空间才能返回它。

但是特征是描述行为而不是数据的抽象，因此可以用u8（单个字节）或更大的结构来实现。

为了能够返回这样的值，您需要将其装箱（请参见Trait Objects）。

第二个问题是返回值必须实现Deserialize特征（而不是LocationTrait）

要解决这些问题：

最简单的方法是使用枚举而不是特征：

#[derive(Serialize, Deserialize)]
#[serde(tag = "type")]
pub enum Location {
    Country(Country),
    City(City)
}

这将适用于{"type" = "Country", name="usa"}之类的文档。检查the Serde doc了解更多选项。

如果您真的要使用特征（例如，能够在此模块外部定义类型），则需要装箱特征和自定义结构，例如：

// The same trait as defined earlier
pub trait LocationTrait {}
impl LocationTrait for Country {}
impl LocationTrait for City {}

// A custom struct on which you can implement the deserialize trait
// Needed as both Deserialize and Box are defined outside this crate.
struct DynLocation(Box<dyn LocationTrait>);

impl<'de> Deserialize<'de> for DynLocation {
    fn deserialize<D>(_deserializer: D) -> Result<Self, D::Error>
    where
        D: Deserializer<'de>,
    {
        // Tricky part ommited here:
        // You will need to partially deserialize you object
        // in order to get a first discriminant before instanciating
        // and deserializing the proper type.
        unimplemented!()
    }
}

// The public method to hide the DynLocation wrapper
pub fn deserialize(item: &str) -> Box<dyn LocationTrait> {
    let location: DynLocation = serde_json::from_str(item).expect("invalid json");
    location.0
}

可以在How can deserialization of polymorphic trait objects be added in Rust if at all?中找到有关同一主题的一些讨论。

如何将BSON反序列化为通用对象？

1 个答案: