Skip to main content

geo_hash/
serde.rs

1use serde::de::{Deserialize, Deserializer};
2use serde::ser::{Serialize, Serializer};
3
4use crate::codec::GeoHash;
5
6impl Serialize for GeoHash {
7    #[inline]
8    fn serialize<SER: Serializer>(&self, ser: SER) -> Result<SER::Ok, SER::Error> {
9        ser.serialize_str(self.as_str())
10    }
11}
12
13struct GeoHashVisitor;
14
15impl<'de> serde::de::Visitor<'de> for GeoHashVisitor {
16    type Value = GeoHash;
17
18    #[inline(always)]
19    fn expecting(&self, formatter: &mut core::fmt::Formatter) -> core::fmt::Result {
20        formatter.write_str("a geohash string with len in range 1..=12")
21    }
22
23    #[inline]
24    fn visit_str<E: serde::de::Error>(self, v: &str) -> Result<Self::Value, E> {
25        match Self::Value::try_from_str(v) {
26            Some(result) => Ok(result),
27            None => Err(serde::de::Error::custom(format_args!("Exceeds buffer capacity({} bytes)", Self::Value::MAX_LEN)))
28        }
29    }
30}
31
32impl<'a> Deserialize<'a> for GeoHash {
33    #[inline]
34    fn deserialize<D: Deserializer<'a>>(des: D) -> Result<Self, D::Error> {
35        des.deserialize_str(GeoHashVisitor)
36    }
37}
38
39#[cfg(test)]
40mod tests {
41    use crate::codec::GeoHash;
42
43    use serde::de::Deserialize;
44    use serde::de::value::{BorrowedStrDeserializer, Error as ValueError};
45
46    #[test]
47    fn should_error_one_exceeding_capacity() {
48        let des = BorrowedStrDeserializer::<ValueError>::new("1234567891234");
49        let res = GeoHash::deserialize(des);
50        assert!(res.is_err());
51    }
52
53    #[test]
54    fn should_ok_within_capacity() {
55        let des = BorrowedStrDeserializer::<ValueError>::new("123456789123");
56        let res = GeoHash::deserialize(des).expect("Unexpected fail");
57        assert_eq!(res.as_str(), "123456789123");
58    }
59}