pgwire 0.38.3

Postgresql wire protocol implemented as a library
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

use bytes::Buf;
use postgres_types::FromSqlOwned;

use crate::api::results::{FieldFormat, FieldInfo};
use crate::error::{PgWireClientError, PgWireClientResult};
use crate::messages::data::DataRow;
use crate::types::FromSqlText;
use crate::types::format::FormatOptions;

#[derive(new, Debug)]
pub struct DataRowsReader {
    fields: Vec<FieldInfo>,
    rows: Vec<DataRow>,
}

impl DataRowsReader {
    pub fn empty() -> DataRowsReader {
        Self {
            fields: vec![],
            rows: vec![],
        }
    }

    /// Generate row decoder for next row
    pub fn next_row(&mut self) -> Option<DataRowDecoder<'_>> {
        if !self.rows.is_empty() {
            let row = self.rows.remove(0);
            Some(DataRowDecoder::new(self.fields.as_slice(), row))
        } else {
            None
        }
    }
}

#[derive(new, Debug)]
pub struct DataRowDecoder<'a> {
    fields: &'a [FieldInfo],
    row: DataRow,
    #[new(default)]
    read_index: usize,
}

impl DataRowDecoder<'_> {
    /// Get value from data row
    pub fn next_value<T>(&mut self) -> PgWireClientResult<Option<T>>
    where
        T: FromSqlOwned + for<'a> FromSqlText<'a>,
    {
        if let Some(field_info) = self.fields.get(self.read_index) {
            // advance read index
            self.read_index += 1;

            let byte_len = self.row.data.get_i32();
            if byte_len < 0 {
                Ok(None)
            } else {
                let bytes = self.row.data.split_to(byte_len as usize);

                if field_info.format() == FieldFormat::Text {
                    T::from_sql_text(
                        field_info.datatype(),
                        bytes.as_ref(),
                        &FormatOptions::default(),
                    )
                    .map_err(PgWireClientError::FromSqlError)
                    .map(Some)
                } else {
                    // binary
                    T::from_sql(field_info.datatype(), bytes.as_ref())
                        .map_err(PgWireClientError::FromSqlError)
                        .map(Some)
                }
            }
        } else {
            Err(PgWireClientError::DataRowIndexOutOfBounds)
        }
    }

    /// Length of fields
    pub fn len(&self) -> usize {
        self.fields.len()
    }

    pub fn is_empty(&self) -> bool {
        self.fields.is_empty()
    }
}

#[cfg(test)]
mod tests {}