sqlx_core_oldapi/mysql/
arguments.rs

1use crate::arguments::Arguments;
2use crate::encode::{Encode, IsNull};
3use crate::mysql::{MySql, MySqlTypeInfo};
4use crate::types::Type;
5
6/// Implementation of [`Arguments`] for MySQL.
7#[derive(Debug, Default, Clone)]
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    #[doc(hidden)]
31    pub fn len(&self) -> usize {
32        self.types.len()
33    }
34}
35
36impl<'q> Arguments<'q> for MySqlArguments {
37    type Database = MySql;
38
39    fn reserve(&mut self, len: usize, size: usize) {
40        self.types.reserve(len);
41        self.values.reserve(size);
42    }
43
44    fn add<T>(&mut self, value: T)
45    where
46        T: Encode<'q, Self::Database> + Type<Self::Database>,
47    {
48        self.add(value)
49    }
50}