fix: append IDENTITY to ADD COLUMN statement if needed

uptrace · Oct 27, 2024 · 694f873 · 694f873
1 parent b1ae32e
commit 694f873
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/dialect/pgdialect/alter_table.go b/dialect/pgdialect/alter_table.go
@@ -122,6 +122,10 @@ func (m *migrator) addColumn(fmter schema.Formatter, b []byte, add *migrate.AddC
 
 	b, _ = add.ColDef.AppendQuery(fmter, b)
 
+	if add.ColDef.IsIdentity {
+		b = appendGeneratedAsIdentity(b)
+	}
+
 	return b, nil
 }