rbdc_mysql/protocol/text/
row.rs

1use bytes::{Buf, Bytes};
2
3use crate::io::MySqlBufExt;
4use crate::protocol::Row;
5use crate::result_set::MySqlColumn;
6use rbdc::io::Decode;
7use rbdc::Error;
8
9#[derive(Debug)]
10pub struct TextRow(pub Row);
11
12impl<'de> Decode<'de, &'de [MySqlColumn]> for TextRow {
13    fn decode_with(mut buf: Bytes, columns: &'de [MySqlColumn]) -> Result<Self, Error> {
14        let storage = buf.clone();
15        let offset = buf.len();
16
17        let mut values = Vec::with_capacity(columns.len());
18
19        for _ in columns {
20            if buf[0] == 0xfb {
21                // NULL is sent as 0xfb
22                values.push(None);
23                buf.advance(1);
24            } else {
25                let size = buf.get_uint_lenenc() as usize;
26                let offset = offset - buf.len();
27
28                // bounds check to avoid panic when server returns non-MySQL-compliant payloads (e.g., ClickHouse MySQL compatibility layer)
29                if size > buf.len() {
30                    return Err(rbdc::Error::protocol(format!(
31                        "text row column length {} exceeds remaining {} bytes (first byte: 0x{:02x})",
32                        size,
33                        buf.len(),
34                        storage[offset] // original first byte of this value in storage snapshot
35                    )));
36                }
37
38                values.push(Some(offset..(offset + size)));
39
40                buf.advance(size);
41            }
42        }
43        Ok(TextRow(Row::from((values, storage.to_vec()))))
44    }
45}