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
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161

use chrono;
use uuid;
use std::net::{SocketAddr, IpAddr};
use std::fmt;
use std::io::Write;

/// SQL format code (text or binary).
#[repr(i16)]
pub enum FormatCode {
    /// Text format
    Text = 0,
    /// Binary format
    Binary = 1,
}

/// Postgres code for types.
#[derive(Debug, Copy, Clone, PartialEq, Eq)]
#[repr(i32)]
pub enum SqlType {
    /// Dynamic type
    None = 0,
}


/// Trait used to convert a type to its SQL representation.
pub trait ToSql {
    /// Postgres type code for this type.
    fn sql_type(&self) -> SqlType {
        SqlType::None
    }
    /// Postgres format code for this type.
    fn format_code(&self) -> FormatCode {
        FormatCode::Text
    }
    /// How to serialize this type in a Postgres-compatible way.
    fn write(&self, w: &mut Vec<u8>);
}

impl ToSql for bool {
    fn sql_type(&self) -> SqlType {
        SqlType::None
    }
    fn write(&self, w: &mut Vec<u8>) {
        if *self {
            write!(w, "TRUE").unwrap()
        } else {
            write!(w, "FALSE").unwrap()
        }
    }
}

impl<Tz: chrono::TimeZone> ToSql for chrono::DateTime<Tz>
    where Tz::Offset: fmt::Display
{
    fn sql_type(&self) -> SqlType {
        SqlType::None
    }
    fn write(&self, w: &mut Vec<u8>) {
        debug!("RFC3339: {:?}", self.to_rfc3339());
        w.write(self.to_rfc3339().as_bytes()).unwrap();
    }
}

/// Parse the timestamp format returned by postgres queries.
pub fn parse_timestamp(s: &str) -> Result<chrono::DateTime<chrono::UTC>, chrono::format::ParseError> {
    debug!("parse_timestamp: {:?}", s);
    Ok(chrono::DateTime::<chrono::UTC>::from_utc(
        chrono::NaiveDateTime::parse_from_str(s, "%F %T%.f+00")?,
        chrono::UTC
    ))
}

impl ToSql for i64 {
    fn sql_type(&self) -> SqlType {
        SqlType::None
    }
    fn write(&self, w: &mut Vec<u8>) {
        write!(w, "{}", *self).unwrap();
    }
}


impl ToSql for i32 {
    fn sql_type(&self) -> SqlType {
        SqlType::None
    }
    fn write(&self, w: &mut Vec<u8>) {
        write!(w, "{}", *self).unwrap();
    }
}

impl ToSql for String {
    fn sql_type(&self) -> SqlType {
        SqlType::None
    }
    fn write(&self, w: &mut Vec<u8>) {
        write!(w, "{}", self).unwrap();
    }
}

impl<'a> ToSql for &'a str {
    fn sql_type(&self) -> SqlType {
        SqlType::None
    }
    fn write(&self, w: &mut Vec<u8>) {
        write!(w, "{}", self).unwrap();
    }
}

impl ToSql for uuid::Uuid {
    fn write(&self, w: &mut Vec<u8>) {
        write!(w, "{}", self.hyphenated()).unwrap();
    }
}

impl ToSql for SocketAddr {
    fn write(&self, w: &mut Vec<u8>) {
        write!(w, "{}", self).unwrap();
    }
}

impl ToSql for IpAddr {
    fn write(&self, w: &mut Vec<u8>) {
        write!(w, "{}", self).unwrap();
    }
}

/// Types representing statements.
pub trait Statement {
    /// SQL statement.
    fn statement(&self) -> &str;

    /// Statement names. The default implementation returns the "unname".
    fn name(&self) -> &str {
        ""
    }
}

impl<'a, S: Statement> Statement for &'a S {
    fn name(&self) -> &str {
        (*self).name()
    }
    fn statement(&self) -> &str {
        (*self).statement()
    }
}

impl<'a> Statement for &'a str {
    fn statement(&self) -> &str {
        self
    }
}

impl<'a, 'b> Statement for (&'a str, &'b str) {
    fn statement(&self) -> &str {
        self.1
    }
    fn name(&self) -> &str {
        self.0
    }
}