Restore visit_bytes for identifying variants and fields

This commit is contained in:
David Tolnay 2017-01-18 21:11:51 -08:00
parent b1fbbfd3ce
commit 0a10116bf5
No known key found for this signature in database
GPG Key ID: F9BA143B95FF6D82
2 changed files with 36 additions and 12 deletions

View File

@ -1,5 +1,5 @@
use syn::{self, aster, Ident};
use quote::Tokens;
use quote::{self, Tokens};
use bound;
use internals::ast::{Body, Field, Item, Style, Variant};
@ -648,7 +648,8 @@ fn deserialize_field_visitor(
item_attrs: &attr::Item,
is_variant: bool,
) -> Tokens {
let field_names = fields.iter().map(|&(ref name, _)| name);
let field_strs = fields.iter().map(|&(ref name, _)| name);
let field_bytes = fields.iter().map(|&(ref name, _)| quote::ByteStr(name));
let field_idents: &Vec<_> = &fields.iter().map(|&(_, ref ident)| ident).collect();
let ignore_variant = if is_variant || item_attrs.deny_unknown_fields() {
@ -690,6 +691,16 @@ fn deserialize_field_visitor(
}
};
let bytes_to_str = if is_variant || item_attrs.deny_unknown_fields() {
Some(quote! {
// TODO https://github.com/serde-rs/serde/issues/666
// update this to use str::from_utf8(value).unwrap_or("<22><><EFBFBD>") on no_std
let value = &::std::string::String::from_utf8_lossy(value);
})
} else {
None
};
quote! {
#[allow(non_camel_case_types)]
enum __Field {
@ -714,11 +725,25 @@ fn deserialize_field_visitor(
{
match value {
#(
#field_names => Ok(__Field::#field_idents),
#field_strs => Ok(__Field::#field_idents),
)*
_ => #fallthrough_arm
}
}
fn visit_bytes<__E>(self, value: &[u8]) -> ::std::result::Result<__Field, __E>
where __E: _serde::de::Error
{
match value {
#(
#field_bytes => Ok(__Field::#field_idents),
)*
_ => {
#bytes_to_str
#fallthrough_arm
}
}
}
}
deserializer.deserialize_struct_field(__FieldVisitor)

View File

@ -3,7 +3,7 @@ use std::net;
use std::path::PathBuf;
use std::time::Duration;
use serde::de::{Deserialize, Type};
use serde::de::Deserialize;
extern crate fnv;
use self::fnv::FnvHasher;
@ -781,6 +781,13 @@ declare_tests! {
Token::Unit,
],
}
test_enum_unit_bytes {
Enum::Unit => &[
Token::EnumStart("Enum"),
Token::Bytes(b"Unit"),
Token::Unit,
],
}
test_box {
Box::new(0i32) => &[Token::I32(0)],
}
@ -933,12 +940,4 @@ declare_error_tests! {
],
Error::InvalidValue("expected variant index 0 <= i < 4".to_owned()),
}
test_enum_unit_bytes<Enum> {
&[
Token::EnumStart("Enum"),
Token::Bytes(b"Unit"),
Token::Unit,
],
Error::InvalidType(Type::Bytes),
}
}