fix issue-489 crud insert is error when the primary key column name i…

…s not id
rbatis · Feb 11, 2024 · a620e94 · a620e94
1 parent 4b66e0d
commit a620e94
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 13 deletions.
diff --git a/src/crud.rs b/src/crud.rs
@@ -83,14 +83,14 @@ macro_rules! impl_insert {
                          `(`
                          trim ',':
                            for k,v in table:
-                              if k == 'id' && v== null:
+                              if v== null:
                                  continue:
                               ${k},
                          `) VALUES `
                       (
                       trim ',':
                        for k,v in table:
-                         if k == 'id' && v== null:
+                         if v== null:
                             continue:
                          #{v},
                       ),

diff --git a/tests/crud_test.rs b/tests/crud_test.rs
@@ -325,17 +325,14 @@ mod test {
             let r = MockTable::insert(&mut rb, &t).await.unwrap();
             let (sql, args) = queue.pop().unwrap();
             println!("{}", sql);
-            assert_eq!(sql, "insert into mock_table (id,name,pc_link,h5_link,pc_banner_img,h5_banner_img,sort,status,remark,create_time,version,delete_flag,count) VALUES (?,?,?,?,?,?,?,?,?,?,?,?,?)");
+            assert_eq!(sql, "insert into mock_table (id,name,pc_link,h5_link,status,remark,create_time,version,delete_flag,count) VALUES (?,?,?,?,?,?,?,?,?,?)");
             assert_eq!(
                 args,
                 vec![
                     to_value!(t.id),
                     to_value!(t.name),
                     to_value!(t.pc_link),
                     to_value!(t.h5_link),
-                    to_value!(t.pc_banner_img),
-                    to_value!(t.h5_banner_img),
-                    to_value!(t.sort),
                     to_value!(t.status),
                     to_value!(t.remark),
                     to_value!(t.create_time),
@@ -376,17 +373,14 @@ mod test {
             let r = MockTable::insert_batch(&mut rb, &ts, 10).await.unwrap();
             let (sql, args) = queue.pop().unwrap();
             println!("{}", sql);
-            assert_eq!(sql, "insert into mock_table (id,name,pc_link,h5_link,pc_banner_img,h5_banner_img,sort,status,remark,create_time,version,delete_flag,count) VALUES (?,?,?,?,?,?,?,?,?,?,?,?,?),(?,?,?,?,?,?,?,?,?,?,?,?,?)");
+            assert_eq!(sql, "insert into mock_table (id,name,pc_link,h5_link,status,remark,create_time,version,delete_flag,count) VALUES (?,?,?,?,?,?,?,?,?,?),(?,?,?,?,?,?,?,?,?,?)");
             assert_eq!(
                 args,
                 vec![
                     to_value!(&ts[0].id),
                     to_value!(&ts[0].name),
                     to_value!(&ts[0].pc_link),
                     to_value!(&ts[0].h5_link),
-                    to_value!(&ts[0].pc_banner_img),
-                    to_value!(&ts[0].h5_banner_img),
-                    to_value!(&ts[0].sort),
                     to_value!(&ts[0].status),
                     to_value!(&ts[0].remark),
                     to_value!(&ts[0].create_time),
@@ -397,9 +391,6 @@ mod test {
                     to_value!(&ts[1].name),
                     to_value!(&ts[1].pc_link),
                     to_value!(&ts[1].h5_link),
-                    to_value!(&ts[1].pc_banner_img),
-                    to_value!(&ts[1].h5_banner_img),
-                    to_value!(&ts[1].sort),
                     to_value!(&ts[1].status),
                     to_value!(&ts[1].remark),
                     to_value!(&ts[1].create_time),