sqlx-xugu 0.8.6

Xugu driver implementation for SQLx. Not for direct use; see the `sqlx` crate for details.
Documentation 
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

use crate::arguments::XuguArgumentValue;
use crate::connection::StatementId;
use crate::protocol::encode_sql_command;
use crate::protocol::statement::ParameterDef;
use crate::protocol::text::ColumnType;
use crate::XuguArguments;
use bytes::BufMut;
use sqlx_core::io::ProtocolEncode;
use sqlx_core::Error;

#[derive(Debug)]
pub struct Execute<'q, 'p> {
    pub st_id: StatementId,
    pub arguments: &'q XuguArguments<'q>,
    pub params: &'p Vec<ParameterDef>,
}

impl Execute<'_, '_> {
    fn encode_params(&self, buf: &mut Vec<u8>) {
        let params = &self.params;
        let args = &self.arguments.values;
        let types = &self.arguments.types;

        let args_count = args.len() as i32;
        buf.put_i32(args_count);

        for i in 0..args.len() {
            let param_name = &params[i].param_name;
            buf.put_i16(param_name.len() as i16);
            buf.put_slice(param_name.as_bytes());
            let inout_type = params[i].ordinal;
            buf.put_i16(inout_type as i16);

            let type_id = types[i].r#type as i32;
            let (arg, type_id) = match &args[i] {
                XuguArgumentValue::Null => ([].as_slice(), ColumnType::NULL as i32),
                XuguArgumentValue::Str(x) => (x.as_bytes(), type_id),
                XuguArgumentValue::Bin(x) => (x.as_ref(), type_id),
            };
            buf.put_i16(type_id as i16);

            buf.put_i32(arg.len() as i32);
            buf.put_slice(arg);
        }
    }
}

impl ProtocolEncode<'_, ()> for Execute<'_, '_> {
    fn encode_with(&self, buf: &mut Vec<u8>, _: ()) -> Result<(), Error> {
        let sql_cmd = format!("? {}", self.st_id);

        encode_sql_command(buf, &sql_cmd);
        self.encode_params(buf);
        Ok(())
    }
}