use crate::err::Error; use crate::sql::ending::number as ending; use crate::sql::error::Error::Parser; use crate::sql::error::IResult; use crate::sql::strand::Strand; use bigdecimal::num_traits::Pow; use bigdecimal::BigDecimal; use bigdecimal::FromPrimitive; use bigdecimal::ToPrimitive; use nom::branch::alt; use nom::character::complete::i64; use nom::number::complete::recognize_float; use nom::Err::Failure; use serde::{Deserialize, Serialize}; use std::cmp::Ordering; use std::fmt::{self, Display, Formatter}; use std::hash; use std::iter::Product; use std::iter::Sum; use std::ops; use std::str::FromStr; pub(crate) const TOKEN: &str = "$surrealdb::private::sql::Number"; #[derive(Clone, Debug, Serialize, Deserialize)] #[serde(rename = "$surrealdb::private::sql::Number")] pub enum Number { Int(i64), Float(f64), Decimal(BigDecimal), // Add new variants here } impl Default for Number { fn default() -> Self { Self::Int(0) } } macro_rules! from_prim_ints { ($($int: ty),*) => { $( impl From<$int> for Number { fn from(i: $int) -> Self { Self::Int(i as i64) } } )* }; } from_prim_ints!(i8, i16, i32, i64, i128, isize, u8, u16, u32, u64, u128, usize); impl From for Number { fn from(f: f32) -> Self { Self::Float(f as f64) } } impl From for Number { fn from(f: f64) -> Self { Self::Float(f) } } impl From for Number { fn from(v: BigDecimal) -> Self { Self::Decimal(v) } } impl FromStr for Number { type Err = (); fn from_str(s: &str) -> Result { Self::try_from(s) } } impl TryFrom for Number { type Error = (); fn try_from(v: String) -> Result { Self::try_from(v.as_str()) } } impl TryFrom for Number { type Error = (); fn try_from(v: Strand) -> Result { Self::try_from(v.as_str()) } } impl TryFrom<&str> for Number { type Error = (); fn try_from(v: &str) -> Result { // Attempt to parse as i64 match v.parse::() { // Store it as an i64 Ok(v) => Ok(Self::Int(v)), // It wasn't parsed as a i64 so parse as a decimal _ => match BigDecimal::from_str(v) { // Store it as a BigDecimal Ok(v) => Ok(Self::Decimal(v)), // It wasn't parsed as a number _ => Err(()), }, } } } macro_rules! try_into_prim { // TODO: switch to one argument per int once https://github.com/rust-lang/rust/issues/29599 is stable ($($int: ty => $to_int: ident),*) => { $( impl TryFrom for $int { type Error = Error; fn try_from(value: Number) -> Result { match value { Number::Int(v) => match v.$to_int() { Some(v) => Ok(v), None => Err(Error::TryFrom(value.to_string(), stringify!($int))), }, Number::Float(v) => match v.$to_int() { Some(v) => Ok(v), None => Err(Error::TryFrom(value.to_string(), stringify!($int))), }, Number::Decimal(ref v) => match v.$to_int() { Some(v) => Ok(v), None => Err(Error::TryFrom(value.to_string(), stringify!($int))), }, } } } )* }; } try_into_prim!( i8 => to_i8, i16 => to_i16, i32 => to_i32, i64 => to_i64, i128 => to_i128, u8 => to_u8, u16 => to_u16, u32 => to_u32, u64 => to_u64, u128 => to_u128, f32 => to_f32, f64 => to_f64 ); impl TryFrom for BigDecimal { type Error = Error; fn try_from(value: Number) -> Result { match value { Number::Int(v) => match BigDecimal::from_i64(v) { Some(v) => Ok(v), None => Err(Error::TryFrom(value.to_string(), "BigDecimal")), }, Number::Float(v) => match BigDecimal::from_f64(v) { Some(v) => Ok(v), None => Err(Error::TryFrom(value.to_string(), "BigDecimal")), }, Number::Decimal(x) => Ok(x), } } } impl Display for Number { fn fmt(&self, f: &mut Formatter) -> fmt::Result { match self { Number::Int(v) => Display::fmt(v, f), Number::Float(v) => Display::fmt(v, f), Number::Decimal(v) => Display::fmt(v, f), } } } impl Number { // ----------------------------------- // Constants // ----------------------------------- pub const NAN: Number = Number::Float(f64::NAN); // ----------------------------------- // Simple number detection // ----------------------------------- pub fn is_nan(&self) -> bool { matches!(self, Number::Float(v) if v.is_nan()) } pub fn is_int(&self) -> bool { matches!(self, Number::Int(_)) } pub fn is_float(&self) -> bool { matches!(self, Number::Float(_)) } pub fn is_decimal(&self) -> bool { matches!(self, Number::Decimal(_)) } pub fn is_integer(&self) -> bool { match self { Number::Int(_) => true, Number::Float(_) => false, Number::Decimal(v) => v.is_integer(), } } pub fn is_truthy(&self) -> bool { match self { Number::Int(v) => v != &0, Number::Float(v) => v != &0.0, Number::Decimal(v) => v != &BigDecimal::default(), } } pub fn is_positive(&self) -> bool { match self { Number::Int(v) => v > &0, Number::Float(v) => v > &0.0, Number::Decimal(v) => v > &BigDecimal::default(), } } pub fn is_negative(&self) -> bool { match self { Number::Int(v) => v < &0, Number::Float(v) => v < &0.0, Number::Decimal(v) => v < &BigDecimal::default(), } } pub fn is_zero_or_positive(&self) -> bool { match self { Number::Int(v) => v >= &0, Number::Float(v) => v >= &0.0, Number::Decimal(v) => v >= &BigDecimal::default(), } } pub fn is_zero_or_negative(&self) -> bool { match self { Number::Int(v) => v <= &0, Number::Float(v) => v <= &0.0, Number::Decimal(v) => v <= &BigDecimal::default(), } } // ----------------------------------- // Simple conversion of number // ----------------------------------- pub fn as_usize(self) -> usize { match self { Number::Int(v) => v as usize, Number::Float(v) => v as usize, Number::Decimal(v) => v.to_usize().unwrap_or_default(), } } pub fn as_int(self) -> i64 { match self { Number::Int(v) => v, Number::Float(v) => v as i64, Number::Decimal(v) => v.to_i64().unwrap_or_default(), } } pub fn as_float(self) -> f64 { match self { Number::Int(v) => v as f64, Number::Float(v) => v, Number::Decimal(v) => v.to_f64().unwrap_or_default(), } } pub fn as_decimal(self) -> BigDecimal { match self { Number::Int(v) => BigDecimal::from_i64(v).unwrap_or_default(), Number::Float(v) => BigDecimal::from_f64(v).unwrap_or_default(), Number::Decimal(v) => v, } } // ----------------------------------- // Complex conversion of number // ----------------------------------- pub fn to_usize(&self) -> usize { match self { Number::Int(v) => *v as usize, Number::Float(v) => *v as usize, Number::Decimal(v) => v.to_usize().unwrap_or_default(), } } pub fn to_int(&self) -> i64 { match self { Number::Int(v) => *v, Number::Float(v) => *v as i64, Number::Decimal(v) => v.to_i64().unwrap_or_default(), } } pub fn to_float(&self) -> f64 { match self { Number::Int(v) => *v as f64, Number::Float(v) => *v, Number::Decimal(v) => v.to_f64().unwrap_or_default(), } } pub fn to_decimal(&self) -> BigDecimal { match self { Number::Int(v) => BigDecimal::from_i64(*v).unwrap_or_default(), Number::Float(v) => BigDecimal::from_f64(*v).unwrap_or_default(), Number::Decimal(v) => v.clone(), } } // ----------------------------------- // // ----------------------------------- pub fn abs(self) -> Self { match self { Number::Int(v) => v.abs().into(), Number::Float(v) => v.abs().into(), Number::Decimal(v) => v.abs().into(), } } pub fn ceil(self) -> Self { match self { Number::Int(v) => v.into(), Number::Float(v) => v.ceil().into(), Number::Decimal(v) => { if v.digits() > 16 { let v = (v.to_f64().unwrap_or_default() + 0.5).round(); BigDecimal::from_f64(v).unwrap_or_default().into() } else { (v + BigDecimal::from_f32(0.5).unwrap()).round(0).into() } } } } pub fn floor(self) -> Self { match self { Number::Int(v) => v.into(), Number::Float(v) => v.floor().into(), Number::Decimal(v) => { if v.digits() > 16 { let v = (v.to_f64().unwrap_or_default() - 0.5).round(); BigDecimal::from_f64(v).unwrap_or_default().into() } else { (v - BigDecimal::from_f32(0.5).unwrap()).round(0).into() } } } } pub fn round(self) -> Self { match self { Number::Int(v) => v.into(), Number::Float(v) => v.round().into(), Number::Decimal(v) => { if v.digits() > 16 { let v = v.to_f64().unwrap_or_default().round(); BigDecimal::from_f64(v).unwrap_or_default().into() } else { v.round(0).into() } } } } pub fn fixed(self, precision: usize) -> Number { match self { Number::Int(v) => format!("{v:.precision$}").try_into().unwrap_or_default(), Number::Float(v) => format!("{v:.precision$}").try_into().unwrap_or_default(), Number::Decimal(v) => format!("{v:.precision$}").try_into().unwrap_or_default(), } } pub fn sqrt(self) -> Self { match self { Number::Int(v) => (v as f64).sqrt().into(), Number::Float(v) => v.sqrt().into(), Number::Decimal(v) => v.sqrt().unwrap_or_default().into(), } } pub fn pow(self, power: Number) -> Number { match (self, power) { (Number::Int(v), Number::Int(p)) if p >= 0 && p < u32::MAX as i64 => { Number::Int(v.pow(p as u32)) } (Number::Decimal(v), Number::Int(p)) if p >= 0 && p < u32::MAX as i64 => { let (as_int, scale) = v.as_bigint_and_exponent(); Number::Decimal(BigDecimal::new(as_int.pow(p as u32), scale * p)) } // TODO: (Number::Decimal(v), Number::Float(p)) => todo!(), // TODO: (Number::Decimal(v), Number::Decimal(p)) => todo!(), (v, p) => Number::Float(v.as_float().pow(p.as_float())), } } } impl Eq for Number {} impl Ord for Number { fn cmp(&self, other: &Self) -> Ordering { self.partial_cmp(other).unwrap_or(Ordering::Equal) } } impl hash::Hash for Number { fn hash(&self, state: &mut H) { match self { Number::Int(v) => v.hash(state), Number::Float(v) => v.to_bits().hash(state), Number::Decimal(v) => v.hash(state), } } } impl PartialEq for Number { fn eq(&self, other: &Self) -> bool { match (self, other) { (Number::Int(v), Number::Int(w)) => v.eq(w), (Number::Float(v), Number::Float(w)) => v.eq(w), (Number::Decimal(v), Number::Decimal(w)) => v.eq(w), // ------------------------------ (Number::Int(v), Number::Float(w)) => (*v as f64).eq(w), (Number::Float(v), Number::Int(w)) => v.eq(&(*w as f64)), // ------------------------------ (Number::Int(v), Number::Decimal(w)) => BigDecimal::from(*v).eq(w), (Number::Decimal(v), Number::Int(w)) => v.eq(&BigDecimal::from(*w)), // ------------------------------ (Number::Float(v), Number::Decimal(w)) => { BigDecimal::from_f64(*v).unwrap_or_default().eq(w) } (Number::Decimal(v), Number::Float(w)) => { v.eq(&BigDecimal::from_f64(*w).unwrap_or_default()) } } } } impl PartialOrd for Number { fn partial_cmp(&self, other: &Self) -> Option { match (self, other) { (Number::Int(v), Number::Int(w)) => v.partial_cmp(w), (Number::Float(v), Number::Float(w)) => v.partial_cmp(w), (Number::Decimal(v), Number::Decimal(w)) => v.partial_cmp(w), // ------------------------------ (Number::Int(v), Number::Float(w)) => (*v as f64).partial_cmp(w), (Number::Float(v), Number::Int(w)) => v.partial_cmp(&(*w as f64)), // ------------------------------ (Number::Int(v), Number::Decimal(w)) => BigDecimal::from(*v).partial_cmp(w), (Number::Decimal(v), Number::Int(w)) => v.partial_cmp(&BigDecimal::from(*w)), // ------------------------------ (Number::Float(v), Number::Decimal(w)) => { BigDecimal::from_f64(*v).unwrap_or_default().partial_cmp(w) } (Number::Decimal(v), Number::Float(w)) => { v.partial_cmp(&BigDecimal::from_f64(*w).unwrap_or_default()) } } } } impl ops::Add for Number { type Output = Self; fn add(self, other: Self) -> Self { match (self, other) { (Number::Int(v), Number::Int(w)) => Number::Int(v + w), (Number::Float(v), Number::Float(w)) => Number::Float(v + w), (Number::Decimal(v), Number::Decimal(w)) => Number::Decimal(v + w), (Number::Int(v), Number::Float(w)) => Number::Float(v as f64 + w), (Number::Float(v), Number::Int(w)) => Number::Float(v + w as f64), (v, w) => Number::from(v.as_decimal() + w.as_decimal()), } } } impl<'a, 'b> ops::Add<&'b Number> for &'a Number { type Output = Number; fn add(self, other: &'b Number) -> Number { match (self, other) { (Number::Int(v), Number::Int(w)) => Number::Int(v + w), (Number::Float(v), Number::Float(w)) => Number::Float(v + w), (Number::Decimal(v), Number::Decimal(w)) => Number::Decimal(v + w), (Number::Int(v), Number::Float(w)) => Number::Float(*v as f64 + w), (Number::Float(v), Number::Int(w)) => Number::Float(v + *w as f64), (v, w) => Number::from(v.to_decimal() + w.to_decimal()), } } } impl ops::Sub for Number { type Output = Self; fn sub(self, other: Self) -> Self { match (self, other) { (Number::Int(v), Number::Int(w)) => Number::Int(v - w), (Number::Float(v), Number::Float(w)) => Number::Float(v - w), (Number::Decimal(v), Number::Decimal(w)) => Number::Decimal(v - w), (Number::Int(v), Number::Float(w)) => Number::Float(v as f64 - w), (Number::Float(v), Number::Int(w)) => Number::Float(v - w as f64), (v, w) => Number::from(v.as_decimal() - w.as_decimal()), } } } impl<'a, 'b> ops::Sub<&'b Number> for &'a Number { type Output = Number; fn sub(self, other: &'b Number) -> Number { match (self, other) { (Number::Int(v), Number::Int(w)) => Number::Int(v - w), (Number::Float(v), Number::Float(w)) => Number::Float(v - w), (Number::Decimal(v), Number::Decimal(w)) => Number::Decimal(v - w), (Number::Int(v), Number::Float(w)) => Number::Float(*v as f64 - w), (Number::Float(v), Number::Int(w)) => Number::Float(v - *w as f64), (v, w) => Number::from(v.to_decimal() - w.to_decimal()), } } } impl ops::Mul for Number { type Output = Self; fn mul(self, other: Self) -> Self { match (self, other) { (Number::Int(v), Number::Int(w)) => Number::Int(v * w), (Number::Float(v), Number::Float(w)) => Number::Float(v * w), (Number::Decimal(v), Number::Decimal(w)) => Number::Decimal(v * w), (Number::Int(v), Number::Float(w)) => Number::Float(v as f64 * w), (Number::Float(v), Number::Int(w)) => Number::Float(v * w as f64), (v, w) => Number::from(v.as_decimal() * w.as_decimal()), } } } impl<'a, 'b> ops::Mul<&'b Number> for &'a Number { type Output = Number; fn mul(self, other: &'b Number) -> Number { match (self, other) { (Number::Int(v), Number::Int(w)) => Number::Int(v * w), (Number::Float(v), Number::Float(w)) => Number::Float(v * w), (Number::Decimal(v), Number::Decimal(w)) => Number::Decimal(v * w), (Number::Int(v), Number::Float(w)) => Number::Float(*v as f64 * w), (Number::Float(v), Number::Int(w)) => Number::Float(v * *w as f64), (v, w) => Number::from(v.to_decimal() * w.to_decimal()), } } } impl ops::Div for Number { type Output = Self; fn div(self, other: Self) -> Self { match (self, other) { (Number::Float(v), Number::Float(w)) => Number::Float(v / w), (Number::Decimal(v), Number::Decimal(w)) => Number::Decimal(v / w), (Number::Int(v), Number::Float(w)) => Number::Float(v as f64 / w), (Number::Float(v), Number::Int(w)) => Number::Float(v / w as f64), (v, w) => Number::from(v.as_decimal() / w.as_decimal()), } } } impl<'a, 'b> ops::Div<&'b Number> for &'a Number { type Output = Number; fn div(self, other: &'b Number) -> Number { match (self, other) { (Number::Float(v), Number::Float(w)) => Number::Float(v / w), (Number::Decimal(v), Number::Decimal(w)) => Number::Decimal(v / w), (Number::Int(v), Number::Float(w)) => Number::Float(*v as f64 / w), (Number::Float(v), Number::Int(w)) => Number::Float(v / *w as f64), (v, w) => Number::from(v.to_decimal() / w.to_decimal()), } } } // ------------------------------ impl Sum for Number { fn sum(iter: I) -> Number where I: Iterator, { iter.fold(Number::Int(0), |a, b| a + b) } } impl<'a> Sum<&'a Self> for Number { fn sum(iter: I) -> Number where I: Iterator, { iter.fold(Number::Int(0), |a, b| &a + b) } } impl Product for Number { fn product(iter: I) -> Number where I: Iterator, { iter.fold(Number::Int(1), |a, b| a * b) } } impl<'a> Product<&'a Self> for Number { fn product(iter: I) -> Number where I: Iterator, { iter.fold(Number::Int(1), |a, b| &a * b) } } pub struct Sorted(pub T); pub trait Sort { fn sorted(&mut self) -> Sorted<&Self> where Self: Sized; } impl Sort for Vec { fn sorted(&mut self) -> Sorted<&Vec> { self.sort(); Sorted(self) } } pub fn number(i: &str) -> IResult<&str, Number> { alt((int, decimal))(i) } pub fn integer(i: &str) -> IResult<&str, i64> { let (i, v) = i64(i)?; let (i, _) = ending(i)?; Ok((i, v)) } fn int(i: &str) -> IResult<&str, Number> { let (i, v) = i64(i)?; let (i, _) = ending(i)?; Ok((i, Number::from(v))) } fn decimal(i: &str) -> IResult<&str, Number> { let (i, v) = recognize_float(i)?; let (i, _) = ending(i)?; Ok((i, Number::try_from(v).map_err(|_| Failure(Parser(i)))?)) } #[cfg(test)] mod tests { use super::*; #[test] fn number_int() { let sql = "123"; let res = number(sql); assert!(res.is_ok()); let out = res.unwrap().1; assert_eq!("123", format!("{}", out)); assert_eq!(out, Number::Int(123)); } #[test] fn number_int_neg() { let sql = "-123"; let res = number(sql); assert!(res.is_ok()); let out = res.unwrap().1; assert_eq!("-123", format!("{}", out)); assert_eq!(out, Number::Int(-123)); } #[test] fn number_float() { let sql = "123.45"; let res = number(sql); assert!(res.is_ok()); let out = res.unwrap().1; assert_eq!("123.45", format!("{}", out)); assert_eq!(out, Number::Float(123.45)); } #[test] fn number_float_neg() { let sql = "-123.45"; let res = number(sql); assert!(res.is_ok()); let out = res.unwrap().1; assert_eq!("-123.45", format!("{}", out)); assert_eq!(out, Number::Float(-123.45)); } #[test] fn number_scientific_lower() { let sql = "12345e-1"; let res = number(sql); assert!(res.is_ok()); let out = res.unwrap().1; assert_eq!("1234.5", format!("{}", out)); assert_eq!(out, Number::Float(1234.5)); } #[test] fn number_scientific_lower_neg() { let sql = "-12345e-1"; let res = number(sql); assert!(res.is_ok()); let out = res.unwrap().1; assert_eq!("-1234.5", format!("{}", out)); assert_eq!(out, Number::Float(-1234.5)); } #[test] fn number_scientific_upper() { let sql = "12345E-02"; let res = number(sql); assert!(res.is_ok()); let out = res.unwrap().1; assert_eq!("123.45", format!("{}", out)); assert_eq!(out, Number::Float(123.45)); } #[test] fn number_scientific_upper_neg() { let sql = "-12345E-02"; let res = number(sql); assert!(res.is_ok()); let out = res.unwrap().1; assert_eq!("-123.45", format!("{}", out)); assert_eq!(out, Number::Float(-123.45)); } #[test] fn number_float_keeps_precision() { let sql = "13.571938471938472"; let res = number(sql); assert!(res.is_ok()); let out = res.unwrap().1; assert_eq!("13.571938471938472", format!("{}", out)); assert_eq!(out, Number::try_from("13.571938471938472").unwrap()); } #[test] fn number_decimal_keeps_precision() { let sql = "13.571938471938471938563985639413947693775636"; let res = number(sql); assert!(res.is_ok()); let out = res.unwrap().1; assert_eq!("13.571938471938471938563985639413947693775636", format!("{}", out)); assert_eq!(out, Number::try_from("13.571938471938471938563985639413947693775636").unwrap()); } #[test] fn number_pow_int() { let res = Number::Int(3).pow(Number::Int(4)); assert_eq!(res, Number::Int(81)); } #[test] fn number_pow_int_negative() { let res = Number::Int(4).pow(Number::Float(-0.5)); assert_eq!(res, Number::Float(0.5)); } #[test] fn number_pow_float() { let res = Number::Float(2.5).pow(Number::Int(2)); assert_eq!(res, Number::Float(6.25)); } #[test] fn number_pow_float_negative() { let res = Number::Int(4).pow(Number::Float(-0.5)); assert_eq!(res, Number::Float(0.5)); } #[test] fn number_pow_decimal_one() { let res = Number::try_from("13.5719384719384719385639856394139476937756394756") .unwrap() .pow(Number::Int(1)); assert_eq!( res, Number::try_from("13.5719384719384719385639856394139476937756394756").unwrap() ); } #[test] fn number_pow_decimal_two() { let res = Number::try_from("13.5719384719384719385639856394139476937756394756") .unwrap() .pow(Number::Int(2)); assert_eq!( res, Number::try_from("184.19751388608358465578173996877942643463869043732548087725588482334195240945031617770904299536").unwrap() ); } }