caisin/sea-orm
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge pull request #175 from MuhannadAlrusayni/master

Browse Source 
Impl `TryGetableMany` for diffrent types of generices
This commit is contained in:
Chris Tsang 2021-09-19 01:54:00 +08:00 committed by GitHub
commit 5d1488f12d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 48 additions and 25 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

73
src/executor/query.rs
View File

@ -314,29 +314,50 @@ where
} }
} }
impl<T> TryGetableMany for (T, T) impl<A, B> TryGetableMany for (A, B)
where where
T: TryGetable, A: TryGetable,
B: TryGetable,
{ {
fn try_get_many(res: &QueryResult, pre: &str, cols: &[String]) -> Result<Self, TryGetError> { fn try_get_many(res: &QueryResult, pre: &str, cols: &[String]) -> Result<Self, TryGetError> {
try_get_many_with_slice_len_of(2, cols)?; try_get_many_with_slice_len_of(2, cols)?;
Ok(( Ok((
T::try_get(res, pre, &cols[0])?, A::try_get(res, pre, &cols[0])?,
T::try_get(res, pre, &cols[1])?, B::try_get(res, pre, &cols[1])?,
)) ))
} }
} }
impl<T> TryGetableMany for (T, T, T) impl<A, B, C> TryGetableMany for (A, B, C)
where where
T: TryGetable, A: TryGetable,
B: TryGetable,
C: TryGetable,
{ {
fn try_get_many(res: &QueryResult, pre: &str, cols: &[String]) -> Result<Self, TryGetError> { fn try_get_many(res: &QueryResult, pre: &str, cols: &[String]) -> Result<Self, TryGetError> {
try_get_many_with_slice_len_of(3, cols)?; try_get_many_with_slice_len_of(3, cols)?;
Ok(( Ok((
T::try_get(res, pre, &cols[0])?, A::try_get(res, pre, &cols[0])?,
T::try_get(res, pre, &cols[1])?, B::try_get(res, pre, &cols[1])?,
T::try_get(res, pre, &cols[2])?, C::try_get(res, pre, &cols[2])?,
))
}
}
impl<A, B, C, D> TryGetableMany for (A, B, C, D)
where
A: TryGetable,
B: TryGetable,
C: TryGetable,
D: TryGetable,
{
fn try_get_many(res: &QueryResult, pre: &str, cols: &[String]) -> Result<Self, TryGetError> {
try_get_many_with_slice_len_of(4, cols)?;
Ok((
A::try_get(res, pre, &cols[0])?,
B::try_get(res, pre, &cols[1])?,
C::try_get(res, pre, &cols[2])?,
D::try_get(res, pre, &cols[3])?,
)) ))
} }
} }
@ -370,15 +391,27 @@ macro_rules! try_from_u64_err {
} }
} }
}; };
}
macro_rules! try_from_u64_tuple { ( $($gen_type: ident),* ) => {
( $type: ty ) => { impl<$( $gen_type, )*> TryFromU64 for ($( $gen_type, )*)
try_from_u64_err!(($type, $type)); where
try_from_u64_err!(($type, $type, $type)); $( $gen_type: TryFromU64, )*
{
fn try_from_u64(_: u64) -> Result<Self, DbErr> {
Err(DbErr::Exec(format!(
"{} cannot be converted from u64",
stringify!(($($gen_type,)*))
)))
}
}
}; };
} }
// impl TryFromU64 for tuples with generic types
try_from_u64_err!(A, B);
try_from_u64_err!(A, B, C);
try_from_u64_err!(A, B, C, D);
macro_rules! try_from_u64_numeric { macro_rules! try_from_u64_numeric {
( $type: ty ) => { ( $type: ty ) => {
impl TryFromU64 for $type { impl TryFromU64 for $type {
@ -393,7 +426,6 @@ macro_rules! try_from_u64_numeric {
}) })
} }
} }
try_from_u64_tuple!($type);
}; };
} }
@ -413,19 +445,10 @@ macro_rules! try_from_u64_string {
Ok(n.to_string()) Ok(n.to_string())
} }
} }
try_from_u64_tuple!($type);
}; };
} }
try_from_u64_string!(String); try_from_u64_string!(String);
macro_rules! try_from_u64_dummy {
( $type: ty ) => {
try_from_u64_err!($type);
try_from_u64_err!(($type, $type));
try_from_u64_err!(($type, $type, $type));
};
}
#[cfg(feature = "with-uuid")] #[cfg(feature = "with-uuid")]
try_from_u64_dummy!(uuid::Uuid); try_from_u64_err!(uuid::Uuid);