extern crate csv;
extern crate serde;
#[macro_use]
extern crate serde_derive;
extern crate serde_qs as qs;
use serde::de::DeserializeOwned;
use std::default::Default;
#[derive(Debug, Deserialize, Serialize)]
struct Query {
#[serde(deserialize_with = "from_csv")]
r: Vec<u8>,
s: u8,
}
fn main() {
let q = "s=12&r=1,2,3";
let q: Query = qs::from_str(q).unwrap();
println!("{:?}", q);
}
fn from_csv<'de, D, T>(deserializer: D) -> Result<Vec<T>, D::Error>
where
D: serde::Deserializer<'de>,
T: DeserializeOwned,
{
deserializer.deserialize_str(CSVVecVisitor::<T>::default())
}
struct CSVVecVisitor<T: DeserializeOwned>(std::marker::PhantomData<T>);
impl<T: DeserializeOwned> Default for CSVVecVisitor<T> {
fn default() -> Self {
CSVVecVisitor(std::marker::PhantomData)
}
}
impl<'de, T: DeserializeOwned> serde::de::Visitor<'de> for CSVVecVisitor<T> {
type Value = Vec<T>;
fn expecting(
&self,
formatter: &mut std::fmt::Formatter,
) -> std::fmt::Result {
write!(formatter, "a str")
}
fn visit_str<E>(self, s: &str) -> std::result::Result<Self::Value, E>
where
E: serde::de::Error,
{
let mut output = Vec::new();
let mut items = csv::Reader::from_reader(s.as_bytes());
for res in items.deserialize() {
let item: T = res.map_err(|e| {
E::custom(format!(
"could not deserialize sequence value: {:?}",
e
))
})?;
output.push(item);
}
Ok(output)
}
}