adjustment for pk column tracking after changing that code

JasperFx · Apr 23, 2024 · b001b79 · b001b79
1 parent 33b1253
commit b001b79
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/src/Weasel.Postgresql/Tables/Table.cs b/src/Weasel.Postgresql/Tables/Table.cs
@@ -54,11 +54,6 @@ public IReadOnlyList<string> PrimaryKeyColumns
     {
         get
         {
-            if (!_primaryKeyColumns.Any())
-            {
-                _primaryKeyColumns.AddRange(_columns.Where(x => x.IsPrimaryKey).Select(x => x.Name));
-            }
-
             return _primaryKeyColumns;
         }
     }
@@ -383,6 +378,7 @@ public ColumnExpression AsPrimaryKey()
         {
             Column.IsPrimaryKey = true;
             Column.AllowNulls = false;
+            _parent._primaryKeyColumns.Fill(Column.Name);
             return this;
         }
 
@@ -501,6 +497,7 @@ public ColumnExpression PartitionByRange()
             _parent.PartitionExpressions.Add(Column.Name);
 
             Column.IsPrimaryKey = true;
+            _parent._primaryKeyColumns.Fill(Column.Name);
 
             return this;
         }