1use serde::de::Visitor;
2use serde::{Deserialize, Serialize};
3use std::fmt;
4use std::num::ParseIntError;
5use std::str::FromStr;
6
7#[derive(Debug, Clone, Copy, Eq, PartialEq, Ord, PartialOrd, Hash)]
8pub struct NumericId(u64);
9
10impl NumericId {
11 pub fn new(id: u64) -> Self {
12 Self(id)
13 }
14 pub fn as_u64(self) -> u64 {
15 self.0
16 }
17}
18
19impl From<NumericId> for u64 {
20 fn from(id: NumericId) -> Self {
21 id.0
22 }
23}
24
25impl From<u64> for NumericId {
26 fn from(id: u64) -> Self {
27 NumericId(id)
28 }
29}
30
31impl<'a> From<&'a u64> for NumericId {
32 fn from(id: &'a u64) -> Self {
33 NumericId(*id)
34 }
35}
36
37impl FromStr for NumericId {
38 type Err = ParseIntError;
39 fn from_str(s: &str) -> Result<Self, Self::Err> {
40 Ok(Self(s.parse()?))
41 }
42}
43
44impl fmt::Display for NumericId {
45 fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
46 f.write_fmt(format_args!("{}", self.0))
47 }
48}
49
50impl PartialEq<u64> for NumericId {
51 fn eq(&self, other: &u64) -> bool {
52 &self.0 == other
53 }
54}
55impl PartialEq<&u64> for NumericId {
56 fn eq(&self, other: &&u64) -> bool {
57 &self.0 == *other
58 }
59}
60
61impl Serialize for NumericId {
62 fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
63 where
64 S: serde::Serializer,
65 {
66 serializer.serialize_str(&self.to_string())
67 }
68}
69
70struct NumericIdVisitor;
71
72impl<'de> Visitor<'de> for NumericIdVisitor {
73 type Value = NumericId;
74 fn expecting(&self, f: &mut fmt::Formatter) -> fmt::Result {
75 f.write_str("a valid unsigned integer or string representing an id")
76 }
77 fn visit_u64<E>(self, v: u64) -> Result<Self::Value, E>
78 where
79 E: serde::de::Error,
80 {
81 Ok(NumericId(v))
82 }
83 fn visit_i64<E>(self, v: i64) -> Result<Self::Value, E>
84 where
85 E: serde::de::Error,
86 {
87 if v < 0 {
88 Err(E::custom("value must be an unsigned integer"))
89 } else {
90 Ok(NumericId(v as u64))
91 }
92 }
93 fn visit_str<E>(self, v: &str) -> Result<Self::Value, E>
94 where
95 E: serde::de::Error,
96 {
97 v.parse().map_err(E::custom)
98 }
99}
100
101impl<'de> Deserialize<'de> for NumericId {
102 fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
103 where
104 D: serde::Deserializer<'de>,
105 {
106 deserializer.deserialize_any(NumericIdVisitor)
107 }
108}
109
110pub trait IntoNumericId: fmt::Display {
111 fn into_id(self) -> NumericId;
112}
113
114impl IntoNumericId for NumericId {
115 fn into_id(self) -> NumericId {
116 self
117 }
118}
119
120impl IntoNumericId for u64 {
121 fn into_id(self) -> NumericId {
122 NumericId(self)
123 }
124}
125
126impl<'a> IntoNumericId for &'a u64 {
127 fn into_id(self) -> NumericId {
128 NumericId(*self)
129 }
130}