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
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103

//! Functions for fetching from quaint result rows

use quaint::connector::ResultRow;

pub trait Getter {
    fn get_expect_string(&self, name: &str) -> String;
    fn get_expect_char(&self, name: &str) -> char;
    fn get_expect_i64(&self, name: &str) -> i64;
    fn get_expect_bool(&self, name: &str) -> bool;
    fn get_expect_string_array(&self, name: &str) -> Vec<String>;

    fn get_string_array(&self, name: &str) -> Option<Vec<String>>;
    fn get_char(&self, name: &str) -> Option<char>;
    fn get_string(&self, name: &str) -> Option<String>;
    fn get_bool(&self, name: &str) -> Option<bool>;
    fn get_u32(&self, name: &str) -> Option<u32>;
    fn get_i64(&self, name: &str) -> Option<i64>;
}

impl Getter for ResultRow {
    #[track_caller]
    fn get_expect_string(&self, name: &str) -> String {
        self.get(name)
            .and_then(|x| x.to_string())
            .ok_or_else(|| format!("Getting {} from Resultrow {:?} as String failed", name, &self))
            .unwrap()
    }

    #[track_caller]
    fn get_expect_char(&self, name: &str) -> char {
        self.get(name)
            .and_then(|x| x.as_char())
            .ok_or_else(|| format!("Getting {} from Resultrow {:?} as char failed", name, &self))
            .unwrap()
    }

    #[track_caller]
    fn get_expect_i64(&self, name: &str) -> i64 {
        self.get(name)
            .and_then(|x| x.as_integer())
            .ok_or_else(|| format!("Getting {} from Resultrow {:?} as i64 failed", name, &self))
            .unwrap()
    }

    #[track_caller]
    fn get_expect_bool(&self, name: &str) -> bool {
        self.get_bool(name)
            .ok_or_else(|| format!("Getting {} from Resultrow {:?} as bool failed", name, &self))
            .unwrap()
    }

    #[track_caller]
    fn get_expect_string_array(&self, name: &str) -> Vec<String> {
        self.get(name)
            .and_then(|x| x.to_vec::<String>())
            .ok_or_else(|| format!("Getting {name} from ResultRow {self:?} as Vec<String> failed"))
            .unwrap()
    }

    fn get_string_array(&self, name: &str) -> Option<Vec<String>> {
        self.get(name).and_then(|x| x.to_vec::<String>())
    }

    fn get_char(&self, name: &str) -> Option<char> {
        self.get(name).and_then(|x| x.as_char())
    }

    // At least on MySQL, the encoding of booleans in the information schema
    // seems to be somewhat flexible, so we try to match "0", "1", 0 and 1
    // additionally. See https://github.com/prisma/prisma/issues/5235 for
    // example.
    fn get_bool(&self, name: &str) -> Option<bool> {
        self.get(name).and_then(|x| {
            x.as_bool()
                .or_else(|| {
                    x.as_i64().and_then(|n| match n {
                        0 => Some(false),
                        1 => Some(true),
                        _ => None,
                    })
                })
                .or_else(|| {
                    x.to_string().and_then(|s| match s.trim() {
                        "0" => Some(false),
                        "1" => Some(true),
                        _ => None,
                    })
                })
        })
    }

    fn get_string(&self, name: &str) -> Option<String> {
        self.get(name).and_then(|x| x.to_string())
    }

    fn get_u32(&self, name: &str) -> Option<u32> {
        self.get(name).and_then(|x| x.as_integer().map(|x| x as u32))
    }

    fn get_i64(&self, name: &str) -> Option<i64> {
        self.get(name).and_then(|x| x.as_integer())
    }
}