Skip to content

Commit

Permalink
feat(serde_with): make Bytes adjustable
Browse files Browse the repository at this point in the history
  • Loading branch information
@sivizius
sivizius committed Oct 10, 2024
1 parent e030aa0 commit 532f190
Show file tree
Hide file tree
Showing 3 changed files with 80 additions and 30 deletions.
50 changes: 37 additions & 13 deletions serde_with/src/de/impls.rs
View file
Original file line number Diff line number Diff line change
Expand Up @@ -1188,7 +1188,10 @@ where
}
}

impl<'de> DeserializeAs<'de, &'de [u8]> for Bytes {
impl<'de, PREFERENCE> DeserializeAs<'de, &'de [u8]> for Bytes<PREFERENCE>
where
PREFERENCE: formats::TypePreference,
{
fn deserialize_as<D>(deserializer: D) -> Result<&'de [u8], D::Error>
where
D: Deserializer<'de>,
Expand All @@ -1207,7 +1210,10 @@ impl<'de> DeserializeAs<'de, &'de [u8]> for Bytes {
// * visit_str
// * visit_string
#[cfg(feature = "alloc")]
impl<'de> DeserializeAs<'de, Vec<u8>> for Bytes {
impl<'de, PREFERENCE> DeserializeAs<'de, Vec<u8>> for Bytes<PREFERENCE>
where
PREFERENCE: formats::TypePreference,
{
fn deserialize_as<D>(deserializer: D) -> Result<Vec<u8>, D::Error>
where
D: Deserializer<'de>,
Expand Down Expand Up @@ -1262,12 +1268,15 @@ impl<'de> DeserializeAs<'de, Vec<u8>> for Bytes {
}

#[cfg(feature = "alloc")]
impl<'de> DeserializeAs<'de, Box<[u8]>> for Bytes {
impl<'de, PREFERENCE> DeserializeAs<'de, Box<[u8]>> for Bytes<PREFERENCE>
where
PREFERENCE: formats::TypePreference,
{
fn deserialize_as<D>(deserializer: D) -> Result<Box<[u8]>, D::Error>
where
D: Deserializer<'de>,
{
<Bytes as DeserializeAs<'de, Vec<u8>>>::deserialize_as(deserializer)
<Bytes<PREFERENCE> as DeserializeAs<'de, Vec<u8>>>::deserialize_as(deserializer)
.map(Vec::into_boxed_slice)
}
}
Expand All @@ -1284,7 +1293,10 @@ impl<'de> DeserializeAs<'de, Box<[u8]>> for Bytes {
// * visit_string
// * visit_seq
#[cfg(feature = "alloc")]
impl<'de> DeserializeAs<'de, Cow<'de, [u8]>> for Bytes {
impl<'de, PREFERENCE> DeserializeAs<'de, Cow<'de, [u8]>> for Bytes<PREFERENCE>
where
PREFERENCE: formats::TypePreference,
{
fn deserialize_as<D>(deserializer: D) -> Result<Cow<'de, [u8]>, D::Error>
where
D: Deserializer<'de>,
Expand Down Expand Up @@ -1354,7 +1366,10 @@ impl<'de> DeserializeAs<'de, Cow<'de, [u8]>> for Bytes {
}
}

impl<'de, const N: usize> DeserializeAs<'de, [u8; N]> for Bytes {
impl<'de, const N: usize, PREFERENCE> DeserializeAs<'de, [u8; N]> for Bytes<PREFERENCE>
where
PREFERENCE: formats::TypePreference,
{
fn deserialize_as<D>(deserializer: D) -> Result<[u8; N], D::Error>
where
D: Deserializer<'de>,
Expand Down Expand Up @@ -1397,7 +1412,10 @@ impl<'de, const N: usize> DeserializeAs<'de, [u8; N]> for Bytes {
}
}

impl<'de, const N: usize> DeserializeAs<'de, &'de [u8; N]> for Bytes {
impl<'de, const N: usize, PREFERENCE> DeserializeAs<'de, &'de [u8; N]> for Bytes<PREFERENCE>
where
PREFERENCE: formats::TypePreference,
{
fn deserialize_as<D>(deserializer: D) -> Result<&'de [u8; N], D::Error>
where
D: Deserializer<'de>,
Expand Down Expand Up @@ -1434,7 +1452,10 @@ impl<'de, const N: usize> DeserializeAs<'de, &'de [u8; N]> for Bytes {
}

#[cfg(feature = "alloc")]
impl<'de, const N: usize> DeserializeAs<'de, Cow<'de, [u8; N]>> for Bytes {
impl<'de, const N: usize, PREFERENCE> DeserializeAs<'de, Cow<'de, [u8; N]>> for Bytes<PREFERENCE>
where
PREFERENCE: formats::TypePreference,
{
fn deserialize_as<D>(deserializer: D) -> Result<Cow<'de, [u8; N]>, D::Error>
where
D: Deserializer<'de>,
Expand Down Expand Up @@ -1531,12 +1552,15 @@ impl<'de, const N: usize> DeserializeAs<'de, Cow<'de, [u8; N]>> for Bytes {
}

#[cfg(feature = "alloc")]
impl<'de, const N: usize> DeserializeAs<'de, Box<[u8; N]>> for Bytes {
impl<'de, const N: usize, PREFERENCE> DeserializeAs<'de, Box<[u8; N]>> for Bytes<PREFERENCE>
where
PREFERENCE: formats::TypePreference,
{
fn deserialize_as<D>(deserializer: D) -> Result<Box<[u8; N]>, D::Error>
where
D: Deserializer<'de>,
{
Bytes::deserialize_as(deserializer).map(Box::new)
<Bytes<PREFERENCE>>::deserialize_as(deserializer).map(Box::new)
}
}

Expand Down Expand Up @@ -1807,17 +1831,17 @@ impl<'de> DeserializeAs<'de, Cow<'de, [u8]>> for BorrowCow {
where
D: Deserializer<'de>,
{
Bytes::deserialize_as(deserializer)
<Bytes<formats::PreferBytes>>::deserialize_as(deserializer)
}
}

Check warning on line 1836 in serde_with/src/de/impls.rs

View workflow job for this annotation

GitHub Actions / Rustfmt (stable) 

Diff in /home/runner/work/serde_with/serde_with/serde_with/src/de/impls.rs

#[cfg(feature = "alloc")]
impl<'de, const N: usize> DeserializeAs<'de, Cow<'de, [u8; N]>> for BorrowCow {
impl<'de, const N: usize,> DeserializeAs<'de, Cow<'de, [u8; N]>> for BorrowCow {
fn deserialize_as<D>(deserializer: D) -> Result<Cow<'de, [u8; N]>, D::Error>
where
D: Deserializer<'de>,
{
Bytes::deserialize_as(deserializer)
<Bytes<formats::PreferBytes>>::deserialize_as(deserializer)
}
}

Expand Down
4 changes: 3 additions & 1 deletion serde_with/src/lib.rs
View file
Original file line number Diff line number Diff line change
Expand Up @@ -1647,7 +1647,9 @@ pub struct TimestampNanoSecondsWithFrac<
/// # ), serde_json::to_value(&test).unwrap());
/// # }
/// ```
pub struct Bytes;
pub struct Bytes<PREFERENCE: formats::TypePreference = formats::PreferBytes>(
PhantomData<PREFERENCE>,
);

/// Deserialize one or many elements
///
Expand Down
56 changes: 40 additions & 16 deletions serde_with/src/ser/impls.rs
View file
Original file line number Diff line number Diff line change
Expand Up @@ -772,80 +772,104 @@ where
}
}

impl SerializeAs<&[u8]> for Bytes {
impl<PREFERENCE> SerializeAs<&[u8]> for Bytes<PREFERENCE>
where
PREFERENCE: formats::TypePreference,
{
fn serialize_as<S>(bytes: &&[u8], serializer: S) -> Result<S::Ok, S::Error>
where
S: Serializer,
{
serializer.serialize_bytes(bytes)
PREFERENCE::serialize_as(*bytes, serializer)
}
}

#[cfg(feature = "alloc")]
impl SerializeAs<Vec<u8>> for Bytes {
impl<PREFERENCE> SerializeAs<Vec<u8>> for Bytes<PREFERENCE>
where
PREFERENCE: formats::TypePreference,
{
fn serialize_as<S>(bytes: &Vec<u8>, serializer: S) -> Result<S::Ok, S::Error>
where
S: Serializer,
{
serializer.serialize_bytes(bytes)
PREFERENCE::serialize_as(bytes.as_slice(), serializer)
}
}

#[cfg(feature = "alloc")]
impl SerializeAs<Box<[u8]>> for Bytes {
impl<PREFERENCE> SerializeAs<Box<[u8]>> for Bytes<PREFERENCE>
where
PREFERENCE: formats::TypePreference,
{
fn serialize_as<S>(bytes: &Box<[u8]>, serializer: S) -> Result<S::Ok, S::Error>
where
S: Serializer,
{
serializer.serialize_bytes(bytes)
PREFERENCE::serialize_as(bytes.as_ref(), serializer)
}
}

#[cfg(feature = "alloc")]
impl<'a> SerializeAs<Cow<'a, [u8]>> for Bytes {
impl<'a, PREFERENCE> SerializeAs<Cow<'a, [u8]>> for Bytes<PREFERENCE>
where
PREFERENCE: formats::TypePreference,
{
fn serialize_as<S>(bytes: &Cow<'a, [u8]>, serializer: S) -> Result<S::Ok, S::Error>
where
S: Serializer,
{
serializer.serialize_bytes(bytes)
PREFERENCE::serialize_as(bytes.as_ref(), serializer)
}
}

impl<const N: usize> SerializeAs<[u8; N]> for Bytes {
impl<const N: usize, PREFERENCE> SerializeAs<[u8; N]> for Bytes<PREFERENCE>
where
PREFERENCE: formats::TypePreference,
{
fn serialize_as<S>(bytes: &[u8; N], serializer: S) -> Result<S::Ok, S::Error>
where
S: Serializer,
{
serializer.serialize_bytes(bytes)
PREFERENCE::serialize_as(bytes.as_ref(), serializer)
}
}

impl<const N: usize> SerializeAs<&[u8; N]> for Bytes {
impl<const N: usize, PREFERENCE> SerializeAs<&[u8; N]> for Bytes<PREFERENCE>
where
PREFERENCE: formats::TypePreference,
{
fn serialize_as<S>(bytes: &&[u8; N], serializer: S) -> Result<S::Ok, S::Error>
where
S: Serializer,
{
serializer.serialize_bytes(*bytes)
PREFERENCE::serialize_as(bytes.as_ref(), serializer)
}
}

#[cfg(feature = "alloc")]
impl<const N: usize> SerializeAs<Box<[u8; N]>> for Bytes {
impl<const N: usize, PREFERENCE> SerializeAs<Box<[u8; N]>> for Bytes<PREFERENCE>
where
PREFERENCE: formats::TypePreference,
{
fn serialize_as<S>(bytes: &Box<[u8; N]>, serializer: S) -> Result<S::Ok, S::Error>
where
S: Serializer,
{
serializer.serialize_bytes(&**bytes)
PREFERENCE::serialize_as(bytes.as_ref(), serializer)
}
}

#[cfg(feature = "alloc")]
impl<'a, const N: usize> SerializeAs<Cow<'a, [u8; N]>> for Bytes {
impl<'a, const N: usize, PREFERENCE> SerializeAs<Cow<'a, [u8; N]>> for Bytes<PREFERENCE>
where
PREFERENCE: formats::TypePreference,
{
fn serialize_as<S>(bytes: &Cow<'a, [u8; N]>, serializer: S) -> Result<S::Ok, S::Error>
where
S: Serializer,
{
serializer.serialize_bytes(bytes.as_ref())
PREFERENCE::serialize_as(bytes.as_ref(), serializer)
}
}

Expand Down

0 comments on commit 532f190

Please sign in to comment.