mikros/serde
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Simplify generated visitors

Browse Source
This commit is contained in:
David Tolnay 2016-08-03 19:53:41 -07:00
parent b289edd4a4
commit d5102a7afd
No known key found for this signature in database
GPG Key ID: F9BA143B95FF6D82
1 changed files with 20 additions and 58 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

78
serde_codegen/src/de.rs
View File

@ -192,31 +192,24 @@ fn deserialize_body(
// Build `__Visitor<A, B, ...>(PhantomData<A>, PhantomData<B>, ...)`
fn deserialize_visitor(
builder: &aster::AstBuilder,
trait_generics: &ast::Generics,
forward_ty_params: Vec<ast::TyParam>,
forward_tys: Vec<P<ast::Ty>>
) -> (P<ast::Item>, P<ast::Ty>, P<ast::Expr>, ast::Generics) {
if trait_generics.ty_params.is_empty() && forward_tys.is_empty() {
generics: &ast::Generics,
) -> (P<ast::Item>, P<ast::Ty>, P<ast::Expr>) {
if generics.ty_params.is_empty() {
(
builder.item().tuple_struct("__Visitor").build(),
builder.item().unit_struct("__Visitor"),
builder.ty().id("__Visitor"),
builder.expr().id("__Visitor"),
trait_generics.clone(),
)
} else {
let placeholders : Vec<_> = trait_generics.ty_params.iter()
let placeholders : Vec<_> = generics.ty_params.iter()
.map(|t| builder.ty().id(t.ident))
.collect();
let mut trait_generics = trait_generics.clone();
let mut ty_params = forward_ty_params.clone();
ty_params.extend(trait_generics.ty_params.into_vec());
trait_generics.ty_params = P::from_vec(ty_params);
(
builder.item().tuple_struct("__Visitor")
.generics().with(trait_generics.clone()).build()
.generics().with(generics.clone()).build()
.with_tys({
let lifetimes = trait_generics.lifetimes.iter()
let lifetimes = generics.lifetimes.iter()
.map(|lifetime_def| {
builder.ty()
.phantom_data()
@ -225,7 +218,7 @@ fn deserialize_visitor(
.unit()
});
let ty_params = trait_generics.ty_params.iter()
let ty_params = generics.ty_params.iter()
.map(|ty_param| {
builder.ty()
.phantom_data()
@ -236,39 +229,22 @@ fn deserialize_visitor(
})
.build(),
builder.ty().path()
.segment("__Visitor").with_generics(trait_generics.clone()).build()
.segment("__Visitor").with_generics(generics.clone()).build()
.build(),
builder.expr().call()
.path().segment("__Visitor")
.with_tys(forward_tys)
.with_tys(placeholders)
.build().build()
.with_args({
let len = trait_generics.lifetimes.len() + trait_generics.ty_params.len();
let len = generics.lifetimes.len() + generics.ty_params.len();
(0 .. len).map(|_| builder.expr().phantom_data())
})
.build(),
trait_generics,
)
}
}
fn deserializer_ty_param(builder: &aster::AstBuilder) -> ast::TyParam {
builder.ty_param("__D")
.trait_bound(builder.path()
.segment("_serde").build()
.segment("de").build()
.id("Deserializer")
.build())
.build()
.build()
}
fn deserializer_ty_arg(builder: &aster::AstBuilder) -> P<ast::Ty>{
builder.ty().id("__D")
}
fn deserialize_unit_struct(
cx: &ExtCtxt,
builder: &aster::AstBuilder,
@ -315,11 +291,9 @@ fn deserialize_tuple(
) -> P<ast::Expr> {
let where_clause = &impl_generics.where_clause;
let (visitor_item, visitor_ty, visitor_expr, visitor_generics) = deserialize_visitor(
let (visitor_item, visitor_ty, visitor_expr) = deserialize_visitor(
builder,
impl_generics,
vec![deserializer_ty_param(builder)],
vec![deserializer_ty_arg(builder)],
);
let is_enum = variant_ident.is_some();
@ -369,7 +343,7 @@ fn deserialize_tuple(
quote_expr!(cx, {
$visitor_item
impl $visitor_generics _serde::de::Visitor for $visitor_ty $where_clause {
impl $impl_generics _serde::de::Visitor for $visitor_ty $where_clause {
type Value = $ty;
$visit_newtype_struct
@ -516,11 +490,9 @@ fn deserialize_struct(
) -> P<ast::Expr> {
let where_clause = &impl_generics.where_clause;
let (visitor_item, visitor_ty, visitor_expr, visitor_generics) = deserialize_visitor(
let (visitor_item, visitor_ty, visitor_expr) = deserialize_visitor(
builder,
&impl_generics,
vec![deserializer_ty_param(builder)],
vec![deserializer_ty_arg(builder)],
impl_generics,
);
let type_path = match variant_ident {
@ -563,7 +535,7 @@ fn deserialize_struct(
$visitor_item
impl $visitor_generics _serde::de::Visitor for $visitor_ty $where_clause {
impl $impl_generics _serde::de::Visitor for $visitor_ty $where_clause {
type Value = $ty;
#[inline]
@ -648,11 +620,9 @@ fn deserialize_item_enum(
}
variant_arms.extend(ignored_arm.into_iter());
let (visitor_item, visitor_ty, visitor_expr, visitor_generics) = deserialize_visitor(
let (visitor_item, visitor_ty, visitor_expr) = deserialize_visitor(
builder,
impl_generics,
vec![deserializer_ty_param(builder)],
vec![deserializer_ty_arg(builder)],
);
quote_expr!(cx, {
@ -660,7 +630,7 @@ fn deserialize_item_enum(
$visitor_item
impl $visitor_generics _serde::de::EnumVisitor for $visitor_ty $where_clause {
impl $impl_generics _serde::de::EnumVisitor for $visitor_ty $where_clause {
type Value = $ty;
fn visit<__V>(&mut self, mut visitor: __V) -> ::std::result::Result<$ty, __V::Error>
@ -879,13 +849,9 @@ fn deserialize_field_visitor(
fn deserialize<__D>(deserializer: &mut __D) -> ::std::result::Result<__Field, __D::Error>
where __D: _serde::de::Deserializer,
{
struct __FieldVisitor<__D> {
phantom: ::std::marker::PhantomData<__D>
}
struct __FieldVisitor;
impl<__D> _serde::de::Visitor for __FieldVisitor<__D>
where __D: _serde::de::Deserializer
{
impl _serde::de::Visitor for __FieldVisitor {
type Value = __Field;
fn visit_usize<__E>(&mut self, value: usize) -> ::std::result::Result<__Field, __E>
@ -907,11 +873,7 @@ fn deserialize_field_visitor(
}
}
deserializer.deserialize_struct_field(
__FieldVisitor::<__D>{
phantom: ::std::marker::PhantomData
}
)
deserializer.deserialize_struct_field(__FieldVisitor)
}
}
).unwrap();