Add SqlxBitFlags derive macro
This commit is contained in:
parent
c2035585ae
commit
656e5e31c0
|
@ -155,3 +155,34 @@ pub fn composite_derive(input: TokenStream) -> TokenStream {
|
||||||
_ => panic!("Composite derive macro only supports structs"),
|
_ => panic!("Composite derive macro only supports structs"),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[proc_macro_derive(SqlxBitFlags)]
|
||||||
|
pub fn sqlx_bitflag_derive(input: TokenStream) -> TokenStream {
|
||||||
|
let ast: syn::DeriveInput = syn::parse(input).unwrap();
|
||||||
|
|
||||||
|
let name = &ast.ident;
|
||||||
|
|
||||||
|
quote!{
|
||||||
|
#[cfg(feature = "sqlx")]
|
||||||
|
impl sqlx::Type<sqlx::MySql> for #name {
|
||||||
|
fn type_info() -> sqlx::mysql::MySqlTypeInfo {
|
||||||
|
u64::type_info()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(feature = "sqlx")]
|
||||||
|
impl<'q> sqlx::Encode<'q, sqlx::MySql> for #name {
|
||||||
|
fn encode_by_ref(&self, buf: &mut <MySql as HasArguments<'q>>::ArgumentBuffer) -> IsNull {
|
||||||
|
u64::encode_by_ref(&self.bits(), buf)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(feature = "sqlx")]
|
||||||
|
impl<'q> sqlx::Decode<'q, sqlx::MySql> for #name {
|
||||||
|
fn decode(value: <MySql as HasValueRef<'q>>::ValueRef) -> Result<Self, BoxDynError> {
|
||||||
|
u64::decode(value).map(|d| #name::from_bits(d).unwrap())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
.into()
|
||||||
|
}
|
Loading…
Reference in New Issue