1use std::str::FromStr;
26
27use bytes::Bytes;
28use rust_decimal::Decimal;
29use serde::{
30 Deserialize, Deserializer, Serialize, Serializer,
31 de::{Error, Unexpected, Visitor},
32 ser::SerializeSeq,
33};
34use ustr::Ustr;
35
36pub mod sorted_hashset {
40 use ahash::AHashSet;
41 use serde::{Deserialize, Deserializer, Serialize, Serializer};
42
43 pub fn serialize<T, S>(set: &AHashSet<T>, s: S) -> Result<S::Ok, S::Error>
50 where
51 T: Serialize + Ord,
52 S: Serializer,
53 {
54 let mut sorted: Vec<&T> = set.iter().collect();
55 sorted.sort_unstable();
56 sorted.serialize(s)
57 }
58
59 pub fn deserialize<'de, T, D>(d: D) -> Result<AHashSet<T>, D::Error>
66 where
67 T: Deserialize<'de> + Eq + std::hash::Hash,
68 D: Deserializer<'de>,
69 {
70 let vec = Vec::<T>::deserialize(d)?;
71 Ok(vec.into_iter().collect())
72 }
73}
74
75struct DecimalVisitor;
80
81impl Visitor<'_> for DecimalVisitor {
82 type Value = Decimal;
83
84 fn expecting(&self, formatter: &mut std::fmt::Formatter) -> std::fmt::Result {
85 formatter.write_str("a decimal number as string, integer, or float")
86 }
87
88 fn visit_str<E: Error>(self, v: &str) -> Result<Self::Value, E> {
90 if v.is_empty() {
91 return Ok(Decimal::ZERO);
92 }
93 parse_decimal_str(v).map_err(E::custom)
94 }
95
96 fn visit_string<E: Error>(self, v: String) -> Result<Self::Value, E> {
98 self.visit_str(&v)
99 }
100
101 fn visit_i64<E: Error>(self, v: i64) -> Result<Self::Value, E> {
103 Ok(Decimal::from(v))
104 }
105
106 fn visit_u64<E: Error>(self, v: u64) -> Result<Self::Value, E> {
107 Ok(Decimal::from(v))
108 }
109
110 fn visit_i128<E: Error>(self, v: i128) -> Result<Self::Value, E> {
111 Ok(Decimal::from(v))
112 }
113
114 fn visit_u128<E: Error>(self, v: u128) -> Result<Self::Value, E> {
115 Ok(Decimal::from(v))
116 }
117
118 fn visit_f64<E: Error>(self, v: f64) -> Result<Self::Value, E> {
120 if !v.is_finite() {
121 return Err(E::invalid_value(Unexpected::Float(v), &self));
122 }
123 Decimal::try_from(v).map_err(E::custom)
124 }
125
126 fn visit_unit<E: Error>(self) -> Result<Self::Value, E> {
128 Ok(Decimal::ZERO)
129 }
130
131 fn visit_none<E: Error>(self) -> Result<Self::Value, E> {
132 Ok(Decimal::ZERO)
133 }
134}
135
136struct OptionalDecimalVisitor;
141
142impl Visitor<'_> for OptionalDecimalVisitor {
143 type Value = Option<Decimal>;
144
145 fn expecting(&self, formatter: &mut std::fmt::Formatter) -> std::fmt::Result {
146 formatter.write_str("null or a decimal number as string, integer, or float")
147 }
148
149 fn visit_str<E: Error>(self, v: &str) -> Result<Self::Value, E> {
152 if v.is_empty() {
153 return Ok(None);
154 }
155 DecimalVisitor.visit_str(v).map(Some)
156 }
157
158 fn visit_string<E: Error>(self, v: String) -> Result<Self::Value, E> {
159 self.visit_str(&v)
160 }
161
162 fn visit_i64<E: Error>(self, v: i64) -> Result<Self::Value, E> {
163 DecimalVisitor.visit_i64(v).map(Some)
164 }
165
166 fn visit_u64<E: Error>(self, v: u64) -> Result<Self::Value, E> {
167 DecimalVisitor.visit_u64(v).map(Some)
168 }
169
170 fn visit_i128<E: Error>(self, v: i128) -> Result<Self::Value, E> {
171 DecimalVisitor.visit_i128(v).map(Some)
172 }
173
174 fn visit_u128<E: Error>(self, v: u128) -> Result<Self::Value, E> {
175 DecimalVisitor.visit_u128(v).map(Some)
176 }
177
178 fn visit_f64<E: Error>(self, v: f64) -> Result<Self::Value, E> {
179 DecimalVisitor.visit_f64(v).map(Some)
180 }
181
182 fn visit_unit<E: Error>(self) -> Result<Self::Value, E> {
184 Ok(None)
185 }
186
187 fn visit_none<E: Error>(self) -> Result<Self::Value, E> {
188 Ok(None)
189 }
190}
191
192fn parse_decimal_str(value: &str) -> Result<Decimal, String> {
193 let parsed = if value.contains('e') || value.contains('E') {
194 Decimal::from_scientific(value)
195 } else {
196 Decimal::from_str(value)
197 };
198
199 match parsed {
200 Ok(decimal) => Ok(decimal),
201 Err(e) => {
202 for scale in (0..=Decimal::MAX_SCALE as usize).rev() {
206 let clamped =
207 decimal_string_clamped_to_scale(value, scale).ok_or_else(|| e.to_string())?;
208
209 if let Ok(decimal) = Decimal::from_str(&clamped) {
210 return Ok(decimal);
211 }
212 }
213 Err(e.to_string())
214 }
215 }
216}
217
218fn decimal_string_clamped_to_scale(value: &str, max_scale: usize) -> Option<String> {
219 let (coefficient, exponent) = match value.find(['e', 'E']) {
220 Some(index) => {
221 let exponent = value[index + 1..].parse::<i32>().ok()?;
222 (&value[..index], exponent)
223 }
224 None => (value, 0),
225 };
226
227 let (sign, unsigned) = match coefficient.as_bytes().first()? {
228 b'+' => ("", &coefficient[1..]),
229 b'-' => ("-", &coefficient[1..]),
230 _ => ("", coefficient),
231 };
232 let (integer, fractional) = decimal_components(unsigned)?;
233 let digits = format!("{integer}{fractional}");
234 if decimal_digits_are_zero(&digits, "") {
235 return Some("0".to_string());
236 }
237
238 let point = i32::try_from(integer.len()).ok()?.checked_add(exponent)?;
239
240 if decimal_integer_digits_exceed_max(&digits, point) {
245 return None;
246 }
247
248 let (integer, fractional) = if point <= 0 {
249 let zero_count = usize::try_from(-point).ok()?.min(max_scale + 1);
252 (
253 "0".to_string(),
254 format!("{}{digits}", "0".repeat(zero_count)),
255 )
256 } else {
257 let point = usize::try_from(point).ok()?;
258 if point >= digits.len() {
259 (
260 format!("{}{}", digits, "0".repeat(point - digits.len())),
261 String::new(),
262 )
263 } else {
264 (digits[..point].to_string(), digits[point..].to_string())
265 }
266 };
267
268 let (integer, fractional) = round_decimal_components(integer, fractional, max_scale);
269 let sign = if sign == "-" && decimal_digits_are_zero(&integer, &fractional) {
270 ""
271 } else {
272 sign
273 };
274
275 if fractional.is_empty() {
276 Some(format!("{sign}{integer}"))
277 } else {
278 Some(format!("{sign}{integer}.{fractional}"))
279 }
280}
281
282fn decimal_components(value: &str) -> Option<(&str, &str)> {
283 let mut split = value.split('.');
284 let integer = split.next()?;
285 let fractional = split.next().unwrap_or("");
286 if split.next().is_some()
287 || (integer.is_empty() && fractional.is_empty())
288 || !integer.chars().all(|c| c.is_ascii_digit())
289 || !fractional.chars().all(|c| c.is_ascii_digit())
290 {
291 return None;
292 }
293 Some((integer, fractional))
294}
295
296fn decimal_integer_digits_exceed_max(digits: &str, point: i32) -> bool {
297 const DECIMAL_MAX_INTEGER_DIGITS: i32 = 29;
298
299 if point <= 0 {
300 return false;
301 }
302
303 let Some(first_non_zero) = digits.bytes().position(|digit| digit != b'0') else {
304 return false;
305 };
306 let Ok(first_non_zero) = i32::try_from(first_non_zero) else {
307 return true;
308 };
309
310 point.saturating_sub(first_non_zero) > DECIMAL_MAX_INTEGER_DIGITS
311}
312
313fn round_decimal_components(
314 mut integer: String,
315 fractional: String,
316 max_scale: usize,
317) -> (String, String) {
318 if fractional.len() <= max_scale {
319 return (integer, fractional);
320 }
321
322 let mut rounded = fractional.as_bytes()[..max_scale].to_vec();
323 if fractional.as_bytes()[max_scale] >= b'5' {
324 increment_decimal_digits(&mut integer, &mut rounded);
325 }
326
327 (
328 integer,
329 String::from_utf8(rounded).expect("decimal digits are ASCII"),
330 )
331}
332
333fn increment_decimal_digits(integer: &mut String, fractional: &mut [u8]) {
334 for digit in fractional.iter_mut().rev() {
335 if *digit < b'9' {
336 *digit += 1;
337 return;
338 }
339 *digit = b'0';
340 }
341
342 let mut integer_digits = integer.as_bytes().to_vec();
343 for digit in integer_digits.iter_mut().rev() {
344 if *digit < b'9' {
345 *digit += 1;
346 *integer = String::from_utf8(integer_digits).expect("decimal digits are ASCII");
347 return;
348 }
349 *digit = b'0';
350 }
351 integer_digits.insert(0, b'1');
352 *integer = String::from_utf8(integer_digits).expect("decimal digits are ASCII");
353}
354
355fn decimal_digits_are_zero(integer: &str, fractional: &str) -> bool {
356 integer
357 .bytes()
358 .chain(fractional.bytes())
359 .all(|digit| digit == b'0')
360}
361
362pub trait Serializable: Serialize + for<'de> Deserialize<'de> {
364 fn from_json_bytes(data: &[u8]) -> Result<Self, serde_json::Error> {
370 serde_json::from_slice(data)
371 }
372
373 fn to_json_bytes(&self) -> Result<Bytes, serde_json::Error> {
379 serde_json::to_vec(self).map(Bytes::from)
380 }
381}
382
383pub use self::msgpack::{FromMsgPack, MsgPackSerializable, ToMsgPack};
384
385pub mod msgpack {
390 use bytes::Bytes;
391 use serde::{Deserialize, Serialize};
392
393 use super::Serializable;
394
395 pub trait FromMsgPack: for<'de> Deserialize<'de> + Sized {
397 fn from_msgpack_bytes(data: &[u8]) -> Result<Self, rmp_serde::decode::Error> {
403 rmp_serde::from_slice(data)
404 }
405 }
406
407 pub trait ToMsgPack: Serialize {
409 fn to_msgpack_bytes(&self) -> Result<Bytes, rmp_serde::encode::Error> {
415 rmp_serde::to_vec_named(self).map(Bytes::from)
416 }
417 }
418
419 pub trait MsgPackSerializable: Serializable + FromMsgPack + ToMsgPack {}
423
424 impl<T> FromMsgPack for T where T: Serializable {}
425
426 impl<T> ToMsgPack for T where T: Serializable {}
427
428 impl<T> MsgPackSerializable for T where T: Serializable {}
429}
430
431#[must_use]
435pub const fn default_true() -> bool {
436 true
437}
438
439#[must_use]
443pub const fn default_false() -> bool {
444 false
445}
446
447pub fn deserialize_decimal<'de, D>(deserializer: D) -> Result<Decimal, D::Error>
470where
471 D: Deserializer<'de>,
472{
473 deserializer.deserialize_any(DecimalVisitor)
474}
475
476pub fn deserialize_optional_decimal<'de, D>(deserializer: D) -> Result<Option<Decimal>, D::Error>
500where
501 D: Deserializer<'de>,
502{
503 deserializer.deserialize_any(OptionalDecimalVisitor)
506}
507
508pub fn serialize_decimal<S: Serializer>(d: &Decimal, s: S) -> Result<S::Ok, S::Error> {
516 rust_decimal::serde::float::serialize(d, s)
517}
518
519pub fn serialize_optional_decimal<S: Serializer>(
525 d: &Option<Decimal>,
526 s: S,
527) -> Result<S::Ok, S::Error> {
528 match d {
529 Some(decimal) => rust_decimal::serde::float::serialize(decimal, s),
530 None => s.serialize_none(),
531 }
532}
533
534pub fn deserialize_decimal_from_str<'de, D>(deserializer: D) -> Result<Decimal, D::Error>
543where
544 D: Deserializer<'de>,
545{
546 let s: std::borrow::Cow<'de, str> = Deserialize::deserialize(deserializer)?;
547 Decimal::from_str(s.as_ref()).map_err(D::Error::custom)
548}
549
550pub fn deserialize_decimal_or_zero<'de, D>(deserializer: D) -> Result<Decimal, D::Error>
558where
559 D: Deserializer<'de>,
560{
561 let s: std::borrow::Cow<'de, str> = Deserialize::deserialize(deserializer)?;
562 if s.is_empty() || s == "0" {
563 Ok(Decimal::ZERO)
564 } else {
565 Decimal::from_str(s.as_ref()).map_err(D::Error::custom)
566 }
567}
568
569pub fn deserialize_optional_decimal_str<'de, D>(
579 deserializer: D,
580) -> Result<Option<Decimal>, D::Error>
581where
582 D: Deserializer<'de>,
583{
584 let s: std::borrow::Cow<'de, str> = Deserialize::deserialize(deserializer)?;
585 if s.is_empty() || s == "0" {
586 Ok(None)
587 } else {
588 Decimal::from_str(s.as_ref())
589 .map(Some)
590 .map_err(D::Error::custom)
591 }
592}
593
594pub fn deserialize_optional_decimal_from_str<'de, D>(
603 deserializer: D,
604) -> Result<Option<Decimal>, D::Error>
605where
606 D: Deserializer<'de>,
607{
608 let opt = Option::<String>::deserialize(deserializer)?;
609 match opt {
610 Some(s) if !s.is_empty() => Decimal::from_str(&s).map(Some).map_err(D::Error::custom),
611 _ => Ok(None),
612 }
613}
614
615pub fn deserialize_optional_decimal_or_zero<'de, D>(deserializer: D) -> Result<Decimal, D::Error>
623where
624 D: Deserializer<'de>,
625{
626 let opt: Option<String> = Deserialize::deserialize(deserializer)?;
627 match opt {
628 None => Ok(Decimal::ZERO),
629 Some(s) if s.is_empty() || s == "0" => Ok(Decimal::ZERO),
630 Some(s) => Decimal::from_str(&s).map_err(D::Error::custom),
631 }
632}
633
634pub fn deserialize_vec_decimal_from_str<'de, D>(deserializer: D) -> Result<Vec<Decimal>, D::Error>
640where
641 D: Deserializer<'de>,
642{
643 let strings = Vec::<String>::deserialize(deserializer)?;
644 strings
645 .into_iter()
646 .map(|s| Decimal::from_str(&s).map_err(D::Error::custom))
647 .collect()
648}
649
650pub fn serialize_decimal_as_str<S>(decimal: &Decimal, serializer: S) -> Result<S::Ok, S::Error>
656where
657 S: Serializer,
658{
659 serializer.serialize_str(&decimal.to_string())
660}
661
662pub fn serialize_optional_decimal_as_str<S>(
668 decimal: &Option<Decimal>,
669 serializer: S,
670) -> Result<S::Ok, S::Error>
671where
672 S: Serializer,
673{
674 match decimal {
675 Some(d) => serializer.serialize_str(&d.to_string()),
676 None => serializer.serialize_none(),
677 }
678}
679
680pub fn serialize_vec_decimal_as_str<S>(
686 decimals: &Vec<Decimal>,
687 serializer: S,
688) -> Result<S::Ok, S::Error>
689where
690 S: Serializer,
691{
692 let mut seq = serializer.serialize_seq(Some(decimals.len()))?;
693 for decimal in decimals {
694 seq.serialize_element(&decimal.to_string())?;
695 }
696 seq.end()
697}
698
699pub fn parse_decimal(s: &str) -> anyhow::Result<Decimal> {
705 Decimal::from_str(s).map_err(|e| anyhow::anyhow!("Failed to parse decimal from '{s}': {e}"))
706}
707
708pub fn parse_optional_decimal(s: &Option<String>) -> anyhow::Result<Option<Decimal>> {
714 match s {
715 None => Ok(None),
716 Some(s) if s.is_empty() => Ok(None),
717 Some(s) => parse_decimal(s).map(Some),
718 }
719}
720
721pub fn deserialize_empty_string_as_none<'de, D>(deserializer: D) -> Result<Option<String>, D::Error>
733where
734 D: Deserializer<'de>,
735{
736 let opt = Option::<String>::deserialize(deserializer)?;
737 Ok(opt.filter(|s| !s.is_empty()))
738}
739
740pub fn deserialize_empty_ustr_as_none<'de, D>(deserializer: D) -> Result<Option<Ustr>, D::Error>
746where
747 D: Deserializer<'de>,
748{
749 let opt = Option::<Ustr>::deserialize(deserializer)?;
750 Ok(opt.filter(|s| !s.is_empty()))
751}
752
753pub fn deserialize_string_to_u8<'de, D>(deserializer: D) -> Result<u8, D::Error>
761where
762 D: Deserializer<'de>,
763{
764 let s: std::borrow::Cow<'de, str> = Deserialize::deserialize(deserializer)?;
765 if s.is_empty() {
766 return Ok(0);
767 }
768 s.as_ref().parse::<u8>().map_err(D::Error::custom)
769}
770
771pub fn deserialize_string_to_u64<'de, D>(deserializer: D) -> Result<u64, D::Error>
779where
780 D: Deserializer<'de>,
781{
782 let s: std::borrow::Cow<'de, str> = Deserialize::deserialize(deserializer)?;
783 if s.is_empty() {
784 Ok(0)
785 } else {
786 s.as_ref().parse::<u64>().map_err(D::Error::custom)
787 }
788}
789
790pub fn deserialize_optional_string_to_u64<'de, D>(deserializer: D) -> Result<Option<u64>, D::Error>
798where
799 D: Deserializer<'de>,
800{
801 let s: Option<String> = Option::deserialize(deserializer)?;
802 match s {
803 Some(s) if s.is_empty() => Ok(None),
804 Some(s) => s.parse().map(Some).map_err(D::Error::custom),
805 None => Ok(None),
806 }
807}
808
809#[cfg(test)]
810mod tests {
811 use ahash::AHashSet;
812 use rstest::*;
813 use rust_decimal::Decimal;
814 use rust_decimal_macros::dec;
815 use serde::{Deserialize, Serialize};
816 use serde_json::json;
817 use ustr::Ustr;
818
819 use super::{
820 Serializable, deserialize_decimal, deserialize_decimal_from_str,
821 deserialize_decimal_or_zero, deserialize_empty_string_as_none,
822 deserialize_empty_ustr_as_none, deserialize_optional_decimal,
823 deserialize_optional_decimal_or_zero, deserialize_optional_decimal_str,
824 deserialize_optional_string_to_u64, deserialize_string_to_u8, deserialize_string_to_u64,
825 deserialize_vec_decimal_from_str,
826 msgpack::{FromMsgPack, ToMsgPack},
827 parse_decimal, parse_optional_decimal, serialize_decimal, serialize_decimal_as_str,
828 serialize_optional_decimal, serialize_optional_decimal_as_str,
829 serialize_vec_decimal_as_str, sorted_hashset,
830 };
831
832 #[derive(Debug, PartialEq, Serialize, Deserialize)]
833 struct SortedSetPayload {
834 #[serde(with = "sorted_hashset")]
835 values: AHashSet<i32>,
836 }
837
838 #[derive(Serialize, Deserialize, PartialEq, Debug)]
839 struct SerializableTestStruct {
840 id: u32,
841 name: String,
842 value: f64,
843 }
844
845 impl Serializable for SerializableTestStruct {}
846
847 #[rstest]
848 fn test_sorted_hashset_serialization_is_deterministic() {
849 let payload = SortedSetPayload {
850 values: [29, 3, 17, 5, 23, 11].into_iter().collect(),
851 };
852
853 let encoded = serde_json::to_string(&payload).unwrap();
854 let expected = serde_json::to_string(&json!({ "values": [3, 5, 11, 17, 23, 29] })).unwrap();
855 let restored: SortedSetPayload = serde_json::from_str(&encoded).unwrap();
856
857 assert_eq!(encoded, expected);
858 assert_eq!(restored, payload);
859 }
860
861 #[rstest]
862 fn test_serializable_json_roundtrip() {
863 let original = SerializableTestStruct {
864 id: 42,
865 name: "test".to_string(),
866 value: std::f64::consts::PI,
867 };
868
869 let json_bytes = original.to_json_bytes().unwrap();
870 let deserialized = SerializableTestStruct::from_json_bytes(&json_bytes).unwrap();
871
872 assert_eq!(original, deserialized);
873 }
874
875 #[rstest]
876 fn test_serializable_msgpack_roundtrip() {
877 let original = SerializableTestStruct {
878 id: 123,
879 name: "msgpack_test".to_string(),
880 value: std::f64::consts::E,
881 };
882
883 let msgpack_bytes = original.to_msgpack_bytes().unwrap();
884 let deserialized = SerializableTestStruct::from_msgpack_bytes(&msgpack_bytes).unwrap();
885
886 assert_eq!(original, deserialized);
887 }
888
889 #[rstest]
890 fn test_serializable_json_invalid_data() {
891 let invalid_json = b"invalid json data";
892 let result = SerializableTestStruct::from_json_bytes(invalid_json);
893 assert!(result.is_err());
894 }
895
896 #[rstest]
897 fn test_serializable_msgpack_invalid_data() {
898 let invalid_msgpack = b"invalid msgpack data";
899 let result = SerializableTestStruct::from_msgpack_bytes(invalid_msgpack);
900 assert!(result.is_err());
901 }
902
903 #[rstest]
904 fn test_serializable_json_empty_values() {
905 let test_struct = SerializableTestStruct {
906 id: 0,
907 name: String::new(),
908 value: 0.0,
909 };
910
911 let json_bytes = test_struct.to_json_bytes().unwrap();
912 let deserialized = SerializableTestStruct::from_json_bytes(&json_bytes).unwrap();
913
914 assert_eq!(test_struct, deserialized);
915 }
916
917 #[rstest]
918 fn test_serializable_msgpack_empty_values() {
919 let test_struct = SerializableTestStruct {
920 id: 0,
921 name: String::new(),
922 value: 0.0,
923 };
924
925 let msgpack_bytes = test_struct.to_msgpack_bytes().unwrap();
926 let deserialized = SerializableTestStruct::from_msgpack_bytes(&msgpack_bytes).unwrap();
927
928 assert_eq!(test_struct, deserialized);
929 }
930
931 #[derive(Deserialize)]
932 struct TestOptionalDecimalStr {
933 #[serde(deserialize_with = "deserialize_optional_decimal_str")]
934 value: Option<Decimal>,
935 }
936
937 #[derive(Deserialize)]
938 struct TestDecimalOrZero {
939 #[serde(deserialize_with = "deserialize_decimal_or_zero")]
940 value: Decimal,
941 }
942
943 #[derive(Deserialize)]
944 struct TestOptionalDecimalOrZero {
945 #[serde(deserialize_with = "deserialize_optional_decimal_or_zero")]
946 value: Decimal,
947 }
948
949 #[derive(Serialize, Deserialize, PartialEq, Debug)]
950 struct TestDecimalRoundtrip {
951 #[serde(
952 serialize_with = "serialize_decimal_as_str",
953 deserialize_with = "deserialize_decimal_from_str"
954 )]
955 value: Decimal,
956 #[serde(
957 serialize_with = "serialize_optional_decimal_as_str",
958 deserialize_with = "super::deserialize_optional_decimal_from_str"
959 )]
960 optional_value: Option<Decimal>,
961 }
962
963 #[rstest]
964 #[case(r#"{"value":"123.45"}"#, Some(dec!(123.45)))]
965 #[case(r#"{"value":"0"}"#, None)]
966 #[case(r#"{"value":""}"#, None)]
967 fn test_deserialize_optional_decimal_str(
968 #[case] json: &str,
969 #[case] expected: Option<Decimal>,
970 ) {
971 let result: TestOptionalDecimalStr = serde_json::from_str(json).unwrap();
972 assert_eq!(result.value, expected);
973 }
974
975 #[rstest]
976 #[case(r#"{"value":"123.45"}"#, dec!(123.45))]
977 #[case(r#"{"value":"0"}"#, Decimal::ZERO)]
978 #[case(r#"{"value":""}"#, Decimal::ZERO)]
979 fn test_deserialize_decimal_or_zero(#[case] json: &str, #[case] expected: Decimal) {
980 let result: TestDecimalOrZero = serde_json::from_str(json).unwrap();
981 assert_eq!(result.value, expected);
982 }
983
984 #[rstest]
985 #[case(r#"{"value":"123.45"}"#, dec!(123.45))]
986 #[case(r#"{"value":"0"}"#, Decimal::ZERO)]
987 #[case(r#"{"value":null}"#, Decimal::ZERO)]
988 fn test_deserialize_optional_decimal_or_zero(#[case] json: &str, #[case] expected: Decimal) {
989 let result: TestOptionalDecimalOrZero = serde_json::from_str(json).unwrap();
990 assert_eq!(result.value, expected);
991 }
992
993 #[rstest]
994 fn test_decimal_serialization_roundtrip() {
995 let original = TestDecimalRoundtrip {
996 value: dec!(123.456789012345678),
997 optional_value: Some(dec!(0.000000001)),
998 };
999
1000 let json = serde_json::to_string(&original).unwrap();
1001
1002 assert!(json.contains("\"123.456789012345678\""));
1004 assert!(json.contains("\"0.000000001\""));
1005
1006 let deserialized: TestDecimalRoundtrip = serde_json::from_str(&json).unwrap();
1007 assert_eq!(original.value, deserialized.value);
1008 assert_eq!(original.optional_value, deserialized.optional_value);
1009 }
1010
1011 #[rstest]
1012 fn test_decimal_optional_none_handling() {
1013 let test_struct = TestDecimalRoundtrip {
1014 value: dec!(42.0),
1015 optional_value: None,
1016 };
1017
1018 let json = serde_json::to_string(&test_struct).unwrap();
1019 assert!(json.contains("null"));
1020
1021 let parsed: TestDecimalRoundtrip = serde_json::from_str(&json).unwrap();
1022 assert_eq!(test_struct.value, parsed.value);
1023 assert_eq!(None, parsed.optional_value);
1024 }
1025
1026 #[derive(Deserialize)]
1027 struct TestEmptyStringAsNone {
1028 #[serde(deserialize_with = "deserialize_empty_string_as_none")]
1029 value: Option<String>,
1030 }
1031
1032 #[rstest]
1033 #[case(r#"{"value":"hello"}"#, Some("hello".to_string()))]
1034 #[case(r#"{"value":""}"#, None)]
1035 #[case(r#"{"value":null}"#, None)]
1036 fn test_deserialize_empty_string_as_none(#[case] json: &str, #[case] expected: Option<String>) {
1037 let result: TestEmptyStringAsNone = serde_json::from_str(json).unwrap();
1038 assert_eq!(result.value, expected);
1039 }
1040
1041 #[derive(Deserialize)]
1042 struct TestEmptyUstrAsNone {
1043 #[serde(deserialize_with = "deserialize_empty_ustr_as_none")]
1044 value: Option<Ustr>,
1045 }
1046
1047 #[rstest]
1048 #[case(r#"{"value":"hello"}"#, Some(Ustr::from("hello")))]
1049 #[case(r#"{"value":""}"#, None)]
1050 #[case(r#"{"value":null}"#, None)]
1051 fn test_deserialize_empty_ustr_as_none(#[case] json: &str, #[case] expected: Option<Ustr>) {
1052 let result: TestEmptyUstrAsNone = serde_json::from_str(json).unwrap();
1053 assert_eq!(result.value, expected);
1054 }
1055
1056 #[derive(Serialize, Deserialize, PartialEq, Debug)]
1057 struct TestVecDecimal {
1058 #[serde(
1059 serialize_with = "serialize_vec_decimal_as_str",
1060 deserialize_with = "deserialize_vec_decimal_from_str"
1061 )]
1062 values: Vec<Decimal>,
1063 }
1064
1065 #[rstest]
1066 fn test_vec_decimal_roundtrip() {
1067 let original = TestVecDecimal {
1068 values: vec![dec!(1.5), dec!(2.25), dec!(100.001)],
1069 };
1070
1071 let json = serde_json::to_string(&original).unwrap();
1072 assert!(json.contains("[\"1.5\",\"2.25\",\"100.001\"]"));
1073
1074 let parsed: TestVecDecimal = serde_json::from_str(&json).unwrap();
1075 assert_eq!(original.values, parsed.values);
1076 }
1077
1078 #[rstest]
1079 fn test_vec_decimal_empty() {
1080 let original = TestVecDecimal { values: vec![] };
1081
1082 let json = serde_json::to_string(&original).unwrap();
1083 let parsed: TestVecDecimal = serde_json::from_str(&json).unwrap();
1084 assert_eq!(original.values, parsed.values);
1085 }
1086
1087 #[derive(Deserialize)]
1088 struct TestStringToU8 {
1089 #[serde(deserialize_with = "deserialize_string_to_u8")]
1090 value: u8,
1091 }
1092
1093 #[rstest]
1094 #[case(r#"{"value":"42"}"#, 42)]
1095 #[case(r#"{"value":"0"}"#, 0)]
1096 #[case(r#"{"value":"255"}"#, 255)]
1097 #[case(r#"{"value":""}"#, 0)]
1098 fn test_deserialize_string_to_u8(#[case] json: &str, #[case] expected: u8) {
1099 let result: TestStringToU8 = serde_json::from_str(json).unwrap();
1100 assert_eq!(result.value, expected);
1101 }
1102
1103 #[rstest]
1104 #[case(r#"{"value":"256"}"#)]
1105 #[case(r#"{"value":"999"}"#)]
1106 #[case(r#"{"value":"abc"}"#)]
1107 fn test_deserialize_string_to_u8_invalid(#[case] json: &str) {
1108 let result: Result<TestStringToU8, _> = serde_json::from_str(json);
1109 assert!(result.is_err());
1110 }
1111
1112 #[derive(Deserialize)]
1113 struct TestStringToU64 {
1114 #[serde(deserialize_with = "deserialize_string_to_u64")]
1115 value: u64,
1116 }
1117
1118 #[rstest]
1119 #[case(r#"{"value":"12345678901234"}"#, 12_345_678_901_234)]
1120 #[case(r#"{"value":"0"}"#, 0)]
1121 #[case(r#"{"value":"18446744073709551615"}"#, u64::MAX)]
1122 #[case(r#"{"value":""}"#, 0)]
1123 fn test_deserialize_string_to_u64(#[case] json: &str, #[case] expected: u64) {
1124 let result: TestStringToU64 = serde_json::from_str(json).unwrap();
1125 assert_eq!(result.value, expected);
1126 }
1127
1128 #[rstest]
1129 #[case(r#"{"value":"18446744073709551616"}"#)]
1130 #[case(r#"{"value":"abc"}"#)]
1131 #[case(r#"{"value":"-1"}"#)]
1132 fn test_deserialize_string_to_u64_invalid(#[case] json: &str) {
1133 let result: Result<TestStringToU64, _> = serde_json::from_str(json);
1134 assert!(result.is_err());
1135 }
1136
1137 #[derive(Deserialize)]
1138 struct TestOptionalStringToU64 {
1139 #[serde(deserialize_with = "deserialize_optional_string_to_u64")]
1140 value: Option<u64>,
1141 }
1142
1143 #[rstest]
1144 #[case(r#"{"value":"12345678901234"}"#, Some(12_345_678_901_234))]
1145 #[case(r#"{"value":"0"}"#, Some(0))]
1146 #[case(r#"{"value":""}"#, None)]
1147 #[case(r#"{"value":null}"#, None)]
1148 fn test_deserialize_optional_string_to_u64(#[case] json: &str, #[case] expected: Option<u64>) {
1149 let result: TestOptionalStringToU64 = serde_json::from_str(json).unwrap();
1150 assert_eq!(result.value, expected);
1151 }
1152
1153 #[rstest]
1154 #[case("123.45", dec!(123.45))]
1155 #[case("0", Decimal::ZERO)]
1156 #[case("0.0", Decimal::ZERO)]
1157 fn test_parse_decimal(#[case] input: &str, #[case] expected: Decimal) {
1158 let result = parse_decimal(input).unwrap();
1159 assert_eq!(result, expected);
1160 }
1161
1162 #[rstest]
1163 fn test_parse_decimal_invalid() {
1164 assert!(parse_decimal("invalid").is_err());
1165 assert!(parse_decimal("").is_err());
1166 }
1167
1168 #[rstest]
1169 #[case(&Some("123.45".to_string()), Some(dec!(123.45)))]
1170 #[case(&Some("0".to_string()), Some(Decimal::ZERO))]
1171 #[case(&Some(String::new()), None)]
1172 #[case(&None, None)]
1173 fn test_parse_optional_decimal(
1174 #[case] input: &Option<String>,
1175 #[case] expected: Option<Decimal>,
1176 ) {
1177 let result = parse_optional_decimal(input).unwrap();
1178 assert_eq!(result, expected);
1179 }
1180
1181 #[derive(Debug, Serialize, Deserialize, PartialEq)]
1184 struct TestFlexibleDecimal {
1185 #[serde(
1186 serialize_with = "serialize_decimal",
1187 deserialize_with = "deserialize_decimal"
1188 )]
1189 value: Decimal,
1190 #[serde(
1191 serialize_with = "serialize_optional_decimal",
1192 deserialize_with = "deserialize_optional_decimal"
1193 )]
1194 optional_value: Option<Decimal>,
1195 }
1196
1197 #[rstest]
1198 #[case(r#"{"value": 123.456, "optional_value": 789.012}"#, dec!(123.456), Some(dec!(789.012)))]
1199 #[case(r#"{"value": "123.456", "optional_value": "789.012"}"#, dec!(123.456), Some(dec!(789.012)))]
1200 #[case(r#"{"value": 100, "optional_value": null}"#, dec!(100), None)]
1201 #[case(r#"{"value": null, "optional_value": null}"#, Decimal::ZERO, None)]
1202 fn test_deserialize_flexible_decimal(
1203 #[case] json: &str,
1204 #[case] expected_value: Decimal,
1205 #[case] expected_optional: Option<Decimal>,
1206 ) {
1207 let result: TestFlexibleDecimal = serde_json::from_str(json).unwrap();
1208 assert_eq!(result.value, expected_value);
1209 assert_eq!(result.optional_value, expected_optional);
1210 }
1211
1212 #[rstest]
1213 fn test_flexible_decimal_roundtrip() {
1214 let original = TestFlexibleDecimal {
1215 value: dec!(123.456),
1216 optional_value: Some(dec!(789.012)),
1217 };
1218
1219 let json = serde_json::to_string(&original).unwrap();
1220 let deserialized: TestFlexibleDecimal = serde_json::from_str(&json).unwrap();
1221
1222 assert_eq!(original.value, deserialized.value);
1223 assert_eq!(original.optional_value, deserialized.optional_value);
1224 }
1225
1226 #[rstest]
1227 fn test_flexible_decimal_scientific_notation() {
1228 let json = r#"{"value": 0.00000001, "optional_value": 12345678.12345}"#;
1232 let parsed: TestFlexibleDecimal = serde_json::from_str(json).unwrap();
1233 assert_eq!(parsed.value, dec!(0.00000001));
1234 assert_eq!(parsed.optional_value, Some(dec!(12345678.12345)));
1235 }
1236
1237 #[rstest]
1238 fn test_flexible_decimal_empty_string_optional() {
1239 let json = r#"{"value": 100, "optional_value": ""}"#;
1240 let parsed: TestFlexibleDecimal = serde_json::from_str(json).unwrap();
1241 assert_eq!(parsed.value, dec!(100));
1242 assert_eq!(parsed.optional_value, None);
1243 }
1244
1245 #[derive(Debug, Deserialize)]
1248 struct TestDecimalOnly {
1249 #[serde(deserialize_with = "deserialize_decimal")]
1250 value: Decimal,
1251 }
1252
1253 #[rstest]
1254 #[case(r#"{"value": "1.5e-8"}"#, dec!(0.000000015))]
1255 #[case(r#"{"value": "1E10"}"#, dec!(10000000000))]
1256 #[case(r#"{"value": "-1.23e5"}"#, dec!(-123000))]
1257 fn test_deserialize_decimal_scientific_string(#[case] json: &str, #[case] expected: Decimal) {
1258 let result: TestDecimalOnly = serde_json::from_str(json).unwrap();
1259 assert_eq!(result.value, expected);
1260 }
1261
1262 #[rstest]
1263 #[case(r#"{"value": 9223372036854775807}"#, dec!(9223372036854775807))] #[case(r#"{"value": -9223372036854775808}"#, dec!(-9223372036854775808))] #[case(r#"{"value": 0}"#, Decimal::ZERO)]
1266 fn test_deserialize_decimal_large_integers(#[case] json: &str, #[case] expected: Decimal) {
1267 let result: TestDecimalOnly = serde_json::from_str(json).unwrap();
1268 assert_eq!(result.value, expected);
1269 }
1270
1271 #[rstest]
1272 #[case(r#"{"value": "-123.456789"}"#, dec!(-123.456789))]
1273 #[case(r#"{"value": -999.99}"#, dec!(-999.99))]
1274 fn test_deserialize_decimal_negative(#[case] json: &str, #[case] expected: Decimal) {
1275 let result: TestDecimalOnly = serde_json::from_str(json).unwrap();
1276 assert_eq!(result.value, expected);
1277 }
1278
1279 #[rstest]
1280 #[case(r#"{"value": "123456789.123456789012345678"}"#)] fn test_deserialize_decimal_high_precision(#[case] json: &str) {
1282 let result: TestDecimalOnly = serde_json::from_str(json).unwrap();
1283 assert_eq!(result.value, dec!(123456789.123456789012345678));
1284 }
1285
1286 #[rstest]
1287 #[case(
1288 r#"{"value": "1.234567890123456789012345678912345e-1"}"#,
1289 "0.1234567890123456789012345679"
1290 )]
1291 #[case(
1292 r#"{"value": "0.1234567890123456789012345678912345"}"#,
1293 "0.1234567890123456789012345679"
1294 )]
1295 #[case(
1296 r#"{"value": "999999999999999999999999999995e-29"}"#,
1297 "10.000000000000000000000000000"
1298 )]
1299 #[case(r#"{"value": "0.5e29"}"#, "50000000000000000000000000000")]
1300 #[case(r#"{"value": "-4e-29"}"#, "0.0000000000000000000000000000")]
1301 #[case(r#"{"value": "1.5e-999999"}"#, "0.0000000000000000000000000000")]
1302 #[case(r#"{"value": "9e-999999"}"#, "0.0000000000000000000000000000")]
1303 #[case(r#"{"value": "0e2000000000"}"#, "0")]
1304 fn test_deserialize_decimal_rounds_high_scale_values(
1305 #[case] json: &str,
1306 #[case] expected: &str,
1307 ) {
1308 let result: TestDecimalOnly = serde_json::from_str(json).unwrap();
1310 assert_eq!(result.value.to_string(), expected);
1311 }
1312
1313 #[rstest]
1314 #[case(r#"{"value": "8e28"}"#)] #[case(r#"{"value": "1e1000000000"}"#)] #[case(r#"{"value": "not-a-number"}"#)]
1317 fn test_deserialize_decimal_rejects_unrepresentable_values(#[case] json: &str) {
1318 let result: Result<TestDecimalOnly, _> = serde_json::from_str(json);
1321 assert!(result.is_err());
1322 }
1323
1324 #[rstest]
1325 fn test_deserialize_optional_decimal_rounds_high_scale_values() {
1326 let json = r#"{"value": "51.234567890123456789012345678912345e-1"}"#;
1327 let result: TestOptionalDecimalOnly = serde_json::from_str(json).unwrap();
1328 assert_eq!(
1329 result.value.map(|v| v.to_string()),
1330 Some("5.1234567890123456789012345679".into()),
1331 );
1332 }
1333
1334 #[derive(Debug, Deserialize)]
1335 struct TestOptionalDecimalOnly {
1336 #[serde(deserialize_with = "deserialize_optional_decimal")]
1337 value: Option<Decimal>,
1338 }
1339
1340 #[rstest]
1341 #[case(r#"{"value": "1.5e-8"}"#, Some(dec!(0.000000015)))]
1342 #[case(r#"{"value": null}"#, None)]
1343 #[case(r#"{"value": ""}"#, None)]
1344 #[case(r#"{"value": 42}"#, Some(dec!(42)))]
1345 #[case(r#"{"value": -100.5}"#, Some(dec!(-100.5)))]
1346 fn test_deserialize_optional_decimal_various(
1347 #[case] json: &str,
1348 #[case] expected: Option<Decimal>,
1349 ) {
1350 let result: TestOptionalDecimalOnly = serde_json::from_str(json).unwrap();
1351 assert_eq!(result.value, expected);
1352 }
1353
1354 use proptest::prelude::*;
1355 use rust_decimal::prelude::ToPrimitive;
1356
1357 fn representable_decimal_strategy() -> impl Strategy<Value = Decimal> {
1358 (
1361 -79_228_162_514_264_337_593_543_950_335i128
1362 ..=79_228_162_514_264_337_593_543_950_335i128,
1363 0u32..=28u32,
1364 )
1365 .prop_map(|(mantissa, scale)| Decimal::from_i128_with_scale(mantissa, scale))
1366 }
1367
1368 fn numeric_string_strategy() -> impl Strategy<Value = String> {
1369 (
1372 proptest::bool::ANY,
1373 "[0-9]{1,30}",
1374 proptest::option::of("[0-9]{1,40}"),
1375 proptest::option::of(-40i32..=40),
1376 )
1377 .prop_map(|(negative, integer, fraction, exponent)| {
1378 let mut value = String::new();
1379 if negative {
1380 value.push('-');
1381 }
1382 value.push_str(&integer);
1383 if let Some(fraction) = fraction {
1384 value.push('.');
1385 value.push_str(&fraction);
1386 }
1387
1388 if let Some(exponent) = exponent {
1389 value.push('e');
1390 value.push_str(&exponent.to_string());
1391 }
1392 value
1393 })
1394 }
1395
1396 proptest! {
1397 #[rstest]
1398 fn prop_deserialize_decimal_roundtrips_representable_values(
1399 expected in representable_decimal_strategy()
1400 ) {
1401 let json = format!(r#"{{"value": "{expected}"}}"#);
1404 let parsed: TestDecimalOnly = serde_json::from_str(&json).unwrap();
1405 prop_assert_eq!(parsed.value, expected);
1406 }
1407
1408 #[rstest]
1409 fn prop_deserialize_decimal_total_on_arbitrary_strings(value in "\\PC{0,64}") {
1410 let json = serde_json::to_string(&serde_json::json!({"value": value})).unwrap();
1412 let _ = serde_json::from_str::<TestDecimalOnly>(&json);
1413 }
1414
1415 #[rstest]
1416 fn prop_deserialize_decimal_tracks_f64_reference(value in numeric_string_strategy()) {
1417 let json = format!(r#"{{"value": "{value}"}}"#);
1420 if let Ok(parsed) = serde_json::from_str::<TestDecimalOnly>(&json) {
1421 let reference: f64 = value.parse().unwrap();
1422 let decoded = parsed.value.to_f64().unwrap();
1423 prop_assert!(
1424 (decoded - reference).abs() <= reference.abs() * 1e-9 + 1e-27,
1425 "decoded {decoded} diverges from reference {reference} for input {value}",
1426 );
1427 }
1428 }
1429
1430 #[rstest]
1431 fn prop_deserialize_optional_decimal_matches_required(
1432 value in numeric_string_strategy()
1433 ) {
1434 let json = format!(r#"{{"value": "{value}"}}"#);
1435 let required = serde_json::from_str::<TestDecimalOnly>(&json);
1436 let optional = serde_json::from_str::<TestOptionalDecimalOnly>(&json);
1437 match (required, optional) {
1438 (Ok(required), Ok(optional)) => {
1439 prop_assert_eq!(optional.value, Some(required.value));
1440 }
1441 (Err(_), Err(_)) => {}
1442 (required, optional) => prop_assert!(
1443 false,
1444 "required and optional decoding disagree for input {}: {:?} vs {:?}",
1445 value,
1446 required.map(|r| r.value),
1447 optional.map(|o| o.value),
1448 ),
1449 }
1450 }
1451 }
1452}