reifydb_type/value/decimal/

mod.rs

// SPDX-License-Identifier: MIT
// Copyright (c) 2025 ReifyDB

use std::{
	cmp::Ordering,
	fmt::{Display, Formatter},
	ops::{Add, Deref, Div, Mul, Sub},
	str::FromStr,
};

use bigdecimal::{BigDecimal as BigDecimalInner, FromPrimitive};
use num_traits::{One, Zero};
use serde::{
	Deserialize, Deserializer, Serialize, Serializer,
	de::{self, Visitor},
};

use super::{int::Int, uint::Uint};
use crate::{error, error::Error, fragment::Fragment, value::r#type::Type};

pub mod parse;

use crate::error::diagnostic::number::invalid_number_format;

#[repr(transparent)]
#[derive(Clone, Debug)]
pub struct Decimal(pub BigDecimalInner);

impl Decimal {
	pub fn zero() -> Self {
		Self(BigDecimalInner::zero())
	}

	pub fn one() -> Self {
		Self(BigDecimalInner::one())
	}
}

impl Deref for Decimal {
	type Target = BigDecimalInner;

	fn deref(&self) -> &Self::Target {
		&self.0
	}
}

impl Decimal {
	pub fn new(value: BigDecimalInner) -> Self {
		Self(value)
	}

	pub fn from_bigdecimal(value: BigDecimalInner) -> Self {
		Self(value)
	}

	pub fn with_scale(value: BigDecimalInner, scale: i64) -> Self {
		Self(value.with_scale(scale))
	}

	pub fn from_i64(value: i64) -> Self {
		Self(BigDecimalInner::from(value))
	}

	pub fn inner(&self) -> &BigDecimalInner {
		&self.0
	}

	pub fn to_bigdecimal(self) -> BigDecimalInner {
		self.0
	}

	pub fn negate(self) -> Self {
		Self(-self.0)
	}
}

impl PartialEq for Decimal {
	fn eq(&self, other: &Self) -> bool {
		self.0 == other.0
	}
}

impl Eq for Decimal {}

impl PartialOrd for Decimal {
	fn partial_cmp(&self, other: &Self) -> Option<Ordering> {
		Some(self.cmp(other))
	}
}

impl Ord for Decimal {
	fn cmp(&self, other: &Self) -> Ordering {
		self.0.cmp(&other.0)
	}
}

impl Display for Decimal {
	fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
		self.0.fmt(f)
	}
}

impl std::hash::Hash for Decimal {
	fn hash<H: std::hash::Hasher>(&self, state: &mut H) {
		self.0.to_string().hash(state);
	}
}

impl FromStr for Decimal {
	type Err = Error;

	fn from_str(s: &str) -> Result<Self, Self::Err> {
		let big_decimal = BigDecimalInner::from_str(s)
			.map_err(|_| error!(invalid_number_format(Fragment::None, Type::Decimal)))?;

		Ok(Self(big_decimal))
	}
}

impl From<i64> for Decimal {
	fn from(value: i64) -> Self {
		Self(BigDecimalInner::from(value))
	}
}

impl From<i8> for Decimal {
	fn from(value: i8) -> Self {
		Self::from(value as i64)
	}
}

impl From<i16> for Decimal {
	fn from(value: i16) -> Self {
		Self::from(value as i64)
	}
}

impl From<i32> for Decimal {
	fn from(value: i32) -> Self {
		Self::from(value as i64)
	}
}

impl From<i128> for Decimal {
	fn from(value: i128) -> Self {
		Self(BigDecimalInner::from(value))
	}
}

impl From<u8> for Decimal {
	fn from(value: u8) -> Self {
		Self::from(value as i64)
	}
}

impl From<u16> for Decimal {
	fn from(value: u16) -> Self {
		Self::from(value as i64)
	}
}

impl From<u32> for Decimal {
	fn from(value: u32) -> Self {
		Self::from(value as i64)
	}
}

