aboutsummaryrefslogtreecommitdiff
path: root/tosin-macros/src/lib.rs
diff options
context:
space:
mode:
authorMelody Horn <melody@boringcactus.com>2021-06-30 15:58:48 -0600
committerMelody Horn <melody@boringcactus.com>2021-06-30 15:58:48 -0600
commitb4c7c3cd480e626ac78b49ab91a95340ccfef26a (patch)
treec1966362001b67b80e8f54a7a9f94be8a47aaf7c /tosin-macros/src/lib.rs
parentbdfdcb98d39d1887b08748e1ea629f0a83f340a4 (diff)
downloadtosin-b4c7c3cd480e626ac78b49ab91a95340ccfef26a.tar.gz
tosin-b4c7c3cd480e626ac78b49ab91a95340ccfef26a.zip
finish inserting save_mut for models
Diffstat (limited to 'tosin-macros/src/lib.rs')
-rw-r--r--tosin-macros/src/lib.rs33
1 files changed, 25 insertions, 8 deletions
diff --git a/tosin-macros/src/lib.rs b/tosin-macros/src/lib.rs
index ff62f3d..1019540 100644
--- a/tosin-macros/src/lib.rs
+++ b/tosin-macros/src/lib.rs
@@ -84,15 +84,22 @@ fn to_field_spec(field: &syn::Field) -> FieldInfo<impl quote::ToTokens> {
// TODO foreign key constraint
FieldInfo {
tosin_field: quote! { ::tosin::db::models::Field::IntField { name: stringify!(#field_name) } },
+ diesel_column: quote! { #field_name -> BigInt },
+ diesel_type: quote! { ::tosin::db::sql_types::BigInt },
+ }
+ } else if field_type == &parse_type("u64") {
+ // TODO allow at all since some dbs can't express that type
+ FieldInfo {
+ tosin_field: quote! { ::tosin::db::models::Field::IntField { name: stringify!(#field_name) } },
diesel_column: quote! { #field_name -> Integer },
diesel_type: quote! { ::tosin::db::sql_types::Integer },
}
- } else if field_type == &parse_type("usize") {
+ } else if field_type == &parse_type("i64") {
// TODO default
FieldInfo {
tosin_field: quote! { ::tosin::db::models::Field::IntField { name: stringify!(#field_name) } },
- diesel_column: quote! { #field_name -> Integer },
- diesel_type: quote! { ::tosin::db::sql_types::Integer },
+ diesel_column: quote! { #field_name -> BigInt },
+ diesel_type: quote! { ::tosin::db::sql_types::BigInt },
}
} else if field_type == &parse_type("String") {
let max_length = model_options
@@ -211,13 +218,22 @@ fn impl_model(ast: &syn::DeriveInput) -> TokenStream {
}
}
- pub fn save_mut(&mut self, connection: &mut tosin::db::backend::Connection) {
+ pub fn save_mut(&mut self, connection: &tosin::db::backend::Connection) {
+ use diesel::prelude::*;
if self.id.is_none() {
// no id yet, so not from db, so insert
- let new_self = tosin::db::insert_into(#lowercase_name::table)
- .values(&self)
- .get_result(connection)
- .expect("error saving to database"); // TODO propagate error
+ // unfortunately InsertStatement::get_result is only supported on pg for now,
+ // so we have to pull this shit
+ let new_self = tosin::db::backend::insert_and_retrieve::<
+ Self, // row
+ #lowercase_name::table, // table
+ #lowercase_name::id, // id
+ >(
+ self,
+ #lowercase_name::table,
+ connection,
+ #lowercase_name::id
+ );
*self = new_self;
} else {
todo!("update existing db item");
@@ -246,6 +262,7 @@ fn impl_model(ast: &syn::DeriveInput) -> TokenStream {
type Values = <(#(#insertable_types,)*) as tosin::db::Insertable<#lowercase_name::table>>::Values;
fn values(self) -> Self::Values {
+ use diesel::ExpressionMethods;
(#(#insertable_values,)*).values()
}
}