impl TryFrom<&[u8]> for NumberAny

Author: samuelcolvin

crates/jiter/src/number_decoder.rs

@@ -133,6 +133,29 @@ impl From<NumberAny> for f64 {
     }
 }
 
+impl TryFrom<&[u8]> for NumberAny {
+    type Error = JsonError;
+
+    fn try_from(value: &[u8]) -> Result<Self, Self::Error> {
+        let first = *value.first().ok_or_else(|| json_error!(InvalidNumber, 0))?;
+        let (int_parse, index) = IntParse::parse(value, 0, first)?;
+        let (number, index) = match int_parse {
+            IntParse::Int(int) => (Self::Int(int), index),
+            IntParse::Float => {
+                let (f, index) = NumberFloat::decode(value, 0, first, false)?;
+                (Self::Float(f), index)
+            }
+            _ => return json_err!(InvalidNumber, index),
+        };
+
+        if index == value.len() {
+            Ok(number)
+        } else {
+            json_err!(InvalidNumber, index)
+        }
+    }
+}
+
 impl AbstractNumberDecoder for NumberAny {
     type Output = NumberAny;
 

crates/jiter/tests/main.rs

@@ -1418,6 +1418,50 @@ fn test_number_int_try_from_bytes() {
     assert_eq!(e.to_string(), "number out of range at index 4301");
 }
 
+#[test]
+fn test_number_any_try_from_bytes() {
+    let n: NumberAny = b"123".as_ref().try_into().unwrap();
+    assert_eq!(n, NumberAny::Int(NumberInt::Int(123)));
+
+    let n: NumberAny = b"0".as_ref().try_into().unwrap();
+    assert_eq!(n, NumberAny::Int(NumberInt::Int(0)));
+
+    let n: NumberAny = b"123.456".as_ref().try_into().unwrap();
+    assert_eq!(n, NumberAny::Float(123.456));
+
+    let n: NumberAny = b"123.45e2".as_ref().try_into().unwrap();
+    assert_eq!(n, NumberAny::Float(12345.0));
+
+    let twenty_nines = "9".repeat(29);
+    let n: NumberAny = twenty_nines.as_bytes().try_into().unwrap();
+    match n {
+        NumberAny::Int(NumberInt::BigInt(v)) => assert_eq!(v.to_string(), twenty_nines),
+        _ => panic!("expected big int"),
+    }
+
+    let e = NumberAny::try_from(b"x23".as_ref()).unwrap_err();
+    assert_eq!(e.to_string(), "invalid number at index 0");
+
+    let e = NumberAny::try_from(b"".as_ref()).unwrap_err();
+    assert_eq!(e.to_string(), "invalid number at index 0");
+
+    let e = NumberAny::try_from(b"2x3".as_ref()).unwrap_err();
+    assert_eq!(e.to_string(), "invalid number at index 1");
+
+    let e = NumberAny::try_from(b"123 ".as_ref()).unwrap_err();
+    assert_eq!(e.to_string(), "invalid number at index 3");
+
+    let e = NumberAny::try_from(b"123.1 ".as_ref()).unwrap_err();
+    assert_eq!(e.to_string(), "invalid number at index 5");
+
+    let e = NumberAny::try_from(b"0123".as_ref()).unwrap_err();
+    assert_eq!(e.to_string(), "invalid number at index 1");
+
+    let too_long = "9".repeat(4309);
+    let e = NumberAny::try_from(too_long.as_bytes()).unwrap_err();
+    assert_eq!(e.to_string(), "number out of range at index 4301");
+}
+
 #[test]
 fn jiter_skip_whole_object() {
     let mut jiter = Jiter::new(br#"{"x": 1}"#);