impl From<u64> for Decimal {
	fn from(value: u64) -> Self {
		Self(BigDecimalInner::from(value))
	}
}

impl From<u128> for Decimal {
	fn from(value: u128) -> Self {
		Self(BigDecimalInner::from(value))
	}
}

impl From<f32> for Decimal {
	fn from(value: f32) -> Self {
		let inner = BigDecimalInner::from_f32(value).unwrap_or_else(|| BigDecimalInner::from(0));
		Self(inner)
	}
}

impl From<f64> for Decimal {
	fn from(value: f64) -> Self {
		let inner = BigDecimalInner::from_f64(value).unwrap_or_else(|| BigDecimalInner::from(0));
		Self(inner)
	}
}

impl From<BigDecimalInner> for Decimal {
	fn from(value: BigDecimalInner) -> Self {
		Self(value)
	}
}

impl From<Int> for Decimal {
	fn from(value: Int) -> Self {
		Self(BigDecimalInner::from_bigint(value.0, 0))
	}
}

impl From<Uint> for Decimal {
	fn from(value: Uint) -> Self {
		Self(BigDecimalInner::from_bigint(value.0, 0))
	}
}

// Arithmetic operations
impl Add for Decimal {
	type Output = Self;

	fn add(self, rhs: Self) -> Self::Output {
		Self(self.0 + rhs.0)
	}
}

impl Sub for Decimal {
	type Output = Self;

	fn sub(self, rhs: Self) -> Self::Output {
		Self(self.0 - rhs.0)
	}
}

impl Mul for Decimal {
	type Output = Self;

	fn mul(self, rhs: Self) -> Self::Output {
		Self(self.0 * rhs.0)
	}
}

impl Div for Decimal {
	type Output = Self;

	fn div(self, rhs: Self) -> Self::Output {
		Self(self.0 / rhs.0)
	}
}

// Reference arithmetic operations (to avoid cloning)
impl Add<&Decimal> for &Decimal {
	type Output = Decimal;

	fn add(self, rhs: &Decimal) -> Self::Output {
		Decimal(&self.0 + &rhs.0)
	}
}

impl Sub<&Decimal> for &Decimal {
	type Output = Decimal;

	fn sub(self, rhs: &Decimal) -> Self::Output {
		Decimal(&self.0 - &rhs.0)
	}
}

impl Mul<&Decimal> for &Decimal {
	type Output = Decimal;

	fn mul(self, rhs: &Decimal) -> Self::Output {
		Decimal(&self.0 * &rhs.0)
	}
}

impl Div<&Decimal> for &Decimal {
	type Output = Decimal;

	fn div(self, rhs: &Decimal) -> Self::Output {
		Decimal(&self.0 / &rhs.0)
	}
}

impl Default for Decimal {
	fn default() -> Self {
		Self::zero()
	}
}

// Serde implementation for string-based serialization
// This works with both JSON and binary formats (bincode, rmp, etc.)
impl Serialize for Decimal {
	fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
	where
		S: Serializer,
	{
		serializer.serialize_str(&self.0.to_string())
	}
}

struct DecimalVisitor;

impl<'de> Visitor<'de> for DecimalVisitor {
	type Value = Decimal;

	fn expecting(&self, formatter: &mut std::fmt::Formatter) -> std::fmt::Result {
		formatter.write_str("a decimal number as a string")
	}

	fn visit_str<E>(self, value: &str) -> Result<Decimal, E>
	where
		E: de::Error,
	{
		BigDecimalInner::from_str(value).map(Decimal).map_err(|e| E::custom(format!("invalid decimal: {}", e)))
	}
}

impl<'de> Deserialize<'de> for Decimal {
	fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
	where
		D: Deserializer<'de>,
	{
		deserializer.deserialize_str(DecimalVisitor)
	}
}

#[cfg(test)]
pub mod tests {
	use super::*;

