postgres_types/
serde_json_1.rs

1use crate::{FromSql, IsNull, ToSql, Type};
2use bytes::{BufMut, BytesMut};
3use serde_1::{Deserialize, Deserializer, Serialize, Serializer};
4use serde_json_1::Value;
5use std::error::Error;
6use std::fmt::Debug;
7use std::io::Read;
8
9/// A wrapper type to allow arbitrary `Serialize`/`Deserialize` types to convert to Postgres JSON values.
10#[derive(Clone, Default, Debug, PartialEq, Eq)]
11pub struct Json<T>(pub T);
12
13impl<T: Serialize> Serialize for Json<T> {
14    fn serialize<S: Serializer>(&self, serializer: S) -> Result<S::Ok, S::Error> {
15        self.0.serialize(serializer)
16    }
17}
18
19impl<'de, T: Deserialize<'de>> Deserialize<'de> for Json<T> {
20    fn deserialize<D: Deserializer<'de>>(deserializer: D) -> Result<Self, D::Error> {
21        T::deserialize(deserializer).map(Self)
22    }
23}
24
25impl<'a, T> FromSql<'a> for Json<T>
26where
27    T: Deserialize<'a>,
28{
29    fn from_sql(ty: &Type, mut raw: &'a [u8]) -> Result<Json<T>, Box<dyn Error + Sync + Send>> {
30        if *ty == Type::JSONB {
31            let mut b = [0; 1];
32            raw.read_exact(&mut b)?;
33            // We only support version 1 of the jsonb binary format
34            if b[0] != 1 {
35                return Err("unsupported JSONB encoding version".into());
36            }
37        }
38        serde_json_1::de::from_slice(raw)
39            .map(Json)
40            .map_err(Into::into)
41    }
42
43    accepts!(JSON, JSONB);
44}
45
46impl<T> ToSql for Json<T>
47where
48    T: Serialize + Debug,
49{
50    fn to_sql(
51        &self,
52        ty: &Type,
53        out: &mut BytesMut,
54    ) -> Result<IsNull, Box<dyn Error + Sync + Send>> {
55        if *ty == Type::JSONB {
56            out.put_u8(1);
57        }
58        serde_json_1::ser::to_writer(out.writer(), &self.0)?;
59        Ok(IsNull::No)
60    }
61
62    accepts!(JSON, JSONB);
63    to_sql_checked!();
64}
65
66impl<'a> FromSql<'a> for Value {
67    fn from_sql(ty: &Type, raw: &[u8]) -> Result<Value, Box<dyn Error + Sync + Send>> {
68        Json::<Value>::from_sql(ty, raw).map(|json| json.0)
69    }
70
71    accepts!(JSON, JSONB);
72}
73
74impl ToSql for Value {
75    fn to_sql(
76        &self,
77        ty: &Type,
78        out: &mut BytesMut,
79    ) -> Result<IsNull, Box<dyn Error + Sync + Send>> {
80        Json(self).to_sql(ty, out)
81    }
82
83    accepts!(JSON, JSONB);
84    to_sql_checked!();
85}
postgres_types/serde_json_1.rs

postgres_types/
serde_json_1.rs
