summaryrefslogtreecommitdiff
path: root/src/main.rs
blob: ba54da8fa0a07b16a0d6175536b72cf2fdb810ac (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
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
use serde::{Deserialize, Serialize};

#[derive(Debug, Serialize, Deserialize)]
struct Specification {
    benchmarks: Vec<Benchmark>,
}

#[derive(Debug, Serialize, Deserialize)]
struct Benchmark {
    benchmark: String,
    backups: Vec<Backup>,
}

#[derive(Debug, Serialize, Deserialize)]
struct Backup {
    changes: Vec<Change>,
}

#[derive(Debug, Serialize, Deserialize)]
enum Change {
    Create(CreateFiles),
    Delete(FileCount),
    Rename(FileCount),
}

#[derive(Debug, Serialize, Deserialize)]
struct CreateFiles {
    files: u64,
    #[serde(with = "file_size")]
    file_size: u64,
}

#[derive(Debug, Serialize, Deserialize)]
struct FileCount {
    files: u64,
}

fn main() {
    let yaml = r#"
benchmarks:
- benchmark: foo
  backups: []
"#;
    let spec: Specification = serde_yaml::from_str(yaml).unwrap();
    serde_json::to_writer(std::io::stdout(), &spec).unwrap();
    println!();
}

mod file_size {
    use serde::{de::Error as _, Deserialize, Deserializer, Serialize, Serializer};

    // This deserializer was originally written with u64 in mind. Then it was made generic by
    // changing u64 to T everywhere and adding boundaries. Same with the serializer.
    pub fn deserialize<'de, D, T>(deserializer: D) -> Result<T, D::Error>
    where
        D: Deserializer<'de>,
        T: std::str::FromStr,
        <T as std::str::FromStr>::Err: std::fmt::Display,
    {
        String::deserialize(deserializer)?
            .parse::<T>()
            .map_err(|e| D::Error::custom(format!("{}", e)))
    }

    pub fn serialize<S, T>(value: &T, serializer: S) -> Result<S::Ok, S::Error>
    where
        S: Serializer,
        T: std::fmt::Display,
    {
        format!("{}", value).serialize(serializer)
    }
}