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
use crate::{
decode::Decode,
encode::{Encode, IsNull},
error::BoxDynError,
postgres::{PgArgumentBuffer, PgHasArrayType, PgTypeInfo, PgValueFormat, PgValueRef, Postgres},
types::Type,
};
use bit_vec::BitVec;
use bytes::Buf;
use std::{io, mem};
impl Type<Postgres> for BitVec {
fn type_info() -> PgTypeInfo {
PgTypeInfo::VARBIT
}
fn compatible(ty: &PgTypeInfo) -> bool {
*ty == PgTypeInfo::BIT || *ty == PgTypeInfo::VARBIT
}
}
impl PgHasArrayType for BitVec {
fn array_type_info() -> PgTypeInfo {
PgTypeInfo::VARBIT_ARRAY
}
fn array_compatible(ty: &PgTypeInfo) -> bool {
*ty == PgTypeInfo::BIT_ARRAY || *ty == PgTypeInfo::VARBIT_ARRAY
}
}
impl Encode<'_, Postgres> for BitVec {
fn encode_by_ref(&self, buf: &mut PgArgumentBuffer) -> IsNull {
buf.extend(&(self.len() as i32).to_be_bytes());
buf.extend(self.to_bytes());
IsNull::No
}
fn size_hint(&self) -> usize {
mem::size_of::<i32>() + self.len()
}
}
impl Decode<'_, Postgres> for BitVec {
fn decode(value: PgValueRef<'_>) -> Result<Self, BoxDynError> {
match value.format() {
PgValueFormat::Binary => {
let mut bytes = value.as_bytes()?;
let len = bytes.get_i32();
if len < 0 {
Err(io::Error::new(
io::ErrorKind::InvalidData,
"Negative VARBIT length.",
))?
}
let bytes_len = (len as usize + 7) / 8;
if bytes.remaining() != bytes_len {
Err(io::Error::new(
io::ErrorKind::InvalidData,
"VARBIT length mismatch.",
))?;
}
let mut bitvec = BitVec::from_bytes(&bytes);
while bitvec.len() > len as usize {
bitvec.pop();
}
Ok(bitvec)
}
PgValueFormat::Text => {
let s = value.as_str()?;
let mut bit_vec = BitVec::with_capacity(s.len());
for c in s.chars() {
match c {
'0' => bit_vec.push(false),
'1' => bit_vec.push(true),
_ => {
Err(io::Error::new(
io::ErrorKind::InvalidData,
"VARBIT data contains other characters than 1 or 0.",
))?;
}
}
}
Ok(bit_vec)
}
}
}
}