wtx 0.28.0

A collection of different transport implementations and related tools focused primarily on web technologies.
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
61

use crate::{
  database::{
    RecordValues,
    client::mysql::{
      EncodeWrapper, Mysql, MysqlStatement,
      flag::Flag,
      mysql_protocol::{MysqlProtocol, encode_wrapper_protocol::EncodeWrapperProtocol},
    },
  },
  misc::Encode,
};

#[derive(Debug)]
pub(crate) struct StmtExecuteReq<'any, 'stmts, RV> {
  pub(crate) rv: RV,
  pub(crate) stmt: &'any MysqlStatement<'stmts>,
}

impl<DO, E, RV> Encode<MysqlProtocol<DO, E>> for StmtExecuteReq<'_, '_, RV>
where
  E: From<crate::Error>,
  RV: RecordValues<Mysql<E>>,
{
  #[inline]
  fn encode(&self, _: &mut (), ew: &mut EncodeWrapperProtocol<'_>) -> Result<(), E> {
    let _ = ew.encode_buffer.extend_from_copyable_slices([
      &[23][..],
      &self.stmt.aux.to_le_bytes(),
      &[0, 1, 0, 0, 0],
    ])?;
    if self.stmt.tys_len > 0 {
      let prev_len = ew.encode_buffer.len();
      let mut idx: usize = 0;
      self.rv.walk(|is_null, _| {
        let bit_offset = idx % 8;
        let byte_index = prev_len.wrapping_add(idx / 8);
        if bit_offset == 0 {
          ew.encode_buffer.push(0)?;
        }
        idx = idx.wrapping_add(1);
        if let Some(elem) = ew.encode_buffer.get_mut(byte_index) {
          *elem |= u8::from(is_null) << bit_offset;
        }
        Ok(())
      })?;
      ew.encode_buffer.push(1)?;
      for ty in self.stmt._tys() {
        let unsigned = u16::from(Flag::Unsigned);
        let value = if ty.flags & unsigned == unsigned { 128 } else { 0 };
        ew.encode_buffer.extend_from_copyable_slice(&[u8::from(ty.ty), value])?;
      }
      let _ = self.rv.encode_values(
        &mut (),
        &mut EncodeWrapper::new(ew.encode_buffer),
        |_, _| 0,
        |_, _, _, _| 0,
      )?;
    }
    Ok(())
  }
}