Skip to content

Enhancement: Optimized Handling of Raw Vectors #871

@Y-jiji

Description

@Y-jiji

I wonder whether serde support for byte-castable vectors can be implemented as a built-in tool in serde_with.
For loading large files containing binary data, I implemented the following to avoid expensive sequence serialization/deserialization:

use std::{marker::PhantomData, mem::MaybeUninit};
use serde::de::Visitor;
use bytemuck::{AnyBitPattern, NoUninit};
use serde_with::{DeserializeAs, SerializeAs};

pub struct Muck;

impl<T: NoUninit> SerializeAs<Vec<T>> for Muck {
    fn serialize_as<S>(source: &Vec<T>, serializer: S) -> Result<S::Ok, S::Error>
        where S: serde::Serializer 
    {
        serializer.serialize_bytes(bytemuck::cast_slice::<T, u8>(source))
    }
}

impl<'de, T: AnyBitPattern + NoUninit> DeserializeAs<'de, Vec<T>> for Muck {
    fn deserialize_as<D>(deserializer: D) -> Result<Vec<T>, D::Error>
        where D: serde::Deserializer<'de> 
    {
        struct MuckVisitor<T>(PhantomData<T>);
        impl<'de, T: AnyBitPattern + NoUninit> Visitor<'de> for MuckVisitor<T> {
            type Value = Vec<T>;
            fn expecting(&self, formatter: &mut std::fmt::Formatter) -> std::fmt::Result {
                write!(formatter, "need bytes")
            }
            fn visit_bytes<E>(self, v: &[u8]) -> Result<Self::Value, E>
                where E: serde::de::Error, 
            {
                let mut target: Vec<T> = unsafe { std::mem::transmute(vec![MaybeUninit::<T>::zeroed(); v.len() / size_of::<T>()]) };
                bytemuck::cast_slice_mut(&mut target).copy_from_slice(v);
                Ok(target)
            }
        }
        deserializer.deserialize_bytes(MuckVisitor(PhantomData))
    }
}

Metadata

Metadata

Assignees

No one assigned

    Labels

    enhancementNew feature or requesthelp wantedExtra attention is needed

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions