cdbc_mysql/
arguments.rs

1use cdbc::arguments::Arguments;
2use cdbc::encode::{Encode, IsNull};
3use crate::{MySql, MySqlTypeInfo};
4use cdbc::types::Type;
5
6/// Implementation of [`Arguments`] for MySQL.
7#[derive(Debug, Default)]
8pub struct MySqlArguments {
9    pub(crate) values: Vec<u8>,
10    pub(crate) types: Vec<MySqlTypeInfo>,
11    pub(crate) null_bitmap: Vec<u8>,
12}
13
14impl MySqlArguments {
15    pub(crate) fn add<'q, T>(&mut self, value: T)
16    where
17        T: Encode<'q, MySql> + Type<MySql>,
18    {
19        let ty = value.produces().unwrap_or_else(T::type_info);
20        let index = self.types.len();
21
22        self.types.push(ty);
23        self.null_bitmap.resize((index / 8) + 1, 0);
24
25        if let IsNull::Yes = value.encode(&mut self.values) {
26            self.null_bitmap[index / 8] |= (1 << (index % 8)) as u8;
27        }
28    }
29}
30
31impl<'q> Arguments<'q> for MySqlArguments {
32    type Database = MySql;
33
34    fn reserve(&mut self, len: usize, size: usize) {
35        self.types.reserve(len);
36        self.values.reserve(size);
37    }
38
39    fn add<T>(&mut self, value: T)
40    where
41        T: Encode<'q, Self::Database> + Type<Self::Database>,
42    {
43        self.add(value)
44    }
45}