Updating the primary key constraints to be present when the parser do…

…esnt load the sql as intellij format
Jacobbrewer1 · Dec 3, 2024 · 7141802 · 7141802
1 parent 105e978
commit 7141802
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/pkg/models/table.go b/pkg/models/table.go
@@ -60,6 +60,16 @@ func (t *Table) setPrimaryKey(con *ast.Constraint) {
 		t.PrimaryKey.Columns[i].InPrimaryKey = true
 		t.PrimaryKey.Columns[i].InUniqueKey = true
 	}
+
+	for _, col := range t.Columns {
+		for _, pk := range t.PrimaryKey.Columns {
+			if col.Name == pk.Name {
+				col.InPrimaryKey = true
+				col.InUniqueKey = true
+				col.Nullable = false
+			}
+		}
+	}
 }
 
 func (t *Table) addKey(con *ast.Constraint) {