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

use crate::io::BufMut;
use crate::postgres::protocol::Encode;
use byteorder::NetworkEndian;

pub struct Query<'a>(pub &'a str);

impl Encode for Query<'_> {
    fn encode(&self, buf: &mut Vec<u8>) {
        buf.push(b'Q');

        // len + query + nul
        buf.put_i32::<NetworkEndian>((4 + self.0.len() + 1) as i32);

        buf.put_str_nul(self.0);
    }
}

#[cfg(test)]
mod tests {
    use super::{Encode, Query};

    const QUERY_SELECT_1: &[u8] = b"Q\0\0\0\rSELECT 1\0";

    #[test]
    fn it_encodes_query() {
        let mut buf = Vec::new();
        let m = Query("SELECT 1");

        m.encode(&mut buf);

        assert_eq!(buf, QUERY_SELECT_1);
    }
}