	#[test]
	fn test_new_decimal_valid() {
		let bd = BigDecimalInner::from_str("123.45").unwrap();
		let decimal = Decimal::new(bd);
		assert_eq!(decimal.to_string(), "123.45");
	}

	#[test]
	fn test_from_str() {
		let decimal = Decimal::from_str("123.45").unwrap();
		assert_eq!(decimal.to_string(), "123.45");
	}

	#[test]
	fn test_comparison() {
		let d1 = Decimal::from_str("123.45").unwrap();
		let d2 = Decimal::from_str("123.46").unwrap();
		let d3 = Decimal::from_str("123.45").unwrap();

		assert!(d1 < d2);
		assert_eq!(d1, d3);
	}

	#[test]
	fn test_display() {
		let decimal = Decimal::from_str("123.45").unwrap();
		assert_eq!(format!("{}", decimal), "123.45");
	}

	#[test]
	fn test_serde_json() {
		let decimal = Decimal::from_str("123.456789").unwrap();
		let json = serde_json::to_string(&decimal).unwrap();
		assert_eq!(json, "\"123.456789\"");

		let deserialized: Decimal = serde_json::from_str(&json).unwrap();
		assert_eq!(deserialized, decimal);
	}

	#[test]
	fn test_serde_json_negative() {
		let decimal = Decimal::from_str("-987.654321").unwrap();
		let json = serde_json::to_string(&decimal).unwrap();
		assert_eq!(json, "\"-987.654321\"");

		let deserialized: Decimal = serde_json::from_str(&json).unwrap();
		assert_eq!(deserialized, decimal);
	}

	#[test]
	fn test_serde_json_zero() {
		let decimal = Decimal::zero();
		let json = serde_json::to_string(&decimal).unwrap();
		assert_eq!(json, "\"0\"");

		let deserialized: Decimal = serde_json::from_str(&json).unwrap();
		assert_eq!(deserialized, decimal);
	}

	#[test]
	fn test_serde_json_high_precision() {
		let decimal = Decimal::from_str("123456789.123456789123456789").unwrap();
		let json = serde_json::to_string(&decimal).unwrap();

		let deserialized: Decimal = serde_json::from_str(&json).unwrap();
		assert_eq!(deserialized, decimal);
	}

	#[test]
	fn test_serde_postcard() {
		let decimal = Decimal::from_str("123.456789").unwrap();
		let encoded = postcard::to_stdvec(&decimal).unwrap();

		let decoded: Decimal = postcard::from_bytes(&encoded).unwrap();
		assert_eq!(decoded, decimal);
	}

	#[test]
	fn test_serde_postcard_negative() {
		let decimal = Decimal::from_str("-987.654321").unwrap();
		let encoded = postcard::to_stdvec(&decimal).unwrap();

		let decoded: Decimal = postcard::from_bytes(&encoded).unwrap();
		assert_eq!(decoded, decimal);
	}

	#[test]
	fn test_serde_postcard_zero() {
		let decimal = Decimal::zero();
		let encoded = postcard::to_stdvec(&decimal).unwrap();

		let decoded: Decimal = postcard::from_bytes(&encoded).unwrap();
		assert_eq!(decoded, decimal);
	}

	#[test]
	fn test_serde_postcard_high_precision() {
		let decimal = Decimal::from_str("123456789.123456789123456789").unwrap();
		let encoded = postcard::to_stdvec(&decimal).unwrap();

		let decoded: Decimal = postcard::from_bytes(&encoded).unwrap();
		assert_eq!(decoded, decimal);
	}

	#[test]
	fn test_serde_postcard_large_number() {
		let decimal = Decimal::from_str("999999999999999999999999999999.999999999999999999999999").unwrap();
		let encoded = postcard::to_stdvec(&decimal).unwrap();

		let decoded: Decimal = postcard::from_bytes(&encoded).unwrap();
		assert_eq!(decoded, decimal);
	}
}