Mirrors/sqlx
2
0
Fork 0
mirror of https://github.com/launchbadge/sqlx synced 2024-11-10 14:34:19 +00:00
Code Issues Projects Releases Packages Wiki Activity

postgres: support null in a record

Browse source
This commit is contained in:
Ryan Leckey 2020-03-17 19:26:15 -07:00
parent 100602187f
commit c1e6b2045c
1 changed files with 10 additions and 29 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

39
sqlx-core/src/postgres/types/record.rs
View file

@ -1,5 +1,5 @@
use crate::decode::Decode;
use crate::encode::Encode;
use crate::encode::{Encode, IsNull};
use crate::io::Buf;
use crate::postgres::protocol::TypeId;
use crate::postgres::{PgTypeInfo, PgValue, Postgres};
@ -42,13 +42,16 @@ impl<'a> PgRecordEncoder<'a> {
self.buf.extend(&[0; 4]);
let start = self.buf.len();
value.encode(self.buf);
if let IsNull::Yes = value.encode_nullable(self.buf) {
// replaces zeros with actual length
self.buf[start - 4..start].copy_from_slice(&(-1_i32).to_be_bytes());
} else {
let end = self.buf.len();
let size = end - start;
let end = self.buf.len();
let size = end - start;
// replaces zeros with actual length
self.buf[start - 4..start].copy_from_slice(&(size as u32).to_be_bytes());
// replaces zeros with actual length
self.buf[start - 4..start].copy_from_slice(&(size as u32).to_be_bytes());
}
// keep track of count
self.num += 1;
@ -57,28 +60,6 @@ impl<'a> PgRecordEncoder<'a> {
}
}
// impl Encode<Postgres> for (bool, i32, i64, f64, String) {
// fn encode(&self, buf: &mut Vec<u8>) {
// PgRecordEncoder::new(buf)
// .encode(self.0)
// .encode(self.1)
// .encode(self.2)
// .encode(self.3)
// .encode(&self.4)
// .finish()
// }
//
// fn size_hint(&self) -> usize {
// // for each field; oid, length, value
// 5 * (4 + 4)
// + (<bool as Encode<Postgres>>::size_hint(&self.0)
// + <i32 as Encode<Postgres>>::size_hint(&self.1)
// + <i64 as Encode<Postgres>>::size_hint(&self.2)
// + <f64 as Encode<Postgres>>::size_hint(&self.3)
// + <String as Encode<Postgres>>::size_hint(&self.4))
// }
// }
pub struct PgRecordDecoder<'de> {
value: PgValue<'de>,
}