Fix incorrect behavior when adding an autoincrement column to a Postg… (

#697) * Fix incorrect behavior when adding an autoincrement column to a Postgres table. * rustfmt
SeaQL · Sep 8, 2023 · 375b7dc · 375b7dc
1 parent 36c03c3
commit 375b7dc
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 3 deletions.
diff --git a/src/backend/postgres/table.rs b/src/backend/postgres/table.rs
@@ -112,7 +112,15 @@ impl TableBuilder for PostgresQueryBuilder {
                     let f = |column_def: &ColumnDef, sql: &mut dyn SqlWriter| {
                         if let Some(column_type) = &column_def.types {
                             write!(sql, " ").unwrap();
-                            self.prepare_column_type(column_type, sql);
+                            if column_def
+                                .spec
+                                .iter()
+                                .any(|v| matches!(v, ColumnSpec::AutoIncrement))
+                            {
+                                self.prepare_column_auto_increment(column_type, sql);
+                            } else {
+                                self.prepare_column_type(column_type, sql);
+                            }
                         }
                     };
                     self.prepare_column_def_common(column, sql, f);

diff --git a/tests/postgres/table.rs b/tests/postgres/table.rs
@@ -519,7 +519,7 @@ fn alter_9() {
 
 #[test]
 fn alter_10() {
-    // https://dbfiddle.uk/PQksflGf
+    // https://dbfiddle.uk/BeiZPvBe
     assert_eq!(
         Table::alter()
             .table(Glyph::Table)
@@ -534,7 +534,7 @@ fn alter_10() {
             .to_string(PostgresQueryBuilder),
         [
             r#"ALTER TABLE "glyph""#,
-            r#"ADD COLUMN "aspect" integer NOT NULL UNIQUE PRIMARY KEY"#,
+            r#"ADD COLUMN "aspect" serial NOT NULL UNIQUE PRIMARY KEY"#,
         ]
         .join(" ")
     );