@@ -95,11 +95,7 @@ impl Generator {
9595 . to_string ( )
9696 . from_case ( Case :: Pascal )
9797 . to_case ( Case :: Snake ) ;
98- let index = self
99- . columns
100- . indexes
101- . values ( )
102- . find ( |idx| idx. field . to_string ( ) == op. by . to_string ( ) ) ;
98+ let index = self . columns . indexes . values ( ) . find ( |idx| idx. field == op. by ) ;
10399
104100 let indexes_columns: Option < Vec < _ > > = {
105101 let columns: Vec < _ > = self
@@ -126,23 +122,14 @@ impl Generator {
126122 } else {
127123 self . gen_non_unique_update ( snake_case_name, name, index_name, idents)
128124 }
129- } else {
130- if self . columns . primary_keys . len ( ) == 1 {
131- if self . columns . primary_keys . first ( ) . unwrap ( ) . to_string ( )
132- == op. by . to_string ( )
133- {
134- self . gen_pk_update (
135- snake_case_name,
136- name,
137- idents,
138- indexes_columns. as_ref ( ) ,
139- )
140- } else {
141- todo ! ( )
142- }
125+ } else if self . columns . primary_keys . len ( ) == 1 {
126+ if * self . columns . primary_keys . first ( ) . unwrap ( ) == op. by {
127+ self . gen_pk_update ( snake_case_name, name, idents, indexes_columns. as_ref ( ) )
143128 } else {
144129 todo ! ( )
145130 }
131+ } else {
132+ todo ! ( )
146133 }
147134 } )
148135 . collect :: < Vec < _ > > ( ) ;
@@ -156,7 +143,7 @@ impl Generator {
156143 & self ,
157144 snake_case_name : String ,
158145 name : & Ident ,
159- idents : & Vec < Ident > ,
146+ idents : & [ Ident ] ,
160147 idx_idents : Option < & Vec < Ident > > ,
161148 ) -> TokenStream {
162149 let pk_ident = & self . pk . as_ref ( ) . unwrap ( ) . ident ;
@@ -228,7 +215,7 @@ impl Generator {
228215 quote ! { }
229216 } ;
230217
231- let process_diff_ident = if let Some ( _ ) = diff {
218+ let process_diff_ident = if diff . is_some ( ) {
232219 quote ! {
233220 self . 0 . indexes. process_difference( link, diffs) ?;
234221 }
@@ -293,7 +280,7 @@ impl Generator {
293280 snake_case_name : String ,
294281 name : & Ident ,
295282 index : & Ident ,
296- idents : & Vec < Ident > ,
283+ idents : & [ Ident ] ,
297284 ) -> TokenStream {
298285 let method_ident = Ident :: new (
299286 format ! ( "update_{snake_case_name}" ) . as_str ( ) ,
@@ -381,7 +368,7 @@ impl Generator {
381368 snake_case_name : String ,
382369 name : & Ident ,
383370 index : & Ident ,
384- idents : & Vec < Ident > ,
371+ idents : & [ Ident ] ,
385372 ) -> TokenStream {
386373 let method_ident = Ident :: new (
387374 format ! ( "update_{snake_case_name}" ) . as_str ( ) ,
0 commit comments