tarantool-rs 0.0.10

Asyncronous tokio-based client for Tarantool
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
57
58
59
60

use std::io::Write;

use crate::{
    codec::{
        consts::{keys, RequestType},
        utils::{write_kv_str, write_kv_tuple, write_kv_u64},
    },
    errors::EncodingError,
    tuple::Tuple,
};

use super::Request;

#[derive(Clone, Debug)]
enum ExecuteStatment<'a> {
    StatementId(u64),
    Query(&'a str),
}

#[derive(Clone, Debug)]
pub(crate) struct Execute<'a, T> {
    statement: ExecuteStatment<'a>,
    binds: T,
}

impl<'a, T: Tuple> Request for Execute<'a, T> {
    fn request_type() -> RequestType
    where
        Self: Sized,
    {
        RequestType::Execute
    }

    // NOTE: `&mut buf: mut` is required since I don't get why compiler complain
    fn encode(&self, mut buf: &mut dyn Write) -> Result<(), EncodingError> {
        rmp::encode::write_map_len(&mut buf, 2)?;
        match self.statement {
            ExecuteStatment::StatementId(x) => write_kv_u64(buf, keys::SQL_STMT_ID, x)?,
            ExecuteStatment::Query(x) => write_kv_str(buf, keys::SQL_TEXT, x)?,
        }
        write_kv_tuple(buf, keys::SQL_BIND, &self.binds)?;
        Ok(())
    }
}

impl<'a, T> Execute<'a, T> {
    pub(crate) fn new_statement_id(statment_id: u64, binds: T) -> Self {
        Self {
            statement: ExecuteStatment::StatementId(statment_id),
            binds,
        }
    }

    pub(crate) fn new_query(query: &'a str, binds: T) -> Self {
        Self {
            statement: ExecuteStatment::Query(query),
            binds,
        }
    }
}