21 lines
673 B
Rust
21 lines
673 B
Rust
|
use quote::quote;
|
||
|
|
|
||
|
|
pub(crate) fn function_body(body: proc_macro2::TokenStream) -> proc_macro2::TokenStream {
|
||
|
|
quote! {
|
||
|
|
let connection = self.pool.get().await;
|
||
|
|
match connection {
|
||
|
|
Ok(mut connection) => {
|
||
|
|
use diesel::associations::HasTable;
|
||
|
|
#body
|
||
|
|
}
|
||
|
|
Err(error) => Err(lib::diesel_crud_trait::CrudError::PoolError(error.to_string())),
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
pub(crate) fn return_type(output: proc_macro2::TokenStream) -> proc_macro2::TokenStream {
|
||
|
|
quote! {
|
||
|
|
std::pin::Pin<Box<dyn core::future::Future<Output = Result<#output, lib::diesel_crud_trait::CrudError>> + Send + 'b>>
|
||
|
|
}
|
||
|
|
}
|