wtx 0.44.1

A collection of different transport implementations and related tools focused primarily on web technologies.
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
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104

use crate::{
  collection::{TryExtend, Vector},
  database::{
    Identifier,
    client::postgres::{
      PostgresRecord, PostgresRecords, PostgresStatement, PostgresStatements, Ty,
      msg_field::MsgField, postgres_column_info::PostgresColumnInfo,
    },
  },
  misc::{Either, net::PartitionedFilledBuffer},
};
use core::ops::Range;

pub(crate) fn data_row<E>(
  begin: usize,
  begin_data: usize,
  net_buffer: &mut PartitionedFilledBuffer,
  records_params: &mut Vector<(Range<usize>, Range<usize>)>,
  stmt: PostgresStatement<'_>,
  values_len: u16,
  values_params: &mut Vector<(bool, Range<usize>)>,
  values_params_offset: usize,
  cb: &mut impl FnMut(PostgresRecord<'_, E>) -> Result<(), E>,
) -> Result<(), E>
where
  E: From<crate::Error>,
{
  let net_buffer_range = begin_data..net_buffer.current_end_idx();
  let mut bytes = net_buffer.all().get(net_buffer_range).unwrap_or_default();
  let record_range_begin = net_buffer.antecedent_end_idx().wrapping_sub(begin);
  let record_range_end = net_buffer.current_end_idx().wrapping_sub(begin_data);
  bytes = bytes.get(record_range_begin..record_range_end).unwrap_or_default();
  let values_params_begin = values_params.len().wrapping_sub(values_params_offset);
  cb(PostgresRecord::parse(bytes, stmt, values_len, values_params)?)?;
  records_params.push((
    record_range_begin..record_range_end,
    values_params_begin..values_params.len().wrapping_sub(values_params_offset),
  ))?;
  Ok(())
}

pub(crate) const fn dummy_stmt_value() -> (PostgresColumnInfo, Ty) {
  (PostgresColumnInfo::new(Identifier::new(), Ty::Any), Ty::Any)
}

pub(crate) fn extend_records<'exec, B, E>(
  begin_data: usize,
  buffer: &mut B,
  net_buffer: &'exec mut PartitionedFilledBuffer,
  records_params: &'exec mut Vector<(Range<usize>, Range<usize>)>,
  stmts: &'exec mut PostgresStatements,
  stmts_identifiers: impl IntoIterator<Item = Either<usize, u64>>,
  values_params: &'exec mut Vector<(bool, Range<usize>)>,
) -> crate::Result<()>
where
  B: TryExtend<[PostgresRecords<'exec, E>; 1]>,
{
  if B::IS_UNIT {
    return Ok(());
  }
  let mut rows_idx: usize = 0;
  let mut values_idx: usize = 0;
  for identifier in stmts_identifiers {
    let opt = match identifier {
      Either::Left(idx) => stmts.get_by_idx(idx),
      Either::Right(id) => stmts.get_by_stmt_cmd_id(id),
    };
    let Some(stmt) = opt else {
      return Err(crate::Error::ProgrammingError);
    };
    let local_rows_idx = rows_idx.wrapping_add(stmt.rows_len);
    let local_values_idx = values_idx.wrapping_add(stmt.columns_len.wrapping_mul(stmt.rows_len));
    let local_rp = records_params.get(rows_idx..local_rows_idx).unwrap_or_default();
    let local_vp = values_params.get(values_idx..local_values_idx).unwrap_or_default();
    rows_idx = local_rows_idx;
    values_idx = local_values_idx;
    buffer.try_extend([PostgresRecords::new(
      net_buffer.all().get(begin_data..net_buffer.current_end_idx()).unwrap_or_default(),
      local_rp,
      stmt,
      local_vp,
    )])?;
  }
  Ok(())
}

pub(crate) fn row_description(
  columns_len: u16,
  rd: &mut &[u8],
  mut cb: impl FnMut(u16, PostgresColumnInfo) -> crate::Result<()>,
) -> crate::Result<()> {
  for idx in 0..columns_len {
    let (read, msg_field) = MsgField::parse(rd)?;
    let ty = Ty::Custom(msg_field.type_oid);
    let pci = PostgresColumnInfo::new(msg_field.name.try_into()?, ty);
    cb(idx, pci)?;
    if let Some(elem @ [_not_empty, ..]) = rd.get(read..) {
      *rd = elem;
    } else {
      break;
    }
  }
  Ok(())
}