summaryrefslogtreecommitdiff
path: root/examples/csv_vectors.rs
blob: 39e95a1ec1be4298e79ca62282049f0bebc57a83 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
extern crate csv;
extern crate serde;
#[macro_use]
extern crate serde_derive;
extern crate serde_qs as qs;

#[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<D>(deserializer: D) -> Result<Vec<u8>, D::Error>
    where D: serde::Deserializer,
{
    deserializer.deserialize_str(CSVVecVisitor)
}

/// Visits a string value of the form "v1,v2,v3" into a vector of bytes Vec<u8>
struct CSVVecVisitor;

impl serde::de::Visitor for CSVVecVisitor {
    type Value = Vec<u8>;

    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_string(s);
        // let items = items.next_str();
        while let csv::NextField::Data(item) = items.next_str() {
            output.push(u8::from_str_radix(item, 10).unwrap());
        }

        Ok(output)
    }
}