byte_unit/unit/
serde_traits.rs

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

byte_unit/unit/
serde_traits.rs
