-
Notifications
You must be signed in to change notification settings - Fork 258
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
feat(Rust): Support polymorphism (#1795)
## What does this PR do?
1. Support polymorphism
Add a class resolver which is used for managing the relationship between
type_id and serializer. The serializer is generated by a generic
function which will be called by dynamic dispatch. When a struct uses
`Box<dyn Any>` as a field type and does not specify the specific type,
we will use the class resolver to load the handler by type_id. In this
situation, there is some performance overhead due to hash lookup, but it
greatly improves convenience.
Use as follow:
```Rust
#[test]
fn any() {
#[derive(Fury, Debug)]
struct Animal {
f3: String,
}
#[derive(Fury, Debug)]
struct Person {
f1: Box<dyn Any>,
}
let person = Person {
f1: Box::new(Animal {
f3: String::from("hello"),
}),
};
let mut fury = Fury::default();
fury.register::<Animal>(999);
fury.register::<Person>(1000);
let bin = fury.serialize(&person);
let obj: Person = fury.deserialize(&bin).expect("");
assert_eq!(true, obj.f1.is::<Animal>())
}
```
2. Add a register function for user to register Struct and id
3. Remove tag and hash which were generate by macro before and were
removed in our protocol now.
## TODO
1. Internal types like String、Set and Map should be registered by fury
by default and lookup by pattern match to avoid hash overhead.
2. More unit testcases.
3. Support `Box<dyn customTrait> `- Loading branch information
1 parent
497fe0a
commit e99b46f
Showing
26 changed files
with
437 additions
and
171 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,116 @@ | ||
| // Licensed to the Apache Software Foundation (ASF) under one | ||
| // or more contributor license agreements. See the NOTICE file | ||
| // distributed with this work for additional information | ||
| // regarding copyright ownership. The ASF licenses this file | ||
| // to you under the Apache License, Version 2.0 (the | ||
| // "License"); you may not use this file except in compliance | ||
| // with the License. You may obtain a copy of the License at | ||
| // | ||
| // http://www.apache.org/licenses/LICENSE-2.0 | ||
| // | ||
| // Unless required by applicable law or agreed to in writing, | ||
| // software distributed under the License is distributed on an | ||
| // "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY | ||
| // KIND, either express or implied. See the License for the | ||
| // specific language governing permissions and limitations | ||
| // under the License. | ||
|
|
||
| use super::context::{ReadContext, WriteContext}; | ||
| use crate::error::Error; | ||
| use crate::fury::Fury; | ||
| use crate::serializer::StructSerializer; | ||
| use std::any::TypeId; | ||
| use std::{any::Any, collections::HashMap}; | ||
|
|
||
| pub struct Harness { | ||
| serializer: fn(&dyn Any, &mut WriteContext), | ||
| deserializer: fn(&mut ReadContext) -> Result<Box<dyn Any>, Error>, | ||
| } | ||
|
|
||
| impl Harness { | ||
| pub fn new( | ||
| serializer: fn(&dyn Any, &mut WriteContext), | ||
| deserializer: fn(&mut ReadContext) -> Result<Box<dyn Any>, Error>, | ||
| ) -> Harness { | ||
| Harness { | ||
| serializer, | ||
| deserializer, | ||
| } | ||
| } | ||
|
|
||
| pub fn get_serializer(&self) -> fn(&dyn Any, &mut WriteContext) { | ||
| self.serializer | ||
| } | ||
|
|
||
| pub fn get_deserializer(&self) -> fn(&mut ReadContext) -> Result<Box<dyn Any>, Error> { | ||
| self.deserializer | ||
| } | ||
| } | ||
|
|
||
| pub struct ClassInfo { | ||
| type_def: Vec<u8>, | ||
| type_id: u32, | ||
| } | ||
|
|
||
| impl ClassInfo { | ||
| pub fn new<T: StructSerializer>(fury: &Fury, type_id: u32) -> ClassInfo { | ||
| ClassInfo { | ||
| type_def: T::type_def(fury), | ||
| type_id, | ||
| } | ||
| } | ||
|
|
||
| pub fn get_type_id(&self) -> u32 { | ||
| self.type_id | ||
| } | ||
|
|
||
| pub fn get_type_def(&self) -> &Vec<u8> { | ||
| &self.type_def | ||
| } | ||
| } | ||
|
|
||
| #[derive(Default)] | ||
| pub struct ClassResolver { | ||
| serialize_map: HashMap<u32, Harness>, | ||
| type_id_map: HashMap<TypeId, u32>, | ||
| class_info_map: HashMap<TypeId, ClassInfo>, | ||
| } | ||
|
|
||
| impl ClassResolver { | ||
| pub fn get_class_info(&self, type_id: TypeId) -> &ClassInfo { | ||
| self.class_info_map.get(&type_id).unwrap() | ||
| } | ||
|
|
||
| pub fn register<T: StructSerializer>(&mut self, class_info: ClassInfo, id: u32) { | ||
| fn serializer<T2: 'static + StructSerializer>(this: &dyn Any, context: &mut WriteContext) { | ||
| let this = this.downcast_ref::<T2>(); | ||
| match this { | ||
| Some(v) => { | ||
| T2::serialize(v, context); | ||
| } | ||
| None => todo!(), | ||
| } | ||
| } | ||
|
|
||
| fn deserializer<T2: 'static + StructSerializer>( | ||
| context: &mut ReadContext, | ||
| ) -> Result<Box<dyn Any>, Error> { | ||
| match T2::deserialize(context) { | ||
| Ok(v) => Ok(Box::new(v)), | ||
| Err(e) => Err(e), | ||
| } | ||
| } | ||
| self.type_id_map.insert(TypeId::of::<T>(), id); | ||
| self.serialize_map | ||
| .insert(id, Harness::new(serializer::<T>, deserializer::<T>)); | ||
| self.class_info_map.insert(TypeId::of::<T>(), class_info); | ||
| } | ||
|
|
||
| pub fn get_harness_by_type(&self, type_id: TypeId) -> Option<&Harness> { | ||
| self.get_harness(*self.type_id_map.get(&type_id).unwrap()) | ||
| } | ||
|
|
||
| pub fn get_harness(&self, id: u32) -> Option<&Harness> { | ||
| self.serialize_map.get(&id) | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.