add impl Deserializer for BatchInfo
All checks were successful
continuous-integration/drone/push Build is passing

This commit is contained in:
Dmitry Belyaev 2023-08-24 22:28:19 +03:00
parent af9f5d8f24
commit f1aed09319

View File

@ -1,4 +1,13 @@
use serde_derive::{Deserialize, Serialize}; use std::{
fmt::{self, Formatter},
marker::PhantomData,
};
use serde::{
de::{MapAccess, SeqAccess, Visitor},
Deserialize, Deserializer,
};
use serde::ser::SerializeStruct; use serde::ser::SerializeStruct;
use bitflags::bitflags; use bitflags::bitflags;
@ -45,33 +54,20 @@ bitflags! {
} }
} }
#[derive(Debug, Default, Clone, Deserialize, PartialEq)] #[derive(Debug, Default, Clone, PartialEq)]
pub struct BatchInfo { pub struct BatchInfo {
#[serde(default)]
pub filename: String, pub filename: String,
#[serde(default)]
pub description: String, pub description: String,
#[serde(default)]
pub author: String, pub author: String,
#[serde(default)]
pub comment: String, pub comment: String,
#[serde(default)]
pub url: String, pub url: String,
#[serde(default)]
pub date: String, pub date: String,
#[serde(default)]
pub processed_by: String, pub processed_by: String,
#[serde(default)]
pub redacted_by: String, pub redacted_by: String,
#[serde(default)]
pub copyright: String, pub copyright: String,
#[serde(default)]
pub theme: String, pub theme: String,
#[serde(default)]
pub kind: String, pub kind: String,
#[serde(default)]
pub source: String, pub source: String,
#[serde(default)]
pub rating: String, pub rating: String,
} }
@ -149,8 +145,8 @@ impl serde::Serialize for BatchInfo {
let mut len = 1; // (+flags) let mut len = 1; // (+flags)
count_string_fields!((self, flags, len, BatchFlags) <- { count_string_fields!((self, flags, len, BatchFlags) <- {
filename: FILENAME, description: DESCRIPTION, author: AUTHOR, comment: COMMENT, url: URL, date: DATE, filename: FILENAME, description: DESCRIPTION, author: AUTHOR, comment: COMMENT, url: URL, date: DATE, processed_by: PROCESSED,
processed_by: PROCESSED, copyright: COPYRIGHT, theme: THEME, kind: KIND, source: SOURCE, rating: RATING redacted_by: REDACTED, copyright: COPYRIGHT, theme: THEME, kind: KIND, source: SOURCE, rating: RATING
}); });
let mut state = serializer.serialize_struct("BatchInfo", len)?; let mut state = serializer.serialize_struct("BatchInfo", len)?;
@ -162,8 +158,8 @@ impl serde::Serialize for BatchInfo {
} }
serialize_fields!((self, flags, state, BatchFlags) <- { serialize_fields!((self, flags, state, BatchFlags) <- {
filename: FILENAME, description: DESCRIPTION, author: AUTHOR, comment: COMMENT, url: URL, date: DATE, filename: FILENAME, description: DESCRIPTION, author: AUTHOR, comment: COMMENT, url: URL, date: DATE, processed_by: PROCESSED,
processed_by: PROCESSED, copyright: COPYRIGHT, theme: THEME, kind: KIND, source: SOURCE, rating: RATING redacted_by: REDACTED, copyright: COPYRIGHT, theme: THEME, kind: KIND, source: SOURCE, rating: RATING
}); });
state.end() state.end()
@ -227,6 +223,269 @@ impl serde::Serialize for Question {
} }
} }
impl<'de> Deserialize<'de> for BatchInfo {
fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
where
D: Deserializer<'de>,
{
#[allow(non_camel_case_types)]
#[doc(hidden)]
enum BatchField {
filename,
description,
author,
comment,
url,
date,
processed_by,
redacted_by,
copyright,
theme,
kind,
source,
rating,
_flags,
__ignore,
}
struct BatchFieldVisitor;
impl<'de> Visitor<'de> for BatchFieldVisitor {
type Value = BatchField;
fn expecting(&self, formatter: &mut Formatter) -> fmt::Result {
Formatter::write_str(formatter, "field identifier")
}
fn visit_u64<E>(self, value: u64) -> Result<Self::Value, E>
where
E: serde::de::Error,
{
match value {
0u64 => Ok(BatchField::filename),
1u64 => Ok(BatchField::description),
2u64 => Ok(BatchField::author),
3u64 => Ok(BatchField::comment),
4u64 => Ok(BatchField::url),
5u64 => Ok(BatchField::date),
6u64 => Ok(BatchField::processed_by),
7u64 => Ok(BatchField::redacted_by),
8u64 => Ok(BatchField::copyright),
9u64 => Ok(BatchField::theme),
10u64 => Ok(BatchField::kind),
11u64 => Ok(BatchField::source),
12u64 => Ok(BatchField::rating),
13u64 => Ok(BatchField::_flags),
_ => Ok(BatchField::__ignore),
}
}
fn visit_str<E>(self, value: &str) -> Result<Self::Value, E>
where
E: serde::de::Error,
{
match value {
"filename" => Ok(BatchField::filename),
"description" => Ok(BatchField::description),
"author" => Ok(BatchField::author),
"comment" => Ok(BatchField::comment),
"url" => Ok(BatchField::url),
"date" => Ok(BatchField::date),
"processed_by" => Ok(BatchField::processed_by),
"redacted_by" => Ok(BatchField::redacted_by),
"copyright" => Ok(BatchField::copyright),
"theme" => Ok(BatchField::theme),
"kind" => Ok(BatchField::kind),
"source" => Ok(BatchField::source),
"rating" => Ok(BatchField::rating),
"_flags" => Ok(BatchField::_flags),
_ => Ok(BatchField::__ignore),
}
}
fn visit_bytes<E>(self, value: &[u8]) -> Result<Self::Value, E>
where
E: serde::de::Error,
{
match value {
b"filename" => Ok(BatchField::filename),
b"description" => Ok(BatchField::description),
b"author" => Ok(BatchField::author),
b"comment" => Ok(BatchField::comment),
b"url" => Ok(BatchField::url),
b"date" => Ok(BatchField::date),
b"processed_by" => Ok(BatchField::processed_by),
b"redacted_by" => Ok(BatchField::redacted_by),
b"copyright" => Ok(BatchField::copyright),
b"theme" => Ok(BatchField::theme),
b"kind" => Ok(BatchField::kind),
b"source" => Ok(BatchField::source),
b"rating" => Ok(BatchField::rating),
b"_flags" => Ok(BatchField::_flags),
_ => Ok(BatchField::__ignore),
}
}
}
impl<'de> Deserialize<'de> for BatchField {
#[inline]
fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
where
D: Deserializer<'de>,
{
Deserializer::deserialize_identifier(deserializer, BatchFieldVisitor)
}
}
#[doc(hidden)]
struct BatchInfoVisitor<'de> {
marker: PhantomData<BatchInfo>,
lifetime: PhantomData<&'de ()>,
}
impl<'de> Visitor<'de> for BatchInfoVisitor<'de> {
type Value = BatchInfo;
fn expecting(&self, formatter: &mut Formatter) -> fmt::Result {
Formatter::write_str(formatter, "struct BatchInfo")
}
#[inline]
fn visit_seq<V>(self, mut seq: V) -> Result<Self::Value, V::Error>
where
V: SeqAccess<'de>,
{
let _flags = BatchFlags::from_bits(
SeqAccess::next_element::<u16>(&mut seq)?.unwrap_or_default(),
)
.unwrap_or_default();
macro_rules! seq_read_strings {
(($flags:ident, $seq:ident, $FlagsType:ident) <- {$($field:ident:$flag:ident),+} ) => {$(
let $field: String = if $flags.intersects($FlagsType::$flag) {
SeqAccess::next_element::<String>(&mut seq)?.unwrap_or_default()
} else {
Default::default()
};
)+}
}
seq_read_strings!((_flags, seq, BatchFlags) <- {
filename: FILENAME, description: DESCRIPTION, author: AUTHOR, comment: COMMENT, url: URL, date: DATE, processed_by: PROCESSED,
redacted_by: REDACTED, copyright: COPYRIGHT, theme: THEME, kind: KIND, source: SOURCE, rating: RATING
});
Ok(BatchInfo {
filename,
description,
author,
comment,
url,
date,
processed_by,
redacted_by,
copyright,
theme,
kind,
source,
rating,
})
}
#[inline]
fn visit_map<V>(self, mut map: V) -> Result<Self::Value, V::Error>
where
V: MapAccess<'de>,
{
let mut filename: Option<String> = None;
let mut description: Option<String> = None;
let mut author: Option<String> = None;
let mut comment: Option<String> = None;
let mut url: Option<String> = None;
let mut date: Option<String> = None;
let mut processed_by: Option<String> = None;
let mut redacted_by: Option<String> = None;
let mut copyright: Option<String> = None;
let mut theme: Option<String> = None;
let mut kind: Option<String> = None;
let mut source: Option<String> = None;
let mut rating: Option<String> = None;
macro_rules! match_map_fields {
(($map:ident, $key:ident, $FieldType:ident) <- {$($field:ident),+} ) => {
match $key {
$(
$FieldType::$field => {
if $field.is_some() {
return Err(<V::Error as serde::de::Error>::duplicate_field(
std::stringify!($field),
));
}
$field = Some(MapAccess::next_value::<String>(&mut $map)?);
},
)+
_ => {
let _ = MapAccess::next_value::<serde::de::IgnoredAny>(&mut $map)?;
}
}
}
}
while let Some(key) = MapAccess::next_key::<BatchField>(&mut map)? {
match_map_fields!((map, key, BatchField) <- {
filename, description, author, comment, url, date, processed_by,
redacted_by, copyright, theme, kind, source, rating
});
}
let filename = filename.unwrap_or_default();
let description = description.unwrap_or_default();
let author = author.unwrap_or_default();
let comment = comment.unwrap_or_default();
let url = url.unwrap_or_default();
let date = date.unwrap_or_default();
let processed_by = processed_by.unwrap_or_default();
let redacted_by = redacted_by.unwrap_or_default();
let copyright = copyright.unwrap_or_default();
let theme = theme.unwrap_or_default();
let kind = kind.unwrap_or_default();
let source = source.unwrap_or_default();
let rating = rating.unwrap_or_default();
Ok(BatchInfo {
filename,
description,
author,
comment,
url,
date,
processed_by,
redacted_by,
copyright,
theme,
kind,
source,
rating,
})
}
}
#[doc(hidden)]
const FIELDS: &[&str] = &[
"filename",
"description",
"author",
"comment",
"url",
"date",
"processed_by",
"redacted_by",
"copyright",
"theme",
"kind",
"source",
"rating",
];
Deserializer::deserialize_struct(
deserializer,
"BatchInfo",
FIELDS,
BatchInfoVisitor {
marker: PhantomData::<BatchInfo>,
lifetime: PhantomData,
},
)
}
}
#[cfg(any(feature = "convert", feature = "convert_async"))] #[cfg(any(feature = "convert", feature = "convert_async"))]
pub mod convert_common { pub mod convert_common {
use super::{BatchInfo, Question}; use super::{BatchInfo, Question};