poem_openapi/types/external/
decimal.rs

1use std::borrow::Cow;
2
3use poem::{http::HeaderValue, web::Field};
4use rust_decimal::Decimal;
5use serde_json::Value;
6
7use crate::{
8    registry::{MetaSchema, MetaSchemaRef},
9    types::{
10        ParseError, ParseFromJSON, ParseFromMultipartField, ParseFromParameter, ParseResult,
11        ToHeader, ToJSON, Type,
12    },
13};
14
15impl Type for Decimal {
16    const IS_REQUIRED: bool = true;
17
18    type RawValueType = Self;
19
20    type RawElementValueType = Self;
21
22    fn name() -> Cow<'static, str> {
23        "string_decimal".into()
24    }
25
26    fn schema_ref() -> MetaSchemaRef {
27        MetaSchemaRef::Inline(Box::new(MetaSchema::new_with_format("string", "decimal")))
28    }
29
30    fn as_raw_value(&self) -> Option<&Self::RawValueType> {
31        Some(self)
32    }
33
34    fn raw_element_iter<'a>(
35        &'a self,
36    ) -> Box<dyn Iterator<Item = &'a Self::RawElementValueType> + 'a> {
37        Box::new(self.as_raw_value().into_iter())
38    }
39}
40
41impl ParseFromJSON for Decimal {
42    fn parse_from_json(value: Option<Value>) -> ParseResult<Self> {
43        let value = value.unwrap_or_default();
44        match value {
45            Value::String(value) => Ok(value.parse()?),
46            Value::Number(num) if num.is_i64() => Ok(Decimal::from(
47                num.as_i64()
48                    .ok_or_else(|| ParseError::custom("Expected a number"))?,
49            )),
50            Value::Number(num) if num.is_u64() => Ok(Decimal::from(
51                num.as_u64()
52                    .ok_or_else(|| ParseError::custom("Expected a number"))?,
53            )),
54            Value::Number(num) if num.is_f64() => Ok(Decimal::try_from(
55                num.as_f64()
56                    .ok_or_else(|| ParseError::custom("Expected a float"))?,
57            )
58            .map_err(|_| ParseError::custom("Float out of range"))?),
59            _ => Err(ParseError::expected_type(value)),
60        }
61    }
62}
63
64impl ParseFromParameter for Decimal {
65    fn parse_from_parameter(value: &str) -> ParseResult<Self> {
66        value.parse().map_err(ParseError::custom)
67    }
68}
69
70impl ParseFromMultipartField for Decimal {
71    async fn parse_from_multipart(field: Option<Field>) -> ParseResult<Self> {
72        match field {
73            Some(field) => Ok(field.text().await?.parse()?),
74            None => Err(ParseError::expected_input()),
75        }
76    }
77}
78
79impl ToJSON for Decimal {
80    fn to_json(&self) -> Option<Value> {
81        Some(Value::String(self.normalize().to_string()))
82    }
83}
84
85impl ToHeader for Decimal {
86    fn to_header(&self) -> Option<HeaderValue> {
87        HeaderValue::from_str(&self.normalize().to_string()).ok()
88    }
89}
poem_openapi/types/external/decimal.rs

poem_openapi/types/external/
decimal.rs
