use crate::{Deserializer, Error, Serializer};
use std::fmt::Display;
use std::str::FromStr;
#[derive(Debug, Copy, Clone, PartialEq, Eq, PartialOrd, Ord, Hash)]
#[cfg_attr(docsrs, doc(cfg(feature = "decimal")))]
pub enum Decimal {
NegInf,
Normalized(rust_decimal::Decimal),
Inf,
NaN,
}
impl Decimal {
pub const ZERO: Self = Decimal::Normalized(rust_decimal::Decimal::ZERO);
pub fn to_vec(&self) -> crate::Result<Vec<u8>> {
let mut serializer = Serializer::new(vec![]);
serializer.serialize_decimal(*self)?;
Ok(serializer.into_inner())
}
pub fn from_slice(bytes: &[u8]) -> crate::Result<Self> {
let mut deserializer = Deserializer::new(bytes);
let t = deserializer.deserialize_decimal()?;
if !deserializer.has_remaining() {
Ok(t)
} else {
Err(Error::TrailingCharacters)
}
}
}
impl From<rust_decimal::Decimal> for Decimal {
fn from(decimal: rust_decimal::Decimal) -> Self {
Decimal::Normalized(decimal)
}
}
impl FromStr for Decimal {
type Err = rust_decimal::Error;
fn from_str(s: &str) -> Result<Self, Self::Err> {
match s {
"nan" | "NaN" => Ok(Decimal::NaN),
"-inf" | "-Inf" => Ok(Decimal::NegInf),
"inf" | "Inf" => Ok(Decimal::Inf),
_ => Ok(Decimal::Normalized(s.parse()?)),
}
}
}
impl Display for Decimal {
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
match self {
Decimal::NaN => write!(f, "NaN"),
Decimal::NegInf => write!(f, "-Inf"),
Decimal::Inf => write!(f, "Inf"),
Decimal::Normalized(n) => write!(f, "{}", n),
}
}
}