Add Result serializer/deserializer

Signed-off-by: Heinz N. Gies <[email protected]>
simd-lite · Apr 3, 2023 · cf3d24d · cf3d24d
1 parent 85fdb65
commit cf3d24d
Show file tree

Hide file tree

Showing 2 changed files with 65 additions and 1 deletion.
diff --git a/src/impls.rs b/src/impls.rs
@@ -57,3 +57,50 @@ where
         }
     }
 }
+
+impl<TOk, TErr> Serialize for std::result::Result<TOk, TErr>
+where
+    TOk: Serialize,
+    TErr: Serialize,
+{
+    #[inline]
+    fn json_write<W>(&self, writer: &mut W) -> io::Result<()>
+    where
+        W: Write,
+    {
+        match self {
+            Ok(e) => {
+                writer.write_all(b"{\"ok\":")?;
+                e.json_write(writer)?;
+                writer.write_all(b"}")
+            }
+            Err(e) => {
+                writer.write_all(b"{\"err\":")?;
+                e.json_write(writer)?;
+                writer.write_all(b"}")
+            }
+        }
+    }
+}
+
+impl<'input, TOk, TErr> Deserialize<'input> for std::result::Result<TOk, TErr>
+where
+    TOk: Deserialize<'input>,
+    TErr: Deserialize<'input>,
+{
+    #[inline]
+    fn from_tape(tape: &mut Tape<'input>) -> simd_json::Result<Self>
+    where
+        Self: std::marker::Sized + 'input,
+    {
+        if let Some(simd_json::Node::Object(1, _)) = tape.next() {
+            match tape.next() {
+                Some(simd_json::Node::String("ok")) => Ok(Ok(TOk::from_tape(tape)?)),
+                Some(simd_json::Node::String("err")) => Ok(Err(TErr::from_tape(tape)?)),
+                _ => Err(simd_json::Error::generic(simd_json::ErrorType::ExpectedMap)),
+            }
+        } else {
+            Err(simd_json::Error::generic(simd_json::ErrorType::ExpectedMap))
+        }
+    }
+}
diff --git a/tests/deser.rs b/tests/deser.rs
@@ -124,12 +124,13 @@ fn event() {
 
 #[test]
 fn enum_ser() {
-    #[derive(Deserialize, PartialEq, Eq, Debug)]
+    #[derive(Deserialize, Serialize, PartialEq, Eq, Debug)]
     pub enum StoredVariants {
         YesNo(bool),
         Small(u8, i8),
         Signy(i64),
         Stringy(String),
+        Res(Result<u8, String>),
     }
 
     let mut s = String::from(r#"{"Small":[1,2]}"#);
@@ -139,4 +140,20 @@ fn enum_ser() {
     let mut s = String::from(r#"{"YesNo":true}"#);
     let e = StoredVariants::from_str(s.as_mut_str()).unwrap();
     assert_eq!(StoredVariants::YesNo(true), e);
+
+    let mut s = String::from(r#"{"Res":{"ok":42}}"#);
+    let e = StoredVariants::from_str(s.as_mut_str()).unwrap();
+    assert_eq!(StoredVariants::Res(Ok(42)), e);
+
+    let mut s = String::from(r#"{"Res":{"err":"snot"}}"#);
+    let e = StoredVariants::from_str(s.as_mut_str()).unwrap();
+    assert_eq!(StoredVariants::Res(Err(String::from("snot"))), e);
+
+    let e = StoredVariants::Res(Ok(42)).json_string().unwrap();
+    assert_eq!(r#"{"Res":{"ok":42}}"#, e);
+
+    let e = StoredVariants::Res(Err(String::from("snot")))
+        .json_string()
+        .unwrap();
+    assert_eq!(r#"{"Res":{"err":"snot"}}"#, e);
 }