reifydb_type/value/decimal/
parse.rs

1// Copyright (c) reifydb.com 2025
2// This file is licensed under the MIT, see license.md file
3
4use 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	// Fast path: check if we need any string processing
19	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}