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 bytebuffer::ByteBuffer;

use crate::encode::Value;

#[allow(dead_code)]
pub(crate) struct ProcedureInvocation<'a> {
    handle: i64,
    is_query: bool,
    params: &'a Vec<&'a dyn Value>,
    query: String,
    pub(crate) slen: i32, // length of pi once serialized
}

pub(crate) fn new_procedure_invocation<'a>(handle: i64, is_query: bool, params: &'a Vec<&dyn Value>, query: &str) -> ProcedureInvocation<'a> {
    ProcedureInvocation {
        handle,
        is_query,
        params,
        query: query.to_string(),
        slen: -1,
    }
}

impl<'a> ProcedureInvocation<'a> {
    fn calculate_length(&mut self) -> i32 {
        if self.slen != -1 {
            return self.slen;
        }
        let mut slen: i32 = 15;
        slen = slen + self.query.len() as i32;
        for item in self.params {
            slen = slen + item.get_write_length()
        }
        self.slen = slen as i32;
        return slen as i32;
    }

    pub(crate) fn bytes(&mut self) -> Vec<u8> {
        let length = self.calculate_length();
        let mut buffer = ByteBuffer::new();
        buffer.write_i32(length);
        buffer.write_u8(0);
        buffer.write_string(self.query.as_str());
        buffer.write_i64(self.handle);
        buffer.write_i16(self.params.len() as i16);
        for item in self.params.iter() {
            item.marshal(&mut buffer)
        }
        return buffer.to_bytes();
    }
}