byte_unit/unit/
serde_traits.rs

1use core::{fmt, fmt::Formatter, str::FromStr};
2
3use serde::{
4    self,
5    de::{Error as DeError, Visitor},
6    Deserialize, Deserializer, Serialize, Serializer,
7};
8
9use super::Unit;
10
11impl Serialize for Unit {
12    #[inline]
13    fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
14    where
15        S: Serializer, {
16        serializer.serialize_str(self.as_str())
17    }
18}
19
20impl<'de> Deserialize<'de> for Unit {
21    #[inline]
22    fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
23    where
24        D: Deserializer<'de>, {
25        struct MyVisitor;
26
27        impl<'de> Visitor<'de> for MyVisitor {
28            type Value = Unit;
29
30            #[inline]
31            fn expecting(&self, f: &mut Formatter<'_>) -> fmt::Result {
32                f.write_str("a string such as \"B\", \"KB\" or \"MiB\"")
33            }
34
35            #[inline]
36            fn visit_str<E>(self, v: &str) -> Result<Self::Value, E>
37            where
38                E: DeError, {
39                Unit::from_str(v).map_err(DeError::custom)
40            }
41        }
42
43        deserializer.deserialize_str(MyVisitor)
44    }
45}
byte_unit/unit/serde_traits.rs

byte_unit/unit/
serde_traits.rs
