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

use byteorder::{ByteOrder, NetworkEndian};

use crate::arguments::Arguments;
use crate::encode::{Encode, IsNull};
use crate::io::BufMut;
use crate::postgres::{PgRawBuffer, PgTypeInfo, Postgres};
use crate::types::Type;

#[derive(Default,Debug)]
pub struct PgArguments {
    // Types of the bind parameters

    pub(super) types: Vec<PgTypeInfo>,

    // Write buffer for serializing bind values

    pub(super) buffer: PgRawBuffer,
}

impl Arguments for PgArguments {
    type Database = super::Postgres;

    fn reserve(&mut self, len: usize, size: usize) {
        self.types.reserve(len);
        self.buffer.reserve(size);
    }

    fn add<T>(&mut self, value: T)
    where
        T: Type<Self::Database> + Encode<Self::Database>,
    {
        // TODO: When/if we receive types that do _not_ support BINARY, we need to check here

        // TODO: There is no need to be explicit unless we are expecting mixed BINARY / TEXT


        self.types.push(<T as Type<Postgres>>::type_info());

        // Reserves space for the length of the value

        let pos = self.buffer.len();
        self.buffer.put_i32::<NetworkEndian>(0);

        let len = if let IsNull::No = value.encode_nullable(&mut self.buffer) {
            (self.buffer.len() - pos - 4) as i32
        } else {
            // Write a -1 for the len to indicate NULL

            // TODO: It is illegal for [encode] to write any data

            //       if IsSql::No; fail a debug assertion

            -1
        };

        // Write-back the len to the beginning of this frame (not including the len of len)

        NetworkEndian::write_i32(&mut self.buffer[pos..], len as i32);
    }
}