async_graphql/types/
id.rs

1use std::{
2    num::ParseIntError,
3    ops::{Deref, DerefMut},
4};
5
6use async_graphql_value::ConstValue;
7#[cfg(feature = "bson")]
8use bson::oid::{self, ObjectId};
9use serde::{Deserialize, Serialize};
10
11use crate::{InputValueError, InputValueResult, Scalar, ScalarType, Value};
12
13/// ID scalar
14///
15/// The input is a `&str`, `String`, `usize` or `uuid::UUID`, and the output is
16/// a string.
17#[derive(Clone, Ord, PartialOrd, Eq, PartialEq, Hash, Debug, Serialize, Deserialize, Default)]
18#[serde(transparent)]
19pub struct ID(pub String);
20
21impl AsRef<str> for ID {
22    fn as_ref(&self) -> &str {
23        self.0.as_str()
24    }
25}
26
27impl Deref for ID {
28    type Target = String;
29
30    fn deref(&self) -> &Self::Target {
31        &self.0
32    }
33}
34
35impl DerefMut for ID {
36    fn deref_mut(&mut self) -> &mut Self::Target {
37        &mut self.0
38    }
39}
40
41impl<T: std::fmt::Display> From<T> for ID {
42    fn from(value: T) -> Self {
43        ID(value.to_string())
44    }
45}
46
47impl From<ID> for String {
48    fn from(id: ID) -> Self {
49        id.0
50    }
51}
52
53impl From<ID> for ConstValue {
54    fn from(id: ID) -> Self {
55        ConstValue::String(id.0)
56    }
57}
58
59macro_rules! try_from_integers {
60    ($($ty:ty),*) => {
61        $(
62           impl TryFrom<ID> for $ty {
63                type Error = ParseIntError;
64
65                fn try_from(id: ID) -> Result<Self, Self::Error> {
66                    id.0.parse()
67                }
68            }
69         )*
70    };
71}
72
73try_from_integers!(i8, i16, i32, i64, i128, u8, u16, u32, u64, u128, isize, usize);
74
75#[cfg(feature = "uuid")]
76impl TryFrom<ID> for uuid::Uuid {
77    type Error = uuid::Error;
78
79    fn try_from(id: ID) -> Result<Self, Self::Error> {
80        uuid::Uuid::parse_str(&id.0)
81    }
82}
83
84#[cfg(feature = "bson")]
85impl TryFrom<ID> for ObjectId {
86    type Error = oid::Error;
87
88    fn try_from(id: ID) -> std::result::Result<Self, oid::Error> {
89        ObjectId::parse_str(id.0)
90    }
91}
92
93impl PartialEq<&str> for ID {
94    fn eq(&self, other: &&str) -> bool {
95        self.0.as_str() == *other
96    }
97}
98
99#[Scalar(internal, name = "ID")]
100impl ScalarType for ID {
101    fn parse(value: Value) -> InputValueResult<Self> {
102        match value {
103            Value::Number(n) if n.is_i64() => Ok(ID(n.to_string())),
104            Value::String(s) => Ok(ID(s)),
105            _ => Err(InputValueError::expected_type(value)),
106        }
107    }
108
109    fn is_valid(value: &Value) -> bool {
110        match value {
111            Value::Number(n) if n.is_i64() => true,
112            Value::String(_) => true,
113            _ => false,
114        }
115    }
116
117    fn to_value(&self) -> Value {
118        Value::String(self.0.clone())
119    }
120}
async_graphql/types/id.rs

async_graphql/types/
id.rs
