reifydb_type/value/decimal/
parse.rs1use std::{borrow::Cow, str::FromStr};
5
6use bigdecimal::BigDecimal as BigDecimalInner;
7
8use crate::{
9 Error, IntoFragment, Type, error::diagnostic::number::invalid_number_format, return_error,
10 value::decimal::Decimal,
11};
12
13pub fn parse_decimal<'a>(fragment: impl IntoFragment<'a>) -> Result<Decimal, Error> {
14 let fragment = fragment.into_fragment();
15 let fragment_owned = fragment.clone().into_owned();
16 let raw_value = fragment.text();
17
18 let needs_trimming = raw_value.as_bytes().first().map_or(false, |&b| b.is_ascii_whitespace())
20 || raw_value.as_bytes().last().map_or(false, |&b| b.is_ascii_whitespace());
21 let has_underscores = raw_value.as_bytes().contains(&b'_');
22
23 let value = match (needs_trimming, has_underscores) {
24 (false, false) => Cow::Borrowed(raw_value),
25 (true, false) => Cow::Borrowed(raw_value.trim()),
26 (false, true) => Cow::Owned(raw_value.replace('_', "")),
27 (true, true) => Cow::Owned(raw_value.trim().replace('_', "")),
28 };
29
30 if value.is_empty() {
31 return_error!(invalid_number_format(fragment_owned.clone(), Type::Decimal));
32 }
33
34 let big_decimal = BigDecimalInner::from_str(&value)
35 .map_err(|_| crate::error!(invalid_number_format(fragment_owned, Type::Decimal)))?;
36
37 Ok(Decimal::new(big_decimal))
38}
39
40#[cfg(test)]
41mod tests {
42 use super::*;
43 use crate::OwnedFragment;
44
45 #[test]
46 fn test_parse_decimal_integer() {
47 let decimal = parse_decimal(OwnedFragment::testing("123")).unwrap();
48 assert_eq!(decimal.to_string(), "123");
49 }
50
51 #[test]
52 fn test_parse_decimal_with_fractional() {
53 let decimal = parse_decimal(OwnedFragment::testing("123.45")).unwrap();
54 assert_eq!(decimal.to_string(), "123.45");
55 }
56
57 #[test]
58 fn test_parse_decimal_with_underscores() {
59 let decimal = parse_decimal(OwnedFragment::testing("1_234.56")).unwrap();
60 assert_eq!(decimal.to_string(), "1234.56");
61 }
62
63 #[test]
64 fn test_parse_decimal_negative() {
65 let decimal = parse_decimal(OwnedFragment::testing("-123.45")).unwrap();
66 assert_eq!(decimal.to_string(), "-123.45");
67 }
68
69 #[test]
70 fn test_parse_decimal_empty() {
71 assert!(parse_decimal(OwnedFragment::testing("")).is_err());
72 }
73
74 #[test]
75 fn test_parse_decimal_invalid() {
76 assert!(parse_decimal(OwnedFragment::testing("not_a_number")).is_err());
77 }
78
79 #[test]
80 fn test_parse_decimal_scientific_notation() {
81 let decimal = parse_decimal(OwnedFragment::testing("1.23e2")).unwrap();
82 assert_eq!(decimal.to_string(), "123");
83 }
84}