dnim/rust-postgres
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add FromSql / ToSql for smol_str::SmolStr.

Browse Source
This commit is contained in:
BratSinot 2022-07-04 12:25:58 +02:00
parent 0984ab1c58
commit 7699c78037
2 changed files with 28 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
postgres-types/Cargo.toml
View File

@ -48,3 +48,5 @@ uuid-08 = { version = "0.8", package = "uuid", optional = true }
uuid-1 = { version = "1.0", package = "uuid", optional = true }
time-02 = { version = "0.2", package = "time", optional = true }
time-03 = { version = "0.3", package = "time", default-features = false, optional = true }
smol_str = { version = "0.1.23", default-features = false, optional = true }

26
postgres-types/src/lib.rs
View File

@ -155,6 +155,7 @@ pub use pg_lsn::PgLsn;
pub use crate::special::{Date, Timestamp};
use bytes::BytesMut;
use smol_str::SmolStr;
// Number of seconds from 1970-01-01 to 2000-01-01
const TIME_SEC_CONVERSION: u64 = 946_684_800;
@ -629,6 +630,18 @@ impl<'a> FromSql<'a> for Box<str> {
}
}
#[cfg(feature = "smol_str")]
impl<'a> FromSql<'a> for smol_str::SmolStr {
fn from_sql(ty: &Type, raw: &'a [u8]) -> Result<SmolStr, Box<dyn Error + Sync + Send>> {
<&str as FromSql>::from_sql(ty, raw)
.map(SmolStr::from)
}
fn accepts(ty: &Type) -> bool {
<&str as FromSql>::accepts(ty)
}
}
impl<'a> FromSql<'a> for &'a str {
fn from_sql(ty: &Type, raw: &'a [u8]) -> Result<&'a str, Box<dyn Error + Sync + Send>> {
match *ty {
@ -1029,6 +1042,19 @@ impl ToSql for Box<str> {
to_sql_checked!();
}
#[cfg(feature = "smol_str")]
impl ToSql for smol_str::SmolStr {
fn to_sql(&self, ty: &Type, w: &mut BytesMut) -> Result<IsNull, Box<dyn Error + Sync + Send>> {
<&str as ToSql>::to_sql(&&**self, ty, w)
}
fn accepts(ty: &Type) -> bool {
<&str as ToSql>::accepts(ty)
}
to_sql_checked!();
}
macro_rules! simple_to {
($t:ty, $f:ident, $($expected:ident),+) => {
impl ToSql for $t {