postgres_types/
bit_vec_06.rs

1use bit_vec_06::BitVec;
2use bytes::BytesMut;
3use postgres_protocol::types;
4use std::error::Error;
5
6use crate::{FromSql, IsNull, ToSql, Type};
7
8impl<'a> FromSql<'a> for BitVec {
9    fn from_sql(_: &Type, raw: &[u8]) -> Result<BitVec, Box<dyn Error + Sync + Send>> {
10        let varbit = types::varbit_from_sql(raw)?;
11        let mut bitvec = BitVec::from_bytes(varbit.bytes());
12        while bitvec.len() > varbit.len() {
13            bitvec.pop();
14        }
15
16        Ok(bitvec)
17    }
18
19    accepts!(BIT, VARBIT);
20}
21
22impl ToSql for BitVec {
23    fn to_sql(&self, _: &Type, out: &mut BytesMut) -> Result<IsNull, Box<dyn Error + Sync + Send>> {
24        types::varbit_to_sql(self.len(), self.to_bytes().into_iter(), out)?;
25        Ok(IsNull::No)
26    }
27
28    accepts!(BIT, VARBIT);
29    to_sql_checked!();
30}