[src/code.rs] Add impl Default for Update

Author: SamuelMarks

src/code.rs

@@ -228,7 +228,9 @@ impl<'a> Struct<'a> {
                     derives_vec.push(derives::PARTIALEQ);
                 }
 
-                derives_vec.push(derives::DEFAULT);
+                if !self.config.options.default_impl {
+                    derives_vec.push(derives::DEFAULT);
+                }
             }
             StructType::Create => derives_vec.extend_from_slice(&[derives::INSERTABLE]),
         }
@@ -824,8 +826,6 @@ pub fn generate_for_table(table: &ParsedTableMacro, config: &GenerationConfig) -
     let struct_name = &table.struct_name;
     let table_options = config.table(&table.name.to_string());
     let generated_columns = table_options.get_autogenerated_columns();
-    println!("struct_name: {}", struct_name);
-    println!("&table.name: {}", &table.name);
     let not_generated = |col: &&ParsedColumnMacro| -> bool {
         !generated_columns.contains(&col.column_name.as_str())
     };
@@ -870,6 +870,25 @@ pub fn generate_for_table(table: &ParsedTableMacro, config: &GenerationConfig) -
     if update_struct.has_code() {
         ret_buffer.push('\n');
         ret_buffer.push_str(update_struct.code());
+        if config.options.default_impl {
+            ret_buffer.push('\n');
+            ret_buffer.push_str(
+                build_default_impl_fn(
+                    &StructType::format(&StructType::Update, &struct_name),
+                    update_struct
+                        .table
+                        .columns
+                        .iter()
+                        .filter(not_generated)
+                        .map(|col| NameTypNullable {
+                            name: col.column_name.to_string(),
+                            typ: col.ty.as_str(),
+                            nullable: col.is_nullable,
+                        }),
+                )
+                .as_str(),
+            );
+        }
     }
 
     // third, push functions - if enabled
@@ -883,15 +902,11 @@ pub fn generate_for_table(table: &ParsedTableMacro, config: &GenerationConfig) -
         ret_buffer.push_str(
             build_default_impl_fn(
                 &struct_name,
-                table
-                    .columns
-                    .iter()
-                    // .filter(not_generated)
-                    .map(|col| NameTypNullable {
-                        name: col.name.to_string().to_owned(),
-                        typ: col.ty.as_str(),
-                        nullable: col.is_nullable,
-                    }),
+                table.columns.iter().map(|col| NameTypNullable {
+                    name: col.name.to_string().to_owned(),
+                    typ: col.ty.as_str(),
+                    nullable: col.is_nullable,
+                }),
             )
             .as_str(),
         );