Line data    Source code

       1              : use crate::{FromSql, Type};
       2              : pub use bytes::BytesMut;
       3              : use std::error::Error;
       4              : 
       5            0 : pub fn read_be_i32(buf: &mut &[u8]) -> Result<i32, Box<dyn Error + Sync + Send>> {
       6            0 :     if buf.len() < 4 {
       7            0 :         return Err("invalid buffer size".into());
       8            0 :     }
       9            0 :     let mut bytes = [0; 4];
      10            0 :     bytes.copy_from_slice(&buf[..4]);
      11            0 :     *buf = &buf[4..];
      12            0 :     Ok(i32::from_be_bytes(bytes))
      13            0 : }
      14              : 
      15            0 : pub fn read_value<'a, T>(
      16            0 :     type_: &Type,
      17            0 :     buf: &mut &'a [u8],
      18            0 : ) -> Result<T, Box<dyn Error + Sync + Send>>
      19            0 : where
      20            0 :     T: FromSql<'a>,
      21            0 : {
      22            0 :     let len = read_be_i32(buf)?;
      23            0 :     let value = if len < 0 {
      24            0 :         None
      25              :     } else {
      26            0 :         if len as usize > buf.len() {
      27            0 :             return Err("invalid buffer size".into());
      28            0 :         }
      29            0 :         let (head, tail) = buf.split_at(len as usize);
      30            0 :         *buf = tail;
      31            0 :         Some(head)
      32              :     };
      33            0 :     T::from_sql_nullable(type_, value)
      34            0 : }