From 694f873d61ed8d2f09032ae0c0dbec4b71c3719e Mon Sep 17 00:00:00 2001
From: dyma solovei <dmytro.y.solovei@gmail.com>
Date: Sun, 27 Oct 2024 16:30:24 +0100
Subject: [PATCH] fix: append IDENTITY to ADD COLUMN statement if needed

---
 dialect/pgdialect/alter_table.go | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/dialect/pgdialect/alter_table.go b/dialect/pgdialect/alter_table.go
index 821191c53..7f9ab4492 100644
--- 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
 }