Vincent Herlemont
57 lines
1.5 KiB
Rust
57 lines
1.5 KiB
Rust
use bincode::{config, Decode, Encode};
|
|
use native_model::Model;
|
|
use native_model_macro::native_model;
|
|
|
|
pub struct Bincode;
|
|
|
|
impl<T: bincode::Encode> native_model::Encode<T> for Bincode {
|
|
type Error = bincode::error::EncodeError;
|
|
fn encode(obj: &T) -> Result<Vec<u8>, bincode::error::EncodeError> {
|
|
bincode::encode_to_vec(obj, config::standard())
|
|
}
|
|
}
|
|
|
|
impl<T: bincode::Decode> native_model::Decode<T> for Bincode {
|
|
type Error = bincode::error::DecodeError;
|
|
fn decode(data: Vec<u8>) -> Result<T, bincode::error::DecodeError> {
|
|
bincode::decode_from_slice(&data, config::standard()).map(|(result, _)| result)
|
|
}
|
|
}
|
|
|
|
#[derive(Debug, Encode, Decode, PartialEq)]
|
|
#[native_model(id = 1, version = 1, with = Bincode)]
|
|
struct Foo1 {
|
|
x: i32,
|
|
}
|
|
|
|
#[derive(Debug, Encode, Decode, PartialEq)]
|
|
#[native_model(id = 1, version = 2, with = Bincode, from = Foo1)]
|
|
struct Foo2 {
|
|
x: i32,
|
|
}
|
|
|
|
impl From<Foo1> for Foo2 {
|
|
fn from(foo1: Foo1) -> Self {
|
|
Foo2 { x: foo1.x }
|
|
}
|
|
}
|
|
|
|
impl From<Foo2> for Foo1 {
|
|
fn from(foo2: Foo2) -> Self {
|
|
Foo1 { x: foo2.x }
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn test_simple() {
|
|
let foo1 = Foo1 { x: 100 };
|
|
let foo2 = Foo2 { x: 200 };
|
|
let foo1_encoded = foo1.native_model_encode().unwrap();
|
|
let foo2_encoded = foo2.native_model_encode().unwrap();
|
|
|
|
let (foo1_decoded, _) = Foo1::native_model_decode(foo1_encoded).unwrap();
|
|
assert!(foo1_decoded == foo1);
|
|
let (foo2_decoded, _) = Foo2::native_model_decode(foo2_encoded).unwrap();
|
|
assert!(foo2_decoded == foo2);
|
|
}
|