Implement `FromStr` for `Decimal`/`UDecimal`.
Signed-off-by: Jean Pierre Dudey <jeandudey@hotmail.com>
This commit is contained in:
parent
2d961412af
commit
07838568f9
|
@ -22,9 +22,11 @@
|
||||||
//!
|
//!
|
||||||
|
|
||||||
use std::{fmt, ops};
|
use std::{fmt, ops};
|
||||||
|
#[cfg(feature = "serde-decimal")] use std::error;
|
||||||
|
#[cfg(feature = "serde-decimal")] use std::str::FromStr;
|
||||||
|
|
||||||
#[cfg(feature = "serde-decimal")] use serde;
|
#[cfg(feature = "serde-decimal")] use serde;
|
||||||
#[cfg(feature = "serde-decimal")] use strason::Json;
|
#[cfg(feature = "serde-decimal")] use strason::{self, Json};
|
||||||
|
|
||||||
/// A fixed-point decimal type
|
/// A fixed-point decimal type
|
||||||
#[derive(Copy, Clone, Debug, Eq, Ord)]
|
#[derive(Copy, Clone, Debug, Eq, Ord)]
|
||||||
|
@ -126,23 +128,10 @@ impl Decimal {
|
||||||
/// Returns whether or not the number is nonnegative
|
/// Returns whether or not the number is nonnegative
|
||||||
#[inline]
|
#[inline]
|
||||||
pub fn nonnegative(&self) -> bool { self.mantissa >= 0 }
|
pub fn nonnegative(&self) -> bool { self.mantissa >= 0 }
|
||||||
}
|
|
||||||
|
|
||||||
|
// Converts a JSON number to a Decimal previously parsed by strason
|
||||||
#[cfg(feature = "serde-decimal")]
|
#[cfg(feature = "serde-decimal")]
|
||||||
impl<'de> serde::Deserialize<'de> for Decimal {
|
fn parse_decimal(s: &str) -> Decimal {
|
||||||
/// Deserialize a `Decimal`.
|
|
||||||
///
|
|
||||||
/// This type is deserialized through [`strason`][1] for the same reason as
|
|
||||||
/// it's explained in the `Serialize` implementation.
|
|
||||||
///
|
|
||||||
/// [1]: https://github.com/apoelstra/strason
|
|
||||||
fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
|
|
||||||
where
|
|
||||||
D: serde::Deserializer<'de>,
|
|
||||||
{
|
|
||||||
let json = Json::deserialize(deserializer)?;
|
|
||||||
match json.num() {
|
|
||||||
Some(s) => {
|
|
||||||
// We know this will be a well-formed Json number, so we can
|
// We know this will be a well-formed Json number, so we can
|
||||||
// be pretty lax about parsing
|
// be pretty lax about parsing
|
||||||
let mut negative = false;
|
let mut negative = false;
|
||||||
|
@ -162,14 +151,43 @@ impl<'de> serde::Deserialize<'de> for Decimal {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
if negative { mantissa *= -1; }
|
if negative { mantissa *= -1; }
|
||||||
Ok(Decimal {
|
Decimal {
|
||||||
mantissa: mantissa,
|
mantissa: mantissa,
|
||||||
exponent: exponent,
|
exponent: exponent,
|
||||||
})
|
|
||||||
}
|
}
|
||||||
None => Err(serde::de::Error::custom("expected decimal, got non-numeric"))
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[cfg(feature = "serde-decimal")]
|
||||||
|
impl FromStr for Decimal {
|
||||||
|
type Err = ParseDecimalError;
|
||||||
|
|
||||||
|
/// Parses a `Decimal` from the given amount string.
|
||||||
|
fn from_str(s: &str) -> Result<Self, Self::Err> {
|
||||||
|
Json::from_str(s)?
|
||||||
|
.num()
|
||||||
|
.map(Decimal::parse_decimal)
|
||||||
|
.ok_or(ParseDecimalError::NotANumber)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(feature = "serde-decimal")]
|
||||||
|
impl<'de> serde::Deserialize<'de> for Decimal {
|
||||||
|
/// Deserialize a `Decimal`.
|
||||||
|
///
|
||||||
|
/// This type is deserialized through [`strason`][1] for the same reason as
|
||||||
|
/// it's explained in the `Serialize` implementation.
|
||||||
|
///
|
||||||
|
/// [1]: https://github.com/apoelstra/strason
|
||||||
|
fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
|
||||||
|
where
|
||||||
|
D: serde::Deserializer<'de>,
|
||||||
|
{
|
||||||
|
Json::deserialize(deserializer)?
|
||||||
|
.num()
|
||||||
|
.map(Decimal::parse_decimal)
|
||||||
|
.ok_or(serde::de::Error::custom("expected decimal, got non-numeric"))
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
#[cfg(feature = "serde-decimal")]
|
#[cfg(feature = "serde-decimal")]
|
||||||
|
@ -260,23 +278,10 @@ impl UDecimal {
|
||||||
self.mantissa * 10u64.pow((exponent - self.exponent) as u32)
|
self.mantissa * 10u64.pow((exponent - self.exponent) as u32)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
|
||||||
|
|
||||||
|
// Converts a JSON number to a Decimal previously parsed by strason
|
||||||
#[cfg(feature = "serde-decimal")]
|
#[cfg(feature = "serde-decimal")]
|
||||||
impl<'de> serde::Deserialize<'de> for UDecimal {
|
fn parse_udecimal(s: &str) -> UDecimal {
|
||||||
/// Deserialize an `UDecimal`.
|
|
||||||
///
|
|
||||||
/// This type is deserialized through [`strason`][1] for the same reason as
|
|
||||||
/// it's explained in the `Serialize` implementation.
|
|
||||||
///
|
|
||||||
/// [1]: https://github.com/apoelstra/strason
|
|
||||||
fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
|
|
||||||
where
|
|
||||||
D: serde::Deserializer<'de>,
|
|
||||||
{
|
|
||||||
let json = Json::deserialize(deserializer)?;
|
|
||||||
match json.num() {
|
|
||||||
Some(s) => {
|
|
||||||
// We know this will be a well-formed Json number, so we can
|
// We know this will be a well-formed Json number, so we can
|
||||||
// be pretty lax about parsing
|
// be pretty lax about parsing
|
||||||
let mut past_dec = false;
|
let mut past_dec = false;
|
||||||
|
@ -293,14 +298,43 @@ impl<'de> serde::Deserialize<'de> for UDecimal {
|
||||||
_ => { /* whitespace or something, just ignore it */ }
|
_ => { /* whitespace or something, just ignore it */ }
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
Ok(UDecimal {
|
UDecimal {
|
||||||
mantissa: mantissa,
|
mantissa: mantissa,
|
||||||
exponent: exponent,
|
exponent: exponent,
|
||||||
})
|
|
||||||
}
|
}
|
||||||
None => Err(serde::de::Error::custom("expected decimal, got non-numeric"))
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[cfg(feature = "serde-decimal")]
|
||||||
|
impl FromStr for UDecimal {
|
||||||
|
type Err = ParseDecimalError;
|
||||||
|
|
||||||
|
/// Parses a `UDecimal` from the given amount string.
|
||||||
|
fn from_str(s: &str) -> Result<Self, Self::Err> {
|
||||||
|
Json::from_str(s)?
|
||||||
|
.num()
|
||||||
|
.map(UDecimal::parse_udecimal)
|
||||||
|
.ok_or(ParseDecimalError::NotANumber)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(feature = "serde-decimal")]
|
||||||
|
impl<'de> serde::Deserialize<'de> for UDecimal {
|
||||||
|
/// Deserialize an `UDecimal`.
|
||||||
|
///
|
||||||
|
/// This type is deserialized through [`strason`][1] for the same reason as
|
||||||
|
/// it's explained in the `Serialize` implementation.
|
||||||
|
///
|
||||||
|
/// [1]: https://github.com/apoelstra/strason
|
||||||
|
fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
|
||||||
|
where
|
||||||
|
D: serde::Deserializer<'de>,
|
||||||
|
{
|
||||||
|
Json::deserialize(deserializer)?
|
||||||
|
.num()
|
||||||
|
.map(UDecimal::parse_udecimal)
|
||||||
|
.ok_or(serde::de::Error::custom("expected decimal, got non-numeric"))
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
#[cfg(feature = "serde-decimal")]
|
#[cfg(feature = "serde-decimal")]
|
||||||
|
@ -321,6 +355,51 @@ impl serde::Serialize for UDecimal {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Errors that occur during `Decimal`/`UDecimal` parsing.
|
||||||
|
#[cfg(feature = "serde-decimal")]
|
||||||
|
#[derive(Debug)]
|
||||||
|
pub enum ParseDecimalError {
|
||||||
|
/// An error ocurred while parsing the JSON number.
|
||||||
|
Json(strason::Error),
|
||||||
|
/// Not a number.
|
||||||
|
NotANumber,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(feature = "serde-decimal")]
|
||||||
|
#[doc(hidden)]
|
||||||
|
impl From<strason::Error> for ParseDecimalError {
|
||||||
|
fn from(e: strason::Error) -> ParseDecimalError {
|
||||||
|
ParseDecimalError::Json(e)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(feature = "serde-decimal")]
|
||||||
|
impl fmt::Display for ParseDecimalError {
|
||||||
|
fn fmt(&self, fmt: &mut fmt::Formatter) -> fmt::Result {
|
||||||
|
match *self {
|
||||||
|
ParseDecimalError::Json(ref e) => fmt::Display::fmt(e, fmt),
|
||||||
|
ParseDecimalError::NotANumber => fmt.write_str("not a valid JSON number")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(feature = "serde-decimal")]
|
||||||
|
impl error::Error for ParseDecimalError {
|
||||||
|
fn description(&self) -> &str {
|
||||||
|
match *self {
|
||||||
|
ParseDecimalError::Json(ref e) => e.description(),
|
||||||
|
ParseDecimalError::NotANumber => "not a valid JSON number",
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn cause(&self) -> Option<&error::Error> {
|
||||||
|
match *self {
|
||||||
|
ParseDecimalError::Json(ref e) => Some(e),
|
||||||
|
_ => None,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
mod tests {
|
mod tests {
|
||||||
use super::*;
|
use super::*;
|
||||||
|
@ -468,4 +547,20 @@ mod tests {
|
||||||
let dec: UDecimal = json.into_deserialize().unwrap();
|
let dec: UDecimal = json.into_deserialize().unwrap();
|
||||||
assert_eq!(dec, UDecimal::new(98000, 7));
|
assert_eq!(dec, UDecimal::new(98000, 7));
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
#[cfg(feature = "serde-decimal")]
|
||||||
|
fn parse_decimal_udecimal() {
|
||||||
|
let dec = "0.00980000".parse::<Decimal>().unwrap();
|
||||||
|
assert_eq!(dec, Decimal::new(980000, 8));
|
||||||
|
|
||||||
|
let dec = "0.00980000".parse::<UDecimal>().unwrap();
|
||||||
|
assert_eq!(dec, UDecimal::new(980000, 8));
|
||||||
|
|
||||||
|
let dec = "0.00980".parse::<Decimal>().unwrap();
|
||||||
|
assert_eq!(dec, Decimal::new(98000, 7));
|
||||||
|
|
||||||
|
let dec = "0.00980".parse::<UDecimal>().unwrap();
|
||||||
|
assert_eq!(dec, UDecimal::new(98000, 7));
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue