OMGeeky/google_bigquery_v2
1
0
Fork 0
mirror of https://github.com/OMGeeky/google_bigquery_v2.git synced 2025-12-26 16:17:24 +01:00
Code Issues Packages Projects Releases Wiki Activity

re-export derive usages

Browse Source
This commit is contained in:
OMGeeky
2023-04-15 15:15:58 +02:00
parent 535a5896b8
commit b7ce1be6b7
2 changed files with 23 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

34
google_bigquery_v2_derive/src/lib.rs
View File

@@ -53,7 +53,7 @@ fn implement_big_query_table_base(
let impl_get_field_value = implement_get_field_value(&ast);
let impl_from_query_result_row = implement_from_query_result_row(&ast);
quote::quote! {
#[async_trait::async_trait]
#[google_bigquery_v2::re_exports::async_trait::async_trait]
impl BigQueryTableBase for #table_ident {
#impl_get_all_params
#impl_get_parameter_from_field
@@ -88,7 +88,7 @@ fn implement_get_all_params(ast: &DeriveInput, table_ident: &Ident) -> TokenStre
quote::quote! {
fn get_all_params(&self) -> google_bigquery_v2::prelude::Result<Vec<Option<google_bigquery_v2::data::QueryParameter>>> {
log::trace!("get_all_params() self:{:?}", self);
google_bigquery_v2::re_exports::log::trace!("get_all_params() self:{:?}", self);
Ok(vec![
#(#fields),*
])
@@ -112,7 +112,7 @@ fn implement_get_parameter_from_field(ast: &DeriveInput, table_ident: &Ident) ->
quote::quote! {
fn get_parameter_from_field(&self, field_name: &str) -> google_bigquery_v2::prelude::Result<Option<google_bigquery_v2::data::QueryParameter>> {
log::trace!("get_parameter_from_field(); field_name: '{}' self:{:?}", field_name, self);
google_bigquery_v2::re_exports::log::trace!("get_parameter_from_field(); field_name: '{}' self:{:?}", field_name, self);
match field_name {
#(#fields)*
_ => Err(format!("Field {} not found", field_name).into()),
@@ -127,7 +127,7 @@ fn implement_get_client(client_field: &Field) -> TokenStream {
let client_ident = client_field.field_ident.clone();
quote::quote! {
fn get_client(&self) -> &BigqueryClient {
log::trace!("get_client() self={:?}", self);
google_bigquery_v2::re_exports::log::trace!("get_client() self={:?}", self);
&self.#client_ident
}
}
@@ -137,7 +137,7 @@ fn implement_set_client(client_field: &Field) -> TokenStream {
let client_ident = client_field.field_ident.clone();
quote::quote! {
fn set_client(&mut self, client: BigqueryClient) {
log::trace!("set_client() self={:?}", self);
google_bigquery_v2::re_exports::log::trace!("set_client() self={:?}", self);
self.#client_ident = client;
}
}
@@ -147,7 +147,7 @@ fn implement_get_pk_field_name(pk_field: &Field) -> TokenStream {
let pk_local_name = pk_field.local_name.clone();
quote::quote! {
fn get_pk_field_name() -> String {
log::trace!("get_pk_field_name()");
google_bigquery_v2::re_exports::log::trace!("get_pk_field_name()");
String::from(#pk_local_name)
}
}
@@ -157,7 +157,7 @@ fn implement_get_pk_db_name(pk_field: &Field) -> TokenStream {
let pk_db_name = pk_field.db_name.clone();
quote::quote! {
fn get_pk_db_name() -> String {
log::trace!("get_pk_db_name()");
google_bigquery_v2::re_exports::log::trace!("get_pk_db_name()");
String::from(#pk_db_name)
}
}
@@ -167,7 +167,7 @@ fn implement_get_pk_value(pk_field: &Field) -> TokenStream {
let pk_ident = &pk_field.field_ident;
quote::quote! {
fn get_pk_value(&self) -> &(dyn google_bigquery_v2::data::param_conversion::BigDataValueType + Send + Sync) {
log::trace!("get_pk_value() self={:?}", self);
google_bigquery_v2::re_exports::log::trace!("get_pk_value() self={:?}", self);
&self.#pk_ident
}
}
@@ -193,7 +193,7 @@ fn implement_get_query_fields(ast: &DeriveInput) -> TokenStream {
quote::quote! {
fn get_query_fields(include_pk: bool) -> std::collections::HashMap<String, String> {
log::trace!("get_query_fields() include_pk={}", include_pk);
google_bigquery_v2::re_exports::log::trace!("get_query_fields() include_pk={}", include_pk);
let mut map = std::collections::HashMap::new();
if(include_pk) {
#pk_insert
@@ -207,7 +207,7 @@ fn implement_get_query_fields(ast: &DeriveInput) -> TokenStream {
fn implement_impl_get_table_name(table_name: &String) -> TokenStream {
quote::quote! {
fn get_table_name() -> String {
log::trace!("get_table_name()");
google_bigquery_v2::re_exports::log::trace!("get_table_name()");
String::from(#table_name)
}
}
@@ -226,8 +226,8 @@ fn implement_set_field_value(ast: &DeriveInput) -> TokenStream {
let fields: Vec<TokenStream> = fields.into_iter().map(write_set_field_value).collect();
quote::quote! {
fn set_field_value(&mut self, field_name: &str, value: &serde_json::Value) -> Result<()>{
log::trace!("set_field_value() self={:?} field_name={} value={:?}", self, field_name, value);
fn set_field_value(&mut self, field_name: &str, value: &google_bigquery_v2::re_exports::serde_json::Value) -> Result<()>{
google_bigquery_v2::re_exports::log::trace!("set_field_value() self={:?} field_name={} value={:?}", self, field_name, value);
use google_bigquery_v2::data::param_conversion::ConvertBigQueryParams;
match field_name {
#(#fields)*
@@ -249,8 +249,8 @@ fn implement_get_field_value(ast: &DeriveInput) -> TokenStream {
let fields: Vec<TokenStream> = fields.into_iter().map(write_get_field_value).collect();
quote::quote! {
fn get_field_value(&self, field_name: &str) -> Result<serde_json::Value> {
log::trace!("get_field_value() self={:?} field_name={}", self, field_name);
fn get_field_value(&self, field_name: &str) -> Result<google_bigquery_v2::re_exports::serde_json::Value> {
google_bigquery_v2::re_exports::log::trace!("get_field_value() self={:?} field_name={}", self, field_name);
use google_bigquery_v2::data::param_conversion::ConvertBigQueryParams;
match field_name {
#(#fields)*
@@ -275,10 +275,10 @@ fn implement_from_query_result_row(ast: &DeriveInput) -> TokenStream {
quote::quote! {
fn new_from_query_result_row(
client: BigqueryClient,
row: &std::collections::HashMap<String, serde_json::Value>,
row: &std::collections::HashMap<String, google_bigquery_v2::re_exports::serde_json::Value>,
) -> Result<Self>
where Self: Sized {
log::trace!("new_from_query_result_row() client={:?} row={:?}", client, row);
google_bigquery_v2::re_exports::log::trace!("new_from_query_result_row() client={:?} row={:?}", client, row);
use google_bigquery_v2::data::param_conversion::ConvertBigQueryParams;
let result = Self{
#client_ident: client,
@@ -296,7 +296,7 @@ fn implement_reload(pk_field: &Field) -> TokenStream {
where
Self: Sized + Send + Sync,
{
log::trace!("reload()");
google_bigquery_v2::re_exports::log::trace!("reload()");
let value = &self.#pk_value;//TODO: this is the problem!. it just does not want to work
Self::get_by_pk(self.get_client().clone(), value).await.map(|mut t| {
*self = t;

6
src/lib.rs
View File

@@ -2,3 +2,9 @@ pub mod client;
pub mod data;
pub mod prelude;
pub mod utils;
pub mod re_exports {
pub use async_trait;
pub use log;
pub use serde_json;
}