86 lines
3.5 KiB
Rust
86 lines
3.5 KiB
Rust
use proc_macro::TokenStream;
|
||
use quote::quote;
|
||
use syn::Data;
|
||
|
||
fn impl_serialize_ex(ast: &syn::DeriveInput) -> TokenStream {
|
||
let name = &ast.ident;
|
||
let fields = match ast.data {
|
||
Data::Struct(ref data) => &data.fields,
|
||
_ => panic!("CustomSerialize 只能用于结构体！"),
|
||
};
|
||
let field_len = fields.len();
|
||
// 迭代器，为每一个字段实现serialize_field
|
||
let serialize_fields = match fields {
|
||
syn::Fields::Named(ref fields) => {
|
||
fields.named.iter().map(|f| {
|
||
let name = &f.ident;
|
||
let ty = &f.ty;
|
||
let mut child_token_stream = proc_macro2::TokenStream::new();
|
||
// 如果字段是数组类型
|
||
if let syn::Type::Array(ref array) = ty {
|
||
let len = &array.len;
|
||
let child_ty = &array.elem;
|
||
// 类型所占字节
|
||
|
||
child_token_stream = quote! {
|
||
children = Some(Vec::new());
|
||
let field_size = std::mem::size_of::<#child_ty>();
|
||
for i in 0..#len {
|
||
let child_value = serde_json::json!({
|
||
"name": format!("{}[{}]", stringify!(#name), i),
|
||
"value": format!("0x{:0width$X}",self.#name[i], width=field_size),
|
||
"field_type": stringify!(#child_ty),
|
||
"size": field_size,
|
||
"offset": (&self.#name[i] as *const _ as usize) - (self as *const _ as usize),
|
||
"children": null
|
||
});
|
||
children.as_mut().unwrap().push(child_value);
|
||
}
|
||
};
|
||
}else{
|
||
child_token_stream = quote! {
|
||
let field_size = std::mem::size_of::<#ty>();
|
||
value = serde_json::json!(format!("0x{:0width$X}",&self.#name, width=field_size));
|
||
};
|
||
}
|
||
quote! {
|
||
let mut value = serde_json::Value::Null;
|
||
let mut children: Option<Vec<serde_json::Value>> = None;
|
||
#child_token_stream
|
||
|
||
let json_value = serde_json::json!({
|
||
"name": stringify!(#name),
|
||
"value": value,
|
||
"field_type": stringify!(#ty),
|
||
"size": std::mem::size_of::<#ty>(),
|
||
"offset": (&self.#name as *const _ as usize) - (self as *const _ as usize),
|
||
"children": &children
|
||
});
|
||
state.serialize_field(stringify!(#name), &json_value)?;
|
||
}
|
||
})
|
||
}
|
||
_ => panic!("CustomSerialize 只能用于结构体！"),
|
||
};
|
||
|
||
let gen = quote! {
|
||
impl serde::Serialize for #name {
|
||
fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
|
||
where
|
||
S: serde::Serializer {
|
||
let mut state = serializer.serialize_struct(stringify!(#name), #field_len)?;
|
||
#(#serialize_fields)*
|
||
state.end()
|
||
}
|
||
}
|
||
};
|
||
gen.into()
|
||
}
|
||
|
||
// 拓展Serialize
|
||
#[proc_macro_derive(SerializeEX)]
|
||
pub fn serialize_ex(input: TokenStream) -> TokenStream {
|
||
let ast = syn::parse(input).unwrap();
|
||
impl_serialize_ex(&ast)
|
||
}